[test] Add more tests for PrivilegedCatalog (#3303)

apache · May 7, 2024 · 4b86018 · 4b86018
1 parent 36eb209
commit 4b86018
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/paimon-core/src/main/java/org/apache/paimon/privilege/PrivilegedCatalog.java b/paimon-core/src/main/java/org/apache/paimon/privilege/PrivilegedCatalog.java
@@ -54,6 +54,14 @@ public PrivilegedCatalog(Catalog wrapped, PrivilegeManager privilegeManager) {
         this.privilegeManager = privilegeManager;
     }
 
+    public Catalog wrapped() {
+        return wrapped;
+    }
+
+    public PrivilegeManager privilegeManager() {
+        return privilegeManager;
+    }
+
     @Override
     public boolean caseSensitive() {
         return wrapped.caseSensitive();

diff --git a/...n/src/test/java/org/apache/paimon/flink/procedure/privilege/PrivilegeProcedureITCase.java b/...n/src/test/java/org/apache/paimon/flink/procedure/privilege/PrivilegeProcedureITCase.java
@@ -18,8 +18,13 @@
 
 package org.apache.paimon.flink.procedure.privilege;
 
+import org.apache.paimon.catalog.Catalog;
+import org.apache.paimon.catalog.FileSystemCatalog;
+import org.apache.paimon.flink.FlinkCatalog;
 import org.apache.paimon.flink.util.AbstractTestBase;
+import org.apache.paimon.privilege.FileBasedPrivilegeManager;
 import org.apache.paimon.privilege.NoPrivilegeException;
+import org.apache.paimon.privilege.PrivilegedCatalog;
 
 import org.apache.flink.table.api.TableEnvironment;
 import org.apache.flink.types.Row;
@@ -74,6 +79,17 @@ public void testUserPrivileges() throws Exception {
                                 + "  'warehouse' = '%s'\n"
                                 + ")",
                         path));
+
+        org.apache.flink.table.catalog.Catalog catalog = tEnv.getCatalog("anonymouscat").get();
+        assertThat(catalog).isInstanceOf(FlinkCatalog.class);
+        Catalog paimonCatalog = ((FlinkCatalog) catalog).catalog();
+        assertThat(paimonCatalog).isInstanceOf(PrivilegedCatalog.class);
+        PrivilegedCatalog privilegedCatalog = (PrivilegedCatalog) paimonCatalog;
+        assertThat(privilegedCatalog.wrapped()).isInstanceOf(FileSystemCatalog.class);
+        assertThat(privilegedCatalog.privilegeManager())
+                .isInstanceOf(FileBasedPrivilegeManager.class);
+        assertThat(privilegedCatalog.privilegeManager().privilegeEnabled()).isTrue();
+
         tEnv.executeSql("USE CATALOG anonymouscat");
         assertNoPrivilege(
                 () -> tEnv.executeSql("INSERT INTO mydb.T1 VALUES (1, 11), (2, 21)").await());