This is an automated email from the ASF dual-hosted git repository.
dinglei pushed a commit to branch develop
in repository https://gitbox.apache.org/repos/asf/rocketmq.git
The following commit(s) were added to refs/heads/develop by this push:
new be69a77 [RIP-10] Add test cases for ServiceThread (#862)
be69a77 is described below
commit be69a773ffe1e69efc12075e1cd59c6e67e5b9b6
Author: froyo2 <[email protected]>
AuthorDate: Tue May 14 15:23:02 2019 +0800
[RIP-10] Add test cases for ServiceThread (#862)
---
.../apache/rocketmq/common/ServiceThreadTest.java | 138 +++++++++++++++++++++
1 file changed, 138 insertions(+)
diff --git
a/common/src/test/java/org/apache/rocketmq/common/ServiceThreadTest.java
b/common/src/test/java/org/apache/rocketmq/common/ServiceThreadTest.java
new file mode 100644
index 0000000..70270dd
--- /dev/null
+++ b/common/src/test/java/org/apache/rocketmq/common/ServiceThreadTest.java
@@ -0,0 +1,138 @@
+/*
+ * 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.rocketmq.common;
+
+import static org.junit.Assert.assertEquals;
+
+import org.junit.Test;
+
+public class ServiceThreadTest {
+
+ @Test
+ public void testShutdown() {
+ shutdown(false, false);
+ shutdown(false, true);
+ shutdown(true, false);
+ shutdown(true, true);
+ }
+
+ @Test
+ public void testStop() {
+ stop(true);
+ stop(false);
+ }
+
+ @Test
+ public void testMakeStop() {
+ ServiceThread testServiceThread = startTestServiceThread();
+ testServiceThread.makeStop();
+ assertEquals(true, testServiceThread.isStopped());
+ }
+
+ @Test
+ public void testWakeup() {
+ ServiceThread testServiceThread = startTestServiceThread();
+ testServiceThread.wakeup();
+ assertEquals(true, testServiceThread.hasNotified.get());
+ assertEquals(0, testServiceThread.waitPoint.getCount());
+ }
+
+ @Test
+ public void testWaitForRunning() {
+ ServiceThread testServiceThread = startTestServiceThread();
+ // test waitForRunning
+ testServiceThread.waitForRunning(1000);
+ assertEquals(false, testServiceThread.hasNotified.get());
+ assertEquals(1, testServiceThread.waitPoint.getCount());
+ // test wake up
+ testServiceThread.wakeup();
+ assertEquals(true, testServiceThread.hasNotified.get());
+ assertEquals(0, testServiceThread.waitPoint.getCount());
+ // repeat waitForRunning
+ testServiceThread.waitForRunning(1000);
+ assertEquals(false, testServiceThread.hasNotified.get());
+ assertEquals(0, testServiceThread.waitPoint.getCount());
+ // repeat waitForRunning again
+ testServiceThread.waitForRunning(1000);
+ assertEquals(false, testServiceThread.hasNotified.get());
+ assertEquals(1, testServiceThread.waitPoint.getCount());
+ }
+
+ private ServiceThread startTestServiceThread() {
+ return startTestServiceThread(false);
+ }
+
+ private ServiceThread startTestServiceThread(boolean daemon) {
+ ServiceThread testServiceThread = new ServiceThread() {
+
+ @Override
+ public void run() {
+ doNothing();
+ }
+
+ private void doNothing() {}
+
+ @Override
+ public String getServiceName() {
+ return "TestServiceThread";
+ }
+ };
+ testServiceThread.thread.setDaemon(daemon);
+ // test start
+ testServiceThread.start();
+ assertEquals(false, testServiceThread.isStopped());
+ return testServiceThread;
+ }
+
+ public void shutdown(boolean daemon, boolean interrupt) {
+ ServiceThread testServiceThread = startTestServiceThread(daemon);
+ shutdown0(interrupt, testServiceThread);
+ // repeat
+ shutdown0(interrupt, testServiceThread);
+ }
+
+ private void shutdown0(boolean interrupt, ServiceThread testServiceThread)
{
+ if (interrupt) {
+ testServiceThread.shutdown(true);
+ } else {
+ testServiceThread.shutdown();
+ }
+ assertEquals(true, testServiceThread.isStopped());
+ assertEquals(true, testServiceThread.hasNotified.get());
+ assertEquals(0, testServiceThread.waitPoint.getCount());
+ }
+
+ public void stop(boolean interrupt) {
+ ServiceThread testServiceThread = startTestServiceThread();
+ stop0(interrupt, testServiceThread);
+ // repeat
+ stop0(interrupt, testServiceThread);
+ }
+
+ private void stop0(boolean interrupt, ServiceThread testServiceThread) {
+ if (interrupt) {
+ testServiceThread.stop(true);
+ } else {
+ testServiceThread.stop();
+ }
+ assertEquals(true, testServiceThread.isStopped());
+ assertEquals(true, testServiceThread.hasNotified.get());
+ assertEquals(0, testServiceThread.waitPoint.getCount());
+ }
+
+}
