You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@iotdb.apache.org by ja...@apache.org on 2022/10/29 11:29:22 UTC
[iotdb] 01/01: Add UT for CreateCQProcedure
This is an automated email from the ASF dual-hosted git repository.
jackietien pushed a commit to branch AddCQProcedureTest
in repository https://gitbox.apache.org/repos/asf/iotdb.git
commit d6a3c50cda46249d66c83769d4793dc70f846cf2
Author: JackieTien97 <ja...@gmail.com>
AuthorDate: Sat Oct 29 19:29:03 2022 +0800
Add UT for CreateCQProcedure
---
confignode/pom.xml | 5 ++
.../procedure/store/ProcedureFactory.java | 1 +
.../iotdb/confignode/service/ConfigNode.java | 6 ++
.../procedure/impl/CreateCQProcedureTest.java | 87 ++++++++++++++++++++++
4 files changed, 99 insertions(+)
diff --git a/confignode/pom.xml b/confignode/pom.xml
index 90a3b855d1..d4829295ec 100644
--- a/confignode/pom.xml
+++ b/confignode/pom.xml
@@ -59,6 +59,11 @@
<groupId>commons-io</groupId>
<artifactId>commons-io</artifactId>
</dependency>
+ <dependency>
+ <groupId>org.powermock</groupId>
+ <artifactId>powermock-api-mockito2</artifactId>
+ <scope>test</scope>
+ </dependency>
</dependencies>
<build>
<plugins>
diff --git a/confignode/src/main/java/org/apache/iotdb/confignode/procedure/store/ProcedureFactory.java b/confignode/src/main/java/org/apache/iotdb/confignode/procedure/store/ProcedureFactory.java
index 9dc6c75e7b..197a2baf99 100644
--- a/confignode/src/main/java/org/apache/iotdb/confignode/procedure/store/ProcedureFactory.java
+++ b/confignode/src/main/java/org/apache/iotdb/confignode/procedure/store/ProcedureFactory.java
@@ -101,6 +101,7 @@ public class ProcedureFactory implements IProcedureFactory {
procedure =
new CreateCQProcedure(
ConfigNode.getInstance().getConfigManager().getCQManager().getExecutor());
+ break;
case DEACTIVATE_TEMPLATE_PROCEDURE:
procedure = new DeactivateTemplateProcedure();
break;
diff --git a/confignode/src/main/java/org/apache/iotdb/confignode/service/ConfigNode.java b/confignode/src/main/java/org/apache/iotdb/confignode/service/ConfigNode.java
index c3a5692f5e..494e9daee8 100644
--- a/confignode/src/main/java/org/apache/iotdb/confignode/service/ConfigNode.java
+++ b/confignode/src/main/java/org/apache/iotdb/confignode/service/ConfigNode.java
@@ -27,6 +27,7 @@ import org.apache.iotdb.commons.service.JMXService;
import org.apache.iotdb.commons.service.RegisterManager;
import org.apache.iotdb.commons.service.metric.MetricService;
import org.apache.iotdb.commons.udf.service.UDFClassLoaderManager;
+import org.apache.iotdb.commons.utils.TestOnly;
import org.apache.iotdb.confignode.client.ConfigNodeRequestType;
import org.apache.iotdb.confignode.client.sync.SyncConfigNodeClientPool;
import org.apache.iotdb.confignode.conf.ConfigNodeConfig;
@@ -295,6 +296,11 @@ public class ConfigNode implements ConfigNodeMBean {
return configManager;
}
+ @TestOnly
+ public void setConfigManager(ConfigManager configManager) {
+ this.configManager = configManager;
+ }
+
private static class ConfigNodeHolder {
private static final ConfigNode INSTANCE = new ConfigNode();
diff --git a/confignode/src/test/java/org/apache/iotdb/confignode/procedure/impl/CreateCQProcedureTest.java b/confignode/src/test/java/org/apache/iotdb/confignode/procedure/impl/CreateCQProcedureTest.java
new file mode 100644
index 0000000000..d46c2572b4
--- /dev/null
+++ b/confignode/src/test/java/org/apache/iotdb/confignode/procedure/impl/CreateCQProcedureTest.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.iotdb.confignode.procedure.impl;
+
+import org.apache.iotdb.confignode.manager.ConfigManager;
+import org.apache.iotdb.confignode.manager.cq.CQManager;
+import org.apache.iotdb.confignode.procedure.impl.cq.CreateCQProcedure;
+import org.apache.iotdb.confignode.procedure.store.ProcedureFactory;
+import org.apache.iotdb.confignode.rpc.thrift.TCreateCQReq;
+import org.apache.iotdb.confignode.service.ConfigNode;
+import org.apache.iotdb.tsfile.utils.PublicBAOS;
+
+import org.junit.Test;
+import org.mockito.Mockito;
+
+import java.io.DataOutputStream;
+import java.nio.ByteBuffer;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.fail;
+
+public class CreateCQProcedureTest {
+
+ @Test
+ public void serializeDeserializeTest() {
+
+ PublicBAOS byteArrayOutputStream = new PublicBAOS();
+ DataOutputStream outputStream = new DataOutputStream(byteArrayOutputStream);
+
+ String sql = "create cq testCq1 BEGIN select s1 into root.backup.d1(s1) from root.sg.d1 END";
+
+ TCreateCQReq req =
+ new TCreateCQReq(
+ "testCq1",
+ 1000,
+ 0,
+ 1000,
+ 0,
+ (byte) 0,
+ "select s1 into root.backup.d1(s1) from root.sg.d1",
+ sql,
+ "Asia",
+ "root");
+ ScheduledExecutorService executor = Executors.newScheduledThreadPool(1);
+ CreateCQProcedure createCQProcedure1 = new CreateCQProcedure(req, executor);
+
+ CQManager cqManager = Mockito.mock(CQManager.class);
+ Mockito.when(cqManager.getExecutor()).thenReturn(executor);
+ ConfigManager configManager = Mockito.mock(ConfigManager.class);
+ Mockito.when(configManager.getCQManager()).thenReturn(cqManager);
+ ConfigNode configNode = ConfigNode.getInstance();
+ configNode.setConfigManager(configManager);
+
+ try {
+ createCQProcedure1.serialize(outputStream);
+ ByteBuffer buffer =
+ ByteBuffer.wrap(byteArrayOutputStream.getBuf(), 0, byteArrayOutputStream.size());
+
+ CreateCQProcedure createCQProcedure2 =
+ (CreateCQProcedure) ProcedureFactory.getInstance().create(buffer);
+ assertEquals(createCQProcedure1, createCQProcedure2);
+ } catch (Exception e) {
+ e.printStackTrace();
+ fail(e.getMessage());
+ } finally {
+ executor.shutdown();
+ }
+ }
+}