AJ-675： add test case about loadBalance

Author: weisun123

test/com/xxdb/LoadBalanceTest.java

@@ -536,7 +536,7 @@ public void Test_getConnection_enableHighAvailability_true_enableLoadBalance_fal
     @Test
     public void Test_DBConnectionPool_enableHighAvailability_false_loadBalance_false() throws SQLException, ClassNotFoundException, IOException, InterruptedException {
         DBConnectionPool pool1 = new ExclusiveDBConnectionPool(HOST,PORT,"admin","123456",100,false,false);
-        Thread.sleep(10000);
+        Thread.sleep(1000);
         DBConnection connection1 = new DBConnection();
         connection1.connect(HOST, PORT, "admin", "123456",false);
         connection1.run("sleep(3000)");
@@ -549,7 +549,7 @@ public void Test_DBConnectionPool_enableHighAvailability_false_loadBalance_false
     @Test
     public void Test_DBConnectionPool_enableHighAvailability_true_loadBalance_false() throws SQLException, ClassNotFoundException, IOException, InterruptedException {
         DBConnectionPool pool1 = new ExclusiveDBConnectionPool(HOST,PORT,"admin","123456",100,false,true);
-        Thread.sleep(10000);
+        Thread.sleep(1000);
         DBConnection connection1 = new DBConnection();
         connection1.connect(HOST, PORT, "admin", "123456",false);
         connection1.run("sleep(3000)");
@@ -643,7 +643,7 @@ public void Test_DBConnectionPool_enableHighAvailability_true_loadBalance_true()
         DBConnectionPool pool1 = new ExclusiveDBConnectionPool(HOST,PORT,"admin","123456",100,true,true,ipports,null, false, false, false);
         DBConnection connection1 = new DBConnection();
         connection1.connect(HOST, PORT, "admin", "123456",true);
-        connection1.run("sleep(1000)");
+        connection1.run("sleep(2000)");
         BasicTable re = (BasicTable) connection1.run("select port ,connectionNum  from rpc(getControllerAlias(),getClusterPerf) where mode= 0");
         for (int i = 0; i < re.rows(); i++) {
             System.out.println("port:" + re.getColumn(0).get(i) + " connectionNum:" + re.getColumn(1).get(i));
@@ -659,34 +659,41 @@ public void Test_DBConnectionPool_enableHighAvailability_true_loadBalance_true_1
         DBConnection controller_conn = new DBConnection();
         controller_conn.connect(controller_host, controller_port, "admin", "123456");
         controller_conn.run("try{stopDataNode('"+HOST+":"+PORT+"')}catch(ex){}");
+        controller_conn.run("sleep(1000)");
         DBConnectionPool pool1 = new ExclusiveDBConnectionPool(HOST,PORT,"admin","123456",100,true,true,ipports,null, false, false, false);
         controller_conn.run("try{startDataNode('"+HOST+":"+PORT+"')}catch(ex){}");
+        controller_conn.run("sleep(1000)");
         DBConnection connection1 = new DBConnection();
         connection1.connect(HOST, PORT, "admin", "123456",false);
         connection1.run("sleep(3000)");
-        BasicTable re = (BasicTable) connection1.run("select port ,connectionNum  from rpc(getControllerAlias(),getClusterPerf) where mode= 0");
+        BasicTable node1 = (BasicTable)connection1.run("select port ,connectionNum  from rpc(getControllerAlias(),getClusterPerf) where mode= 0 and port ="+ipports[1].split(":")[1]);
+        System.out.println(node1.getString());
+        Assert.assertEquals(true, Integer.valueOf(node1.getColumn(1).get(0).toString())>=50);
+
+        BasicTable node2 = (BasicTable)connection1.run("select port ,connectionNum  from rpc(getControllerAlias(),getClusterPerf) where mode= 0 and port ="+ipports[2].split(":")[1]);
+        System.out.println(node2.getString());
+        Assert.assertEquals(true, Integer.valueOf(node2.getColumn(1).get(0).toString())>=25);
+        pool1.shutdown();
+    }
+    @Test
+    public void Test_DBConnectionPool_enableHighAvailability_false_loadBalance_true() throws SQLException, ClassNotFoundException, IOException {
+        DBConnectionPool pool1 = new ExclusiveDBConnectionPool(HOST,PORT,"admin","123456",100,true,false,null,null, false, false, false);
+        DBConnection connection1 = new DBConnection();
+        connection1.connect(HOST, PORT, "admin", "123456",false);
+        connection1.run("sleep(3000)");
+        BasicTable re = (BasicTable) connection1.run("select port ,connectionNum  from rpc(getControllerAlias(),getClusterPerf) where mode in [0,4];");
         for (int i = 0; i < re.rows(); i++) {
             System.out.println("port:" + re.getColumn(0).get(i) + " connectionNum:" + re.getColumn(1).get(i));
             String port = re.getColumn(0).get(i).toString();
             String connectionNum = re.getColumn(1).get(i).toString();
             if(Integer.valueOf(port)!=PORT) {
-                assertEquals(true, Integer.valueOf(connectionNum) > 25);
+                assertEquals(true, Integer.valueOf(connectionNum) > 20);
                 assertEquals(true, Integer.valueOf(connectionNum) < 50);
             }
         }
         pool1.shutdown();
     }
     @Test
-    public void Test_DBConnectionPool_enableHighAvailability_false_loadBalance_true() throws SQLException, ClassNotFoundException, IOException {
-        String re = null;
-        try{
-            DBConnectionPool pool1 = new ExclusiveDBConnectionPool(HOST,PORT,"admin","123456",100,true,false,null,null, false, false, false);
-        }catch(Exception ex){
-            re = ex.getMessage();
-        }
-        Assert.assertEquals("Cannot only enable loadbalance but not enable highAvailablity.",re);
-    }
-    @Test
     public void Test_DBConnectionPool_enableHighAvailability_true_loadBalance_false_site_not_null() throws SQLException, ClassNotFoundException, IOException {
         DBConnection controller_conn = new DBConnection();
         controller_conn.connect(controller_host, controller_port, "admin", "123456");
@@ -705,16 +712,16 @@ public void Test_DBConnectionPool_enableHighAvailability_true_loadBalance_false_
         controller_conn.run("5000");
         List<DBTask> tasks = new ArrayList<>();
         for (int i = 0; i < 100; i++){
-            BasicDBTask task = new BasicDBTask("t = streamTable(10:0,`a`b,[INT,INT]);\n insert into t values(1,1);");
+            BasicDBTask task = new BasicDBTask("getNodePort();");
             tasks.add(task);
         }
         pool1.execute(tasks);
+        pool1.waitForThreadCompletion();
         BasicTable node2 = (BasicTable)controller_conn.run("select port ,connectionNum  from rpc(getControllerAlias(),getClusterPerf) where mode= 0 and port ="+PORT);
         System.out.println(node2.getString());
         Assert.assertEquals(true, Integer.valueOf(node2.getColumn(1).get(0).toString())>=100);
         controller_conn.run("try{startDataNode('"+HOST+":"+ipportArray[0].split(":")[1]+"')}catch(ex){}");
         controller_conn.run("2000");
         pool1.shutdown();
-
     }
 }