Wait for Firebase Functions to complete during test setup.

Author: matthias_schaub

mapswipe_workers/tests/integration/test_transfer_results.py

@@ -26,9 +26,10 @@ def test_changes_given_project_id(self):
 
         pg_db = auth.postgresDB()
         sql_query = (
-            "SELECT * FROM results WHERE project_id = '{0}' AND user_id = '{0}'".format(
-                self.project_id
-            )
+            f"SELECT * "
+            f"FROM results "
+            f"WHERE project_id = '{self.project_id}' "
+            f"AND user_id = '{self.project_id}'"
         )
         result = pg_db.retr_query(sql_query)
         self.assertIsNotNone(result)
@@ -43,9 +44,10 @@ def test_changes(self):
 
         pg_db = auth.postgresDB()
         sql_query = (
-            "SELECT * FROM results WHERE project_id = '{0}' and user_id = '{0}'".format(
-                self.project_id
-            )
+            f"SELECT * "
+            f"FROM results "
+            f"WHERE project_id = '{self.project_id}' "
+            f"AND user_id = '{self.project_id}'"
         )
         result = pg_db.retr_query(sql_query)
         self.assertIsNotNone(result)
@@ -56,9 +58,9 @@ def test_user_not_in_postgres(self):
 
         # Make sure user and results are not yet in Postgres
         sql_query = (
-            "DELETE FROM results WHERE user_id = '{0}' and project_id = '{0}'".format(
-                self.project_id
-            )
+            f"DELETE FROM results "
+            f"WHERE user_id = '{self.project_id}' "
+            f"AND project_id = '{self.project_id}'"
         )
         pg_db.query(sql_query)
         sql_query = "DELETE FROM users WHERE user_id = '{0}'".format(self.project_id)
@@ -71,9 +73,10 @@ def test_user_not_in_postgres(self):
         self.assertIsNotNone(result)
 
         sql_query = (
-            "SELECT * FROM results WHERE project_id = '{0}' AND user_id = '{0}'".format(
-                self.project_id
-            )
+            f"SELECT * "
+            f"FROM results "
+            f"WHERE project_id = '{self.project_id}' "
+            f"AND user_id = '{self.project_id}'"
         )
         result = pg_db.retr_query(sql_query)
         self.assertIsNotNone(result)

mapswipe_workers/tests/unittests/test_user_stats.py

@@ -4,9 +4,7 @@
 import pandas as pd
 from pandas.testing import assert_frame_equal
 
-from mapswipe_workers.generate_stats.user_stats import (
-    get_agg_results_by_user_id,
-)
+from mapswipe_workers.generate_stats.user_stats import get_agg_results_by_user_id
 
 
 class TestUserStats(unittest.TestCase):