def setUp(self):
self.database = get_database()
self.db_cleanup()
self.user_data = EndpointShared().user_data
def test_save_result(service):
with service.client:
request_data = {
"aliases": [{
"first_name": "John",
"last_name": "Doe",
"middle_name": "Test",
"birth_date": "01/01/1980",
}]
}
record = CrawlerFactory.create()
service.login(service.user_data["user1"]["email"],
service.user_data["user1"]["password"])
g.database = get_database()
stats.save_result(request_data, record)
g.database.connection.commit()
service.check_search_result_saved(service.user_data["user1"]["user_id"],
request_data,
num_eligible_charges=6,
num_charges=9)
def check_search_event_saved(self, user_id, request_data):
with self.app.app_context():
g.database = get_database()
hashed_search_params = self.__hash_search_params(user_id, request_data)
g.database.cursor.execute(
"""
SELECT * FROM SEARCH_RESULTS
WHERE hashed_search_params = %(hashed_search_params)s
;
""",
{"hashed_search_params": hashed_search_params},
)
result = g.database.cursor.fetchone()._asdict()
def setup(self):
self.app.add_url_rule("/api/test/user_protected", view_func=UserProtectedView.as_view("user_protected"))
self.app.add_url_rule("/api/test/admin_protected", view_func=AdminProtectedView.as_view("admin_protected"))
with self.app.app_context():
g.database = get_database()
self.__db_cleanup()
self.__create_test_user("user1")
self.__create_test_user("user2")
self.__create_test_user("admin")
g.database.connection.commit()
g.database.close_connection()
def check_search_result_saved(self, user_id, request_data, num_eligible_charges, num_charges):
with self.app.app_context():
g.database = get_database()
hashed_search_params = self.__hash_search_params(user_id, request_data)
g.database.cursor.execute(
"""
SELECT * FROM SEARCH_RESULTS
WHERE hashed_search_params = %(hashed_search_params)s
;
""",
{"hashed_search_params": hashed_search_params},
)
result = g.database.cursor.fetchone()._asdict()
assert result["num_eligible_charges"] == num_eligible_charges
assert result["num_charges"] == num_charges
def test_get_database_connection(self):
with self.app.app_context():
g.database = get_database()
assert g.database
query = "SELECT * FROM users;"
g.database.cursor.execute(query, ())
rows = g.database.cursor.fetchall()
assert rows or rows == []
g.database.close_connection()
assert g.database.cursor.closed
def test_save_search_event(service):
with service.client:
request_data = {
"aliases": [{
"first_name": "John",
"last_name": "Doe",
"middle_name": "Test",
"birth_date": "01/01/1980",
}]
}
service.login(service.user_data["user1"]["email"],
service.user_data["user1"]["password"])
g.database = get_database()
save_search_event(request_data["aliases"])
g.database.connection.commit()
service.check_search_event_saved(service.user_data["user1"]["user_id"],
request_data)
def before():
g.database = get_database()
def setUp(self):
self.database = get_database()
self.db_cleanup()
def teardown(self):
with self.app.app_context():
g.database = get_database()
self.__db_cleanup()
g.database.close_connection()
