Add database setup

Dockerfile.test_db

@@ -0,0 +1,4 @@
+FROM postgres
+ENV POSTGRES_PASSWORD=postgres
+ENV POSTGRES_DB=mainnet
+COPY ./database/01_table_creation.sql /docker-entrypoint-initdb.d/

Makefile

@@ -0,0 +1,23 @@
+DOCKER_IMAGE_NAME=test_db_image
+DOCKER_CONTAINER_NAME=test_db_container
+DB_PORT=5432
+
+install:
+	pip install -r requirements.txt
+
+imbalances:
+	python -m src.imbalances_script
+
+daemon:
+	python -m src.daemon
+
+test_db:
+	docker build -t $(DOCKER_IMAGE_NAME) -f Dockerfile.test_db .
+	docker run -d --name $(DOCKER_CONTAINER_NAME) -p $(DB_PORT):$(DB_PORT) $(DOCKER_IMAGE_NAME)
+
+stop_test_db:
+	docker stop $(DOCKER_CONTAINER_NAME) || true
+	docker rm $(DOCKER_CONTAINER_NAME) || true
+	docker rmi $(DOCKER_IMAGE_NAME) || true
+
+.PHONY: install imbalances daemon test_db run_test_db stop_test_db clean

README.md

@@ -24,5 +24,10 @@ python -m src.imbalances_script
 python -m src.daemon
 ```
 
-**To run the basic test in the `tests/` folder:**
- using pytest, simply run the command: `pytest` 
+## Tests
+
+To build and start a local database for testing use the command
+```sh
+docker build -t test_db_image -f Dockerfile.test_db .
+docker run -d --name test_db_container -p 5432:5432 test_db_image
+```

database/01_table_creation.sql

@@ -0,0 +1,23 @@
+CREATE TABLE token_info (
+    token_address bytea PRIMARY KEY,
+    symbol varchar NOT NULL,
+    decimals int NOT NULL
+);
+
+CREATE TABLE token_times (
+    time timestamp NOT NULL,
+    token_address bytea NOT NULL,
+    block_number bigint NOT NULL,
+    tx_hash bytea NOT NULL,
+
+    PRIMARY KEY (time, token_address, tx_hash)
+);
+
+CREATE TABLE prices (
+    time timestamp NOT NULL,
+    token_address bytea NOT NULL,
+    price numeric(60, 18) NOT NULL,
+    source varchar NOT NULL,
+
+    PRIMARY KEY (time, token_address, source)
+);