add unit test for pg_catalog.pg_collation table

Author: Giangblackk

rust/cubesql/cubesql/src/compile/engine/context_postgresql.rs

@@ -19,9 +19,9 @@ use super::information_schema::postgres::{
     InfoSchemaTestingBlockingProvider, InfoSchemaTestingDatasetProvider, PgCatalogAmProvider,
     PgCatalogAttrdefProvider, PgCatalogAttributeProvider, PgCatalogAuthMembersProvider,
     PgCatalogAvailableExtensionVersionsProvider, PgCatalogCastProvider, PgCatalogClassProvider,
-    PgCatalogConstraintProvider, PgCatalogDatabaseProvider, PgCatalogDependProvider,
-    PgCatalogDescriptionProvider, PgCatalogEnumProvider, PgCatalogEventTriggerProvider,
-    PgCatalogExtensionProvider, PgCatalogForeignDataWrapperProvider,
+    PgCatalogCollationProvider, PgCatalogConstraintProvider, PgCatalogDatabaseProvider,
+    PgCatalogDependProvider, PgCatalogDescriptionProvider, PgCatalogEnumProvider,
+    PgCatalogEventTriggerProvider, PgCatalogExtensionProvider, PgCatalogForeignDataWrapperProvider,
     PgCatalogForeignServerProvider, PgCatalogForeignTableProvider, PgCatalogIndexProvider,
     PgCatalogInheritsProvider, PgCatalogLanguageProvider, PgCatalogLocksProvider,
     PgCatalogMatviewsProvider, PgCatalogNamespaceProvider, PgCatalogOperatorProvider,
@@ -99,6 +99,8 @@ impl DatabaseProtocol {
             "pg_catalog.pg_index".to_string()
         } else if let Some(_) = any.downcast_ref::<PgCatalogClassProvider>() {
             "pg_catalog.pg_class".to_string()
+        } else if let Some(_) = any.downcast_ref::<PgCatalogCollationProvider>() {
+            "pg_catalog.pg_collation".to_string()
         } else if let Some(_) = any.downcast_ref::<PgCatalogProcProvider>() {
             "pg_catalog.pg_proc".to_string()
         } else if let Some(_) = any.downcast_ref::<PgCatalogSettingsProvider>() {
@@ -377,6 +379,7 @@ impl DatabaseProtocol {
                 "pg_class" => {
                     return Some(Arc::new(PgCatalogClassProvider::new(&context.meta.tables)))
                 }
+                "pg_collation" => return Some(Arc::new(PgCatalogCollationProvider::new())),
                 "pg_proc" => return Some(Arc::new(PgCatalogProcProvider::new())),
                 "pg_settings" => {
                     return Some(Arc::new(PgCatalogSettingsProvider::new(

rust/cubesql/cubesql/src/compile/mod.rs

@@ -17698,4 +17698,17 @@ LIMIT {{ limit }}{% endif %}"#.to_string(),
             }
         )
     }
+
+    #[tokio::test]
+    async fn test_pg_collation() -> Result<(), CubeError> {
+        insta::assert_snapshot!(
+            "pg_collation",
+            execute_query(
+                "SELECT * FROM pg_catalog.pg_collation".to_string(),
+                DatabaseProtocol::PostgreSQL
+            )
+            .await?
+        );
+        Ok(())
+    }
 }

rust/cubesql/cubesql/src/compile/snapshots/cubesql__compile__tests__pg_collation.snap

@@ -0,0 +1,13 @@
+---
+source: cubesql/src/compile/mod.rs
+expression: "execute_query(\"SELECT * FROM pg_catalog.pg_collation\".to_string(),\nDatabaseProtocol::PostgreSQL).await?"
+---
++-----+-----------+---------------+-----------+--------------+---------------------+--------------+---------------+-------------+-----------+------------+--------------+-------------+
+| oid | collname  | collnamespace | collowner | collprovider | collisdeterministic | collencoding | colliculocale | collcollate | collctype | colllocale | collicurules | collversion |
++-----+-----------+---------------+-----------+--------------+---------------------+--------------+---------------+-------------+-----------+------------+--------------+-------------+
+| 100 | default   | 11            | 10        | d            | true                | -1           | NULL          | NULL        | NULL      | NULL       | NULL         | NULL        |
+| 950 | C         | 11            | 10        | c            | true                | -1           | C             | C           | C         | NULL       | NULL         | NULL        |
+| 951 | POSIX     | 11            | 10        | c            | true                | -1           | POSIX         | POSIX       | POSIX     | NULL       | NULL         | NULL        |
+| 962 | usc_basic | 11            | 10        | c            | true                | 6            | C             | C           | C         | NULL       | NULL         | NULL        |
+| 963 | unicode   | 11            | 10        | i            | true                | -1           | NULL          | NULL        | NULL      | und        | NULL         | 153.121     |
++-----+-----------+---------------+-----------+--------------+---------------------+--------------+---------------+-------------+-----------+------------+--------------+-------------+