diff --git a/fe/fe-core/src/test/java/org/apache/doris/analysis/ShowCreateMaterializedViewTest.java b/fe/fe-core/src/test/java/org/apache/doris/analysis/ShowCreateMaterializedViewTest.java new file mode 100644 index 0000000000..70bbd617c8 --- /dev/null +++ b/fe/fe-core/src/test/java/org/apache/doris/analysis/ShowCreateMaterializedViewTest.java @@ -0,0 +1,93 @@ +// Licensed to the Apache Software Foundation (ASF) under one +// or more contributor license agreements. See the NOTICE file +// distributed with this work for additional information +// regarding copyright ownership. The ASF licenses this file +// to you under the Apache License, Version 2.0 (the +// "License"); you may not use this file except in compliance +// with the License. You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, +// software distributed under the License is distributed on an +// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +// KIND, either express or implied. See the License for the +// specific language governing permissions and limitations +// under the License. + +package org.apache.doris.analysis; + +import org.apache.doris.common.AnalysisException; +import org.apache.doris.mysql.privilege.AccessControllerManager; +import org.apache.doris.mysql.privilege.PrivPredicate; +import org.apache.doris.qe.ConnectContext; +import org.apache.doris.qe.SessionVariable; +import org.apache.doris.qe.ShowResultSetMetaData; + +import mockit.Mock; +import mockit.MockUp; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; +import org.junit.jupiter.api.Assertions; + +import java.util.concurrent.atomic.AtomicBoolean; + +public class ShowCreateMaterializedViewTest { + private Analyzer analyzer; + private ConnectContext ctx = new ConnectContext(); + + @Before + public void setUp() { + analyzer = AccessTestUtil.fetchAdminAnalyzer(true); + ctx.setSessionVariable(new SessionVariable()); + ctx.setThreadLocalInfo(); + } + + @After + public void tearDown() { + ConnectContext.remove(); + } + + @Test + public void testAnalyse() throws Exception { + AtomicBoolean privilege = new AtomicBoolean(false); + new MockUp() { + @Mock + public boolean checkTblPriv(ConnectContext ctx, String qualifiedDb, String tbl, PrivPredicate wanted) { + return privilege.get(); + } + }; + + /* case 1 */ { + ShowCreateMaterializedViewStmt stmt = new ShowCreateMaterializedViewStmt("test", new TableName("a.b.c")); + Assertions.assertThrows(AnalysisException.class, () -> stmt.analyze(analyzer)); + } + + /* case 2 */ { + ShowCreateMaterializedViewStmt stmt = new ShowCreateMaterializedViewStmt("test", new TableName("", "", "")); + Assertions.assertThrows(AnalysisException.class, () -> stmt.analyze(analyzer)); + } + + /* case 3 */ { + ShowCreateMaterializedViewStmt stmt = new ShowCreateMaterializedViewStmt("test", new TableName("", "", "tbl")); + Assertions.assertThrows(AnalysisException.class, () -> stmt.analyze(analyzer)); + } + + /* case success */ { + privilege.set(true); + ShowCreateMaterializedViewStmt stmt = new ShowCreateMaterializedViewStmt("test", new TableName("", "", "tbl")); + stmt.analyze(analyzer); + Assertions.assertEquals(stmt.toSql(), "SHOW CREATE MATERIALIZED VIEW `test` ON `testDb`.`tbl`"); + Assertions.assertEquals(stmt.getMvName(), "test"); + Assertions.assertEquals(stmt.getTableName(), new TableName("testDb.tbl")); + } + } + + @Test + public void getMetaData() { + ShowCreateMaterializedViewStmt stmt = new ShowCreateMaterializedViewStmt("", null); + ShowResultSetMetaData result = stmt.getMetaData(); + Assertions.assertEquals(result.getColumnCount(), 3); + } +}