Fixes and tests updates

Updates tests so they run with the db_manager and fixes some bugs introduced in d7d42c9.

test/unit/conftest.py

@@ -4,7 +4,10 @@ import requests
 from time import sleep
 from threading import Thread
 
+from pisa.conf import DB_PATH
 from pisa.api import start_api
+from pisa.watcher import Watcher
+from pisa.db_manager import DBManager
 from test.simulator.bitcoind_sim import run_simulator, HOST, PORT
 
 
@@ -20,7 +23,10 @@ def run_bitcoind():
 
 @pytest.fixture(scope='session')
 def run_api():
-    api_thread = Thread(target=start_api)
+    db_manager = DBManager(DB_PATH)
+    watcher = Watcher(db_manager)
+
+    api_thread = Thread(target=start_api, args=[watcher])
     api_thread.daemon = True
     api_thread.start()
 
@@ -33,6 +39,11 @@ def prng_seed():
     random.seed(0)
 
 
+@pytest.fixture(scope='session')
+def db_manager():
+    return DBManager('test_db')
+
+
 def get_random_value_hex(nbytes):
     pseudo_random_value = random.getrandbits(8*nbytes)
     prv_hex = '{:x}'.format(pseudo_random_value)

test/unit/test_cleaner.py

@@ -1,5 +1,4 @@
 import random
-from os import urandom
 from uuid import uuid4
 
 from pisa import logging
@@ -59,23 +58,26 @@ def set_up_jobs(total_jobs):
     return jobs, tx_job_map
 
 
-def test_delete_expired_appointment():
+def test_delete_expired_appointment(db_manager):
+
     for _ in range(ITERATIONS):
         appointments, locator_uuid_map = set_up_appointments(MAX_ITEMS)
         expired_appointments = random.sample(list(appointments.keys()), k=ITEMS)
 
-        Cleaner.delete_expired_appointment(expired_appointments, appointments, locator_uuid_map)
+        Cleaner.delete_expired_appointment(expired_appointments, appointments, locator_uuid_map, db_manager)
 
         assert not set(expired_appointments).issubset(appointments.keys())
 
 
-def test_delete_completed_jobs():
+def test_delete_completed_jobs(db_manager):
+    height = 0
+
     for _ in range(ITERATIONS):
         jobs, tx_job_map = set_up_jobs(MAX_ITEMS)
         selected_jobs = random.sample(list(jobs.keys()), k=ITEMS)
 
         completed_jobs = [(job, 6) for job in selected_jobs]
 
-        Cleaner.delete_completed_jobs(jobs, tx_job_map, completed_jobs, 0)
+        Cleaner.delete_completed_jobs(jobs, tx_job_map, completed_jobs, height, db_manager)
 
         assert not set(completed_jobs).issubset(jobs.keys())

test/unit/test_responder.py

@@ -15,8 +15,8 @@ from pisa.conf import BTC_RPC_USER, BTC_RPC_PASSWD, BTC_RPC_HOST, BTC_RPC_PORT
 
 
 @pytest.fixture(scope="module")
-def responder():
-    return Responder()
+def responder(db_manager):
+    return Responder(db_manager)
 
 
 def create_dummy_job_data(random_txid=False, justice_rawtx=None):
@@ -94,7 +94,9 @@ def test_add_response(responder):
     # setting the state to awake.
     responder.asleep = False
 
-    receipt = responder.add_response(uuid, job.dispute_txid, job.justice_txid, job.justice_rawtx, job.appointment_end)
+    # The block_hash passed to add_response does not matter much now. It will in the future to deal with errors
+    receipt = responder.add_response(uuid, job.dispute_txid, job.justice_txid, job.justice_rawtx, job.appointment_end,
+                                     block_hash=get_random_value_hex(32))
 
     assert receipt.delivered is True
 
@@ -235,12 +237,12 @@ def test_get_txs_to_rebroadcast(responder):
     assert txs_to_rebroadcast == list(txs_missing_too_many_conf.keys())
 
 
-def test_get_completed_jobs():
+def test_get_completed_jobs(db_manager):
     bitcoin_cli = AuthServiceProxy("http://%s:%s@%s:%d" % (BTC_RPC_USER, BTC_RPC_PASSWD, BTC_RPC_HOST, BTC_RPC_PORT))
     initial_height = bitcoin_cli.getblockcount()
 
     # Let's use a fresh responder for this to make it easier to compare the results
-    responder = Responder()
+    responder = Responder(db_manager)
 
     # A complete job is a job that has reached the appointment end with enough confirmations (> MIN_CONFIRMATIONS)
     # We'll create three type of transactions: end reached + enough conf, end reached + no enough conf, end not reached
@@ -285,8 +287,8 @@ def test_get_completed_jobs():
     assert set(completed_jobs_ids) == set(ended_jobs_keys)
 
 
-def test_rebroadcast():
-    responder = Responder()
+def test_rebroadcast(db_manager):
+    responder = Responder(db_manager)
     responder.asleep = False
 
     txs_to_rebroadcast = []
@@ -305,7 +307,8 @@ def test_rebroadcast():
         if (i % 2) == 0:
             txs_to_rebroadcast.append(justice_txid)
 
-    receipts = responder.rebroadcast(txs_to_rebroadcast)
+    # The block_hash passed to rebroadcast does not matter much now. It will in the future to deal with errors
+    receipts = responder.rebroadcast(txs_to_rebroadcast, get_random_value_hex(32))
 
     # All txs should have been delivered and the missed confirmation reset
     for txid, receipt in receipts:

test/unit/test_watcher.py

@@ -38,8 +38,8 @@ with open(PISA_SECRET_KEY, "r") as key_file:
 
 
 @pytest.fixture(scope="module")
-def watcher():
-    return Watcher()
+def watcher(db_manager):
+    return Watcher(db_manager)
 
 
 def generate_dummy_appointment():