migrate-junit5-client-module

Author: sherlock-lin

bookkeeper-server/src/test/java/org/apache/bookkeeper/client/BookKeeperAdminTest.java

@@ -1,83 +1,79 @@
 /*
-*
-* 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.
-*
-*/
+ *
+ * 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.bookkeeper.client;
 
 import org.apache.bookkeeper.test.BookKeeperClusterTestCase;
 import org.apache.bookkeeper.test.TestCallbacks.AddCallbackFuture;
 import org.apache.bookkeeper.zookeeper.ZooKeeperWatcherBase;
 import org.apache.zookeeper.ZooKeeper;
-import org.junit.Test;
+import org.junit.jupiter.api.Test;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 /**
  * Test the bookkeeper client while losing a ZK session.
  */
 public class BookKeeperClientZKSessionExpiry extends BookKeeperClusterTestCase {
-    private static final Logger LOG = LoggerFactory.getLogger(BookKeeperClientZKSessionExpiry.class);
 
-    public BookKeeperClientZKSessionExpiry() {
-        super(4);
-    }
+  private static final Logger LOG = LoggerFactory.getLogger(BookKeeperClientZKSessionExpiry.class);
 
-    @Test
-    public void testSessionLossWhileWriting() throws Exception {
+  public BookKeeperClientZKSessionExpiry() {
+    super(4);
+  }
 
-        Thread expiryThread = new Thread() {
-                @Override
-                public void run() {
-                    try {
-                        while (true) {
-                            Thread.sleep(5000);
-                            long sessionId = bkc.getZkHandle().getSessionId();
-                            byte[] sessionPasswd = bkc.getZkHandle().getSessionPasswd();
+  @Test
+  void sessionLossWhileWriting() throws Exception {
 
-                            try {
-                                ZooKeeperWatcherBase watcher = new ZooKeeperWatcherBase(10000, false);
-                                ZooKeeper zk = new ZooKeeper(zkUtil.getZooKeeperConnectString(), 10000,
-                                                             watcher, sessionId, sessionPasswd);
-                                zk.close();
-                            } catch (Exception e) {
-                                LOG.info("Error killing session", e);
-                            }
-                        }
-                    } catch (InterruptedException ie) {
-                        Thread.currentThread().interrupt();
-                        return;
-                    }
-                }
-            };
-        expiryThread.start();
+    Thread expiryThread = new Thread() {
+      @Override
+      public void run() {
+        try {
+          while (true) {
+            Thread.sleep(5000);
+            long sessionId = bkc.getZkHandle().getSessionId();
+            byte[] sessionPasswd = bkc.getZkHandle().getSessionPasswd();
 
-        for (int i = 0; i < 3; i++) {
-            LedgerHandle lh = bkc.createLedger(3, 3, 2, BookKeeper.DigestType.MAC, "foobar".getBytes());
-            for (int j = 0; j < 100; j++) {
-                lh.asyncAddEntry("foobar".getBytes(), new AddCallbackFuture(j), null);
+            try {
+              ZooKeeperWatcherBase watcher = new ZooKeeperWatcherBase(10000, true);
+              ZooKeeper zk = new ZooKeeper(zkUtil.getZooKeeperConnectString(), 10000, watcher,
+                  sessionId,
+                  sessionPasswd);
+              zk.close();
+            } catch (Exception e) {
+              LOG.info("Error killing session", e);
             }
-            startNewBookie();
-            killBookie(0);
+          }
+        } catch (InterruptedException ie) {
+          Thread.currentThread().interrupt();
+          return;
+        }
+      }
+    };
+    expiryThread.start();
 
-            lh.addEntry("lastEntry".getBytes());
+    for (int i = 0; i < 3; i++) {
+      LedgerHandle lh = bkc.createLedger(3, 3, 2, BookKeeper.DigestType.MAC, "foobar".getBytes());
+      for (int j = 0; j < 100; j++) {
+        lh.asyncAddEntry("foobar".getBytes(), new AddCallbackFuture(j), null);
+      }
+      startNewBookie();
+      killBookie(0);
 
-            lh.close();
-        }
+      lh.addEntry("lastEntry".getBytes());
+
+      lh.close();
     }
+  }
 }