Use docker-compose to spin up a local CockroachDB cluster for testing

Author: ckoehn

Makefile

@@ -1,3 +1,4 @@
+COMPOSE=docker-compose -f docker-compose.yml
 ENV_BASE=~/envs
 ENV=${ENV_BASE}/sqlalchemy-cockroachdb
 TOX=${ENV}/bin/tox
@@ -38,3 +39,15 @@ clean:
 .PHONY: detox
 detox: clean
 	rm -rf .tox
+
+.PHONY: db-up
+db-up:
+	${COMPOSE} up -d
+
+.PHONY: db-down
+db-down:
+	${COMPOSE} down
+
+.PHONY: db-recreate
+db-recreate:
+	${COMPOSE} exec cockroach-0 ./cockroach sql --insecure -e 'drop database defaultdb; create database defaultdb;'

docker-compose.yml

@@ -0,0 +1,31 @@
+version: "3.7"
+
+x-cockroach-base: &cockroach-base
+  image: cockroachdb/cockroach:latest-v21.1
+
+services:
+  cockroach-0:
+    <<: *cockroach-base
+    container_name: cockroach-0
+    command: start --insecure --advertise-addr=cockroach-0 --join=cockroach-0,cockroach-1,cockroach-2 --store=type=mem,size=1GB
+    ports:
+      - "26257:26257"
+
+  cockroach-1:
+    <<: *cockroach-base
+    container_name: cockroach-1
+    command: start --insecure --advertise-addr=cockroach-1 --join=cockroach-0,cockroach-1,cockroach-2 --store=type=mem,size=1GB
+
+  cockroach-2:
+    <<: *cockroach-base
+    container_name: cockroach-2
+    command: start --insecure --advertise-addr=cockroach-2 --join=cockroach-0,cockroach-1,cockroach-2 --store=type=mem,size=1GB
+
+  cockroach-init:
+    <<: *cockroach-base
+    container_name: cockroach-init
+    depends_on:
+      - cockroach-0
+      - cockroach-1
+      - cockroach-2
+    command: init --insecure --host=cockroach-0