This is an automated email from the ASF dual-hosted git repository.
yiguolei pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/doris.git
The following commit(s) were added to refs/heads/master by this push:
new 0bf0f945a2d [test](ut) added UT cases for show broker stmt (#28629)
0bf0f945a2d is described below
commit 0bf0f945a2dd341a207d8c1d79c99d90383be6c8
Author: Nitin-Kashyap <[email protected]>
AuthorDate: Fri Dec 29 23:04:01 2023 +0700
[test](ut) added UT cases for show broker stmt (#28629)
---
.../apache/doris/analysis/ShowBrokerStmtTest.java | 87 ++++++++++++++++++++++
1 file changed, 87 insertions(+)
diff --git
a/fe/fe-core/src/test/java/org/apache/doris/analysis/ShowBrokerStmtTest.java
b/fe/fe-core/src/test/java/org/apache/doris/analysis/ShowBrokerStmtTest.java
new file mode 100644
index 00000000000..bb361f06061
--- /dev/null
+++ b/fe/fe-core/src/test/java/org/apache/doris/analysis/ShowBrokerStmtTest.java
@@ -0,0 +1,87 @@
+// 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.catalog.ScalarType;
+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 ShowBrokerStmtTest {
+ 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 testAnalyze() throws AnalysisException {
+ AtomicBoolean privilege = new AtomicBoolean(false);
+ new MockUp<AccessControllerManager>() {
+ @Mock
+ public boolean checkGlobalPriv(ConnectContext ctx, PrivPredicate
wanted) {
+ return privilege.get();
+ }
+ };
+
+ ShowBrokerStmt stmt = new ShowBrokerStmt();
+ Assertions.assertThrows(AnalysisException.class, () ->
stmt.analyze(analyzer));
+
+ privilege.set(true);
+ stmt.analyze(analyzer);
+ }
+
+ @Test
+ public void getMetaData() {
+ ShowBrokerStmt stmt = new ShowBrokerStmt();
+ ShowResultSetMetaData result = stmt.getMetaData();
+ Assertions.assertEquals(result.getColumnCount(), 7);
+ result.getColumns().forEach(col ->
Assertions.assertEquals(col.getType(), ScalarType.createVarchar(30)));
+ }
+
+
+ @Test
+ public void getRedirectStatus() {
+ ShowBrokerStmt stmt = new ShowBrokerStmt();
+ Assertions.assertEquals(RedirectStatus.FORWARD_NO_SYNC,
stmt.getRedirectStatus());
+
+ ctx.getSessionVariable().forwardToMaster = false;
+ Assertions.assertEquals(RedirectStatus.NO_FORWARD,
stmt.getRedirectStatus());
+ }
+}
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
