[Yugabyte] Update tests

malwaregarry · malwaregarry · commit 1af322b4aaf4 · 2024-09-03T11:59:44.000+08:00
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -522,8 +522,11 @@ jobs:
           docker run -d --name yugabyte -p7000:7000 -p9000:9000 -p5433:5433 -p9042:9042 yugabytedb/yugabyte:latest bin/yugabyted start --daemon=false
           sleep 5
       - name: Run Tests
-        run: YUGABYTE_AVAILABLE=true mvn -Dtest=TestYugabyte test
-
+        run: |
+          YUGABYTE_AVAILABLE=true mvn -Dtest=TestYSQLNoREC test
+          YUGABYTE_AVAILABLE=true mvn -Dtest=TestYSQLTLP test
+          YUGABYTE_AVAILABLE=true mvn -Dtest=TestYSQLPQS test
+          YUGABYTE_AVAILABLE=true mvn -Dtest=TestYCQL test
 
   doris:
     name: DBMS Tests (Apache Doris)
diff --git a/test/sqlancer/dbms/TestConfig.java b/test/sqlancer/dbms/TestConfig.java
@@ -8,6 +8,7 @@ public class TestConfig {
     public static final String DORIS_ENV = "DORIS_AVAILABLE";
     public static final String MARIADB_ENV = "MARIADB_AVAILABLE";
     public static final String POSTGRES_ENV = "POSTGRES_AVAILABLE";
+    public static final String YUGABYTE_ENV = "YUGABYTE_AVAILABLE";
 
     public static boolean isEnvironmentTrue(String key) {
         String value = System.getenv(key);
diff --git a/test/sqlancer/dbms/TestYCQL.java b/test/sqlancer/dbms/TestYCQL.java
@@ -0,0 +1,18 @@
+package sqlancer.dbms;
+
+import org.junit.jupiter.api.Test;
+import sqlancer.Main;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assumptions.assumeTrue;
+
+public class TestYCQL {
+    @Test
+    public void testYCQL() {
+        assumeTrue(TestConfig.isEnvironmentTrue(TestConfig.YUGABYTE_ENV));
+        assertEquals(0,
+                Main.executeMain("--random-seed", "0", "--timeout-seconds", TestConfig.SECONDS, "--username",
+                        "cassandra", "--password", "cassandra", "--num-threads", "1", "--num-queries",
+                        TestConfig.NUM_QUERIES, "ycql"));
+    }
+}
diff --git a/test/sqlancer/dbms/TestYSQLNoREC.java b/test/sqlancer/dbms/TestYSQLNoREC.java
@@ -0,0 +1,19 @@
+package sqlancer.dbms;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assumptions.assumeTrue;
+
+import org.junit.jupiter.api.Test;
+
+import sqlancer.Main;
+
+public class TestYSQLNoREC {
+    @Test
+    public void testYSQLNoREC() {
+        assumeTrue(TestConfig.isEnvironmentTrue(TestConfig.YUGABYTE_ENV));
+        assertEquals(0,
+                Main.executeMain("--random-seed", "0", "--timeout-seconds", TestConfig.SECONDS, "--username",
+                        "yugabyte", "--password", "yugabyte", "--num-threads", "1", "--num-queries",
+                        TestConfig.NUM_QUERIES, "ysql", "--oracle", "NOREC"));
+    }
+}
diff --git a/test/sqlancer/dbms/TestYSQLPQS.java b/test/sqlancer/dbms/TestYSQLPQS.java
@@ -0,0 +1,19 @@
+package sqlancer.dbms;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assumptions.assumeTrue;
+
+import org.junit.jupiter.api.Test;
+
+import sqlancer.Main;
+
+public class TestYSQLPQS {
+    @Test
+    public void testYSQLPQS() {
+        assumeTrue(TestConfig.isEnvironmentTrue(TestConfig.YUGABYTE_ENV));
+        assertEquals(0,
+                Main.executeMain("--random-seed", "0", "--timeout-seconds", TestConfig.SECONDS, "--username",
+                        "yugabyte", "--password", "yugabyte", "--num-threads", "1", "--num-queries",
+                        TestConfig.NUM_QUERIES, "ysql", "--oracle", "PQS"));
+    }
+}
diff --git a/test/sqlancer/dbms/TestYSQLTLP.java b/test/sqlancer/dbms/TestYSQLTLP.java
@@ -0,0 +1,19 @@
+package sqlancer.dbms;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assumptions.assumeTrue;
+
+import org.junit.jupiter.api.Test;
+
+import sqlancer.Main;
+
+public class TestYSQLTLP {
+    @Test
+    public void testYSQLTLP() {
+        assumeTrue(TestConfig.isEnvironmentTrue(TestConfig.YUGABYTE_ENV));
+        assertEquals(0,
+                Main.executeMain("--random-seed", "0", "--timeout-seconds", TestConfig.SECONDS, "--username",
+                        "yugabyte", "--password", "yugabyte", "--num-threads", "1", "--num-queries",
+                        TestConfig.NUM_QUERIES, "ysql", "--oracle", "QUERY_PARTITIONING"));
+    }
+}
diff --git a/test/sqlancer/dbms/TestYugabyte.java b/test/sqlancer/dbms/TestYugabyte.java
