[CockroachDB] Add CERT test

malwaregarry · malwaregarry · commit 880ec6f3f269 · 2024-10-04T14:13:56.000+08:00
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -173,6 +173,7 @@ jobs:
         run: |
           COCKROACHDB_AVAILABLE=true mvn -Dtest=TestCockroachDBNoREC test
           COCKROACHDB_AVAILABLE=true mvn -Dtest=TestCockroachDBTLP test
+          COCKROACHDB_AVAILABLE=true mvn -Dtest=TestCockroachDBCERT test
 
   cockroachdb-qpg:
     name: QPG Tests (CockroachDB)
diff --git a/test/sqlancer/dbms/TestCockroachDBCERT.java b/test/sqlancer/dbms/TestCockroachDBCERT.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 TestCockroachDBCERT {
+
+    @Test
+    public void testCockroachDBCERT() {
+        assumeTrue(TestConfig.isEnvironmentTrue(TestConfig.COCKROACHDB_ENV));
+        assertEquals(0, Main.executeMain(new String[] { "--random-seed", "0", "--timeout-seconds", TestConfig.SECONDS,
+                "--num-queries", TestConfig.NUM_QUERIES, "cockroachdb", "--oracle", "CERT" }));
+    }
+
+}
