From 616116a84120220adb8e36cb364f08b0a638807e Mon Sep 17 00:00:00 2001 From: Frederick Yin Date: Thu, 3 Feb 2022 16:11:20 +0800 Subject: Adapt test suites --- jimbrella/test/test_routine.py | 40 +++++++++++++++++++++------------------- 1 file changed, 21 insertions(+), 19 deletions(-) (limited to 'jimbrella/test/test_routine.py') diff --git a/jimbrella/test/test_routine.py b/jimbrella/test/test_routine.py index 0b15717..acb1f1e 100644 --- a/jimbrella/test/test_routine.py +++ b/jimbrella/test/test_routine.py @@ -5,20 +5,19 @@ import os import shutil import logging from .jform_data import mock_jform_data -from ..routine import sync_database, process_overdue -from ..database import Database +from ..routine import sync_jform, process_overdue +from ..umbrellas import Umbrellas from ..jform import JForm from ..admin_log import AdminLog from ..config import * -"""Set up logging.""" +"""Set up logging. logging.basicConfig( filename="/tmp/jimbrella.log", encoding="utf-8", level=logging.INFO, format="[%(asctime)s] %(levelname)s:%(name)s:%(message)s", -) -logging.info("----- BEGIN ROUTINE TEST -----") +)""" """Spin up a simple Flask app to mimic jForm.""" jform = Flask(__name__) @@ -58,18 +57,21 @@ def api(endpoint): Thread(target=jform.run, kwargs={"port": 5001}).start() -"""Initialize Database and JForm.""" -TEST_DATABASE_PATH = "/tmp/jimbrella.db.csv" -TEST_ADMIN_LOG_PATH = "/tmp/jimbrella.admin.log" -shutil.copyfile(DATABASE_PATH, TEST_DATABASE_PATH) -db = Database(TEST_DATABASE_PATH) -takeaway_jform = JForm("takeaway", "http://localhost:5001/takeaway", "/tmp") -giveback_jform = JForm("giveback", "http://localhost:5001/giveback", "/tmp") -admin = AdminLog(TEST_ADMIN_LOG_PATH) -sync_database(takeaway_jform, giveback_jform, db, admin) -process_overdue(db) +if __name__ == "__main__": + logging.info("----- BEGIN ROUTINE TEST -----") + + """Initialize Database and JForm.""" + TEST_DATABASE_PATH = "/tmp/jimbrella.db" + TEST_ADMIN_LOG_PATH = "/tmp/jimbrella.admin.log" + shutil.copyfile(DATABASE_PATH, TEST_DATABASE_PATH) + db = Umbrellas(TEST_DATABASE_PATH) + takeaway_jform = JForm("takeaway", "http://localhost:5001/takeaway", "/tmp") + giveback_jform = JForm("giveback", "http://localhost:5001/giveback", "/tmp") + admin = AdminLog(TEST_ADMIN_LOG_PATH) + sync_jform(takeaway_jform, giveback_jform, db, admin) + process_overdue(db, admin) -"""Cleanup""" -os.remove(takeaway_jform._bookmark_fp) -os.remove(giveback_jform._bookmark_fp) -logging.info("----- END ROUTINE TEST -----") + """Cleanup""" + os.remove(takeaway_jform._bookmark_fp) + os.remove(giveback_jform._bookmark_fp) + logging.info("----- END ROUTINE TEST -----") -- cgit v1.2.3