diff --git a/cockroachdb/__init__.py b/cockroachdb/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/cockroachdb/config.json b/cockroachdb/config.json new file mode 100644 index 0000000..3ed5fa5 --- /dev/null +++ b/cockroachdb/config.json @@ -0,0 +1,17 @@ +{ + "embedded": "no", + "dbms": "cockroachdb", + "image": "cockroachdb/cockroach:v24.1.0", + "container_name": "cockroachdb-sqlancer", + "port": 26257, + "startup_cmd": [ + "start-single-node", + "--insecure" + ], + "username": "root", + "password": "", + "oracle": "NOREC", + "num_threads": 4, + "timeout_seconds": 60 + } + \ No newline at end of file diff --git a/config.json b/config.json index 4a8cc01..dc2b8f7 100644 --- a/config.json +++ b/config.json @@ -1,5 +1,5 @@ { - "dbms_list": ["mysql", "postgres", "sqlite", "tidb"], + "dbms_list": ["mysql", "postgres", "sqlite", "tidb", "cockroachdb"], "dbms": "mysql", "container_name": "mysql-custom", "image": "mysql-custom", diff --git a/test.py b/test.py index e78b96f..75407f5 100644 --- a/test.py +++ b/test.py @@ -18,6 +18,7 @@ def start_db_container(dbms, cfg, script_log, docker_log): image = cfg.get("image") container_name = cfg.get("container_name", f"{dbms}-sqlancer") env_dict = cfg.get("env", {}) + startup_cmd = cfg.get("startup_cmd", []) if not image: script_log.error( "Missing 'image' field in config.json") @@ -36,7 +37,8 @@ def start_db_container(dbms, cfg, script_log, docker_log): "--name", container_name, "--network", "sqlancer-net", *env_vars, - image + image, + *startup_cmd ], docker_log) script_log.info("Starting DBMS container: %s ...", container_name)