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 TestYugabyte {

    String yugabyteAvailable = System.getenv("YUGABYTE_AVAILABLE");
    boolean yugabyteIsAvailable = yugabyteAvailable != null && yugabyteAvailable.equalsIgnoreCase("true");

    @Test
    public void testYCQL() {
        assumeTrue(yugabyteIsAvailable);
        assertEquals(0, Main.executeMain("--random-seed", "0", "--timeout-seconds", TestConfig.SECONDS, "--username",
                "cassandra", "--password", "cassandra", "--num-threads",
                "4", "--num-queries", TestConfig.NUM_QUERIES, "ycql"));
    }

    @Test
    public void testYSQL() {
        assumeTrue(yugabyteIsAvailable);
        assertEquals(0,
                Main.executeMain("--random-seed", "0", "--timeout-seconds", TestConfig.SECONDS, "--username",
                        "yugabyte", "--password", "yugabyte", "--num-threads", "4", "--num-queries",
                        TestConfig.NUM_QUERIES, "ysql"));
    }
}

@@ -505,16 +505,32 @@
 quals(0,
+%0A               
  Main.ex
@@ -632,16 +632,24 @@
         
+        
 %22cassand
@@ -696,32 +696,16 @@
 hreads%22,
-%0A               
  %224%22, %22-
@@ -718,16 +718,40 @@
 ueries%22,
+%0A                       
  TestCon

Reformat YCQL test