class TestRepository(unittest.TestCase): def setUp(self): """Call before every test case.""" self.db = DemoRepository(settings.DB_TEST_CONNECTION_STRING) #self.db = DemoRepository(settings.DB_CONNECTION_STRING) log.setLevel(logging.DEBUG) def tearDown(self): self.db.end_session def test_get_ballots(self): result = self.db.get_ballots() print(result) self.assertTrue(len(result) > 0) def test_search_voters(self): result = self.db.search_voters(['jones']) print(result) self.assertTrue(len(result) > 0) result = self.db.search_voters(['jones', 'ellen']) print(result) self.assertTrue(len(result) == 1) def test_get_identities(self): api.debug_create_test_request(1) result = self.db.get_identities() print(result) self.assertTrue(len(result) > 0) def test_get_identity(self): api.debug_create_test_request(1) result = api.verifier_list_requests() id = self.db.get_identity(result['result'][0]['id']) print(id) self.assertIsNotNone(id)
""" The flask application package. """ from flask import Flask from werkzeug.contrib.cache import FileSystemCache from verifier import settings from modules.helpers import setup_logging log = setup_logging(settings.APP_LOGS, "verifier.log", settings.LOG_LEVEL_FILE, settings.LOG_LEVEL_CONSOLE) from verifier.data.demo_repository import DemoRepository app = Flask(__name__) app.config.from_pyfile("settings.py") cache = FileSystemCache(settings.APP_CACHE) # you should be able to use any repository that derives from ballot_box.data.base_repository.BaseRepository db = DemoRepository(settings.DB_CONNECTION_STRING) import modules.context_processor import views from modules import helpers @app.teardown_appcontext def shutdown_session(exception=None): db.end_session()
def setUp(self): """Call before every test case.""" self.db = DemoRepository(settings.DB_TEST_CONNECTION_STRING) #self.db = DemoRepository(settings.DB_CONNECTION_STRING) log.setLevel(logging.DEBUG)