[rocketmq] branch develop updated: [RIP-10] Add test cases for CountDownLatch2 (#868)

[di...@apache.org]

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 6a42489  [RIP-10] Add test cases for CountDownLatch2 (#868)
6a42489 is described below

commit 6a42489197a2e881487fd407691b94cac7f27007
Author: flyonsnow <Ni...@gmail.com>
AuthorDate: Tue May 14 16:01:22 2019 +0800

    [RIP-10] Add test cases for CountDownLatch2 (#868)
---
 .../rocketmq/common/CountDownLatch2Test.java       | 123 +++++++++++++++++++++
 1 file changed, 123 insertions(+)

diff --git a/common/src/test/java/org/apache/rocketmq/common/CountDownLatch2Test.java b/common/src/test/java/org/apache/rocketmq/common/CountDownLatch2Test.java
new file mode 100644
index 0000000..3570db6
--- /dev/null
+++ b/common/src/test/java/org/apache/rocketmq/common/CountDownLatch2Test.java
@@ -0,0 +1,123 @@
+/*
+ * 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 org.junit.Test;
+
+import java.util.concurrent.TimeUnit;
+
+import static org.hamcrest.core.Is.is;
+import static org.hamcrest.core.StringContains.containsString;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * CountDownLatch2 Unit Test
+ *
+ * @see CountDownLatch2
+ */
+public class CountDownLatch2Test {
+
+    /**
+     * test constructor with invalid init param
+     *
+     * @see CountDownLatch2#CountDownLatch2(int)
+     */
+    @Test
+    public void testConstructorError() {
+        int count = -1;
+        try {
+            CountDownLatch2 latch = new CountDownLatch2(count);
+        } catch (IllegalArgumentException e) {
+            assertThat(e.getMessage(), is("count < 0"));
+        }
+    }
+
+    /**
+     * test constructor with valid init param
+     *
+     * @see CountDownLatch2#CountDownLatch2(int)
+     */
+    @Test
+    public void testConstructor() {
+        int count = 10;
+        CountDownLatch2 latch = new CountDownLatch2(count);
+        assertEquals("Expected equal", count, latch.getCount());
+        assertThat("Expected contain", latch.toString(), containsString("[Count = " + count + "]"));
+    }
+
+    /**
+     * test await timeout
+     *
+     * @see CountDownLatch2#await(long, TimeUnit)
+     */
+    @Test
+    public void testAwaitTimeout() throws InterruptedException {
+        int count = 1;
+        CountDownLatch2 latch = new CountDownLatch2(count);
+        boolean await = latch.await(10, TimeUnit.MILLISECONDS);
+        assertFalse("Expected false", await);
+
+        latch.countDown();
+        boolean await2 = latch.await(10, TimeUnit.MILLISECONDS);
+        assertTrue("Expected true", await2);
+    }
+
+
+    /**
+     * test reset
+     *
+     * @see CountDownLatch2#countDown()
+     */
+    @Test(timeout = 1000)
+    public void testCountDownAndGetCount() throws InterruptedException {
+        int count = 2;
+        CountDownLatch2 latch = new CountDownLatch2(count);
+        assertEquals("Expected equal", count, latch.getCount());
+        latch.countDown();
+        assertEquals("Expected equal", count - 1, latch.getCount());
+        latch.countDown();
+        latch.await();
+        assertEquals("Expected equal", 0, latch.getCount());
+    }
+
+
+    /**
+     * test reset
+     *
+     * @see CountDownLatch2#reset()
+     */
+    @Test
+    public void testReset() throws InterruptedException {
+        int count = 2;
+        CountDownLatch2 latch = new CountDownLatch2(count);
+        latch.countDown();
+        assertEquals("Expected equal", count - 1, latch.getCount());
+        latch.reset();
+        assertEquals("Expected equal", count, latch.getCount());
+        latch.countDown();
+        latch.countDown();
+        latch.await();
+        assertEquals("Expected equal", 0, latch.getCount());
+        // coverage Sync#tryReleaseShared, c==0
+        latch.countDown();
+        assertEquals("Expected equal", 0, latch.getCount());
+    }
+}