[Postgres] Add CERT test

malwaregarry · malwaregarry · commit e0d072bec939 · 2024-10-01T16:28:47.000+08:00
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -482,6 +482,7 @@ jobs:
           POSTGRES_AVAILABLE=true mvn -Dtest=TestPostgresPQS test
           POSTGRES_AVAILABLE=true mvn -Dtest=TestPostgresTLP test
           POSTGRES_AVAILABLE=true mvn -Dtest=TestPostgresNoREC test
+          POSTGRES_AVAILABLE=true mvn -Dtest=TestPostgresCERT test
 
   presto:
     name: DBMS Tests (Presto)
diff --git a/test/sqlancer/dbms/TestPostgresCERT.java b/test/sqlancer/dbms/TestPostgresCERT.java
@@ -0,0 +1,20 @@
+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 TestPostgresCERT {
+
+    @Test
+    public void testCERT() {
+        assumeTrue(TestConfig.isEnvironmentTrue(TestConfig.POSTGRES_ENV));
+        assertEquals(0,
+                Main.executeMain(new String[] { "--random-seed", "0", "--timeout-seconds", TestConfig.SECONDS,
+                        "--num-threads", "4", "--num-queries", TestConfig.NUM_QUERIES, "postgres", "--test-collations",
+                        "false", "--oracle", "CERT" }));
+    }
+}
