(paimon) branch master updated: [test] Add more tests for PrivilegedCatalog (#3303)

[lzljs3620320]

This is an automated email from the ASF dual-hosted git repository.

lzljs3620320 pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/paimon.git


The following commit(s) were added to refs/heads/master by this push:
     new 4b860187c [test] Add more tests for PrivilegedCatalog (#3303)
4b860187c is described below

commit 4b860187c533e79a7e826b8e6741b17ddb23591a
Author: tsreaper <tsreape...@gmail.com>
AuthorDate: Tue May 7 15:12:46 2024 +0800

    [test] Add more tests for PrivilegedCatalog (#3303)
---
 .../org/apache/paimon/privilege/PrivilegedCatalog.java   |  8 ++++++++
 .../procedure/privilege/PrivilegeProcedureITCase.java    | 16 ++++++++++++++++
 2 files changed, 24 insertions(+)

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
index 7c2ab9493..851cc302a 100644
--- 
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 class PrivilegedCatalog implements Catalog {
         this.privilegeManager = privilegeManager;
     }
 
+    public Catalog wrapped() {
+        return wrapped;
+    }
+
+    public PrivilegeManager privilegeManager() {
+        return privilegeManager;
+    }
+
     @Override
     public boolean caseSensitive() {
         return wrapped.caseSensitive();
diff --git 
a/paimon-flink/paimon-flink-common/src/test/java/org/apache/paimon/flink/procedure/privilege/PrivilegeProcedureITCase.java
 
b/paimon-flink/paimon-flink-common/src/test/java/org/apache/paimon/flink/procedure/privilege/PrivilegeProcedureITCase.java
index d432c02ea..e0c5f5d66 100644
--- 
a/paimon-flink/paimon-flink-common/src/test/java/org/apache/paimon/flink/procedure/privilege/PrivilegeProcedureITCase.java
+++ 
b/paimon-flink/paimon-flink-common/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 class PrivilegeProcedureITCase extends 
AbstractTestBase {
                                 + "  '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());