enums

Author: juleswritescode

crates/pgt_hover/tests/hover_integration_tests.rs

@@ -520,7 +520,7 @@ async fn test_grant_table_hover(test_db: PgPool) {
 }
 
 #[sqlx::test(migrator = "pgt_test_utils::MIGRATIONS")]
-async fn hover_on_types(test_db: PgPool) {
+async fn hover_on_composite_type(test_db: PgPool) {
     let setup = r#"create type compfoo as (f1 int, f2 text);"#;
 
     let query = format!(
@@ -536,3 +536,15 @@ async fn hover_on_types(test_db: PgPool) {
     )
     .await;
 }
+
+#[sqlx::test(migrator = "pgt_test_utils::MIGRATIONS")]
+async fn hover_on_enum_type(test_db: PgPool) {
+    let setup = r#"create type compfoo as ENUM ('yes', 'no');"#;
+
+    let query = format!(
+        "create function getfoo() returns setof comp{}foo as $$ select fooid, fooname from foo $$ language sql;",
+        QueryWithCursorPosition::cursor_marker()
+    );
+
+    test_hover_at_cursor("hover_custom_type_enum", query, Some(setup), &test_db).await;
+}

crates/pgt_hover/tests/snapshots/hover_custom_type_enum.snap

@@ -0,0 +1,24 @@
+---
+source: crates/pgt_hover/tests/hover_integration_tests.rs
+expression: snapshot
+---
+# Input
+```sql
+create function getfoo() returns setof compfoo as $$ select fooid, fooname from foo $$ language sql;
+                                           ↑ hovered here
+```
+
+# Hover Results
+### `public.compfoo` (Custom Type)  
+```plain  
+Enum Permutations:
+- yes
+- no
+
+  
+```  
+---    
+```plain  
+
+  
+```