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()