writing tests

Author: Matthias

tests/test_export.py

@@ -1,8 +1,15 @@
+import pickle
 from mapswipe_workers.basic import BaseFunctions
 
 
 def test_transfer_results():
-    BaseFunctions.run_export('production', [])
+
+    filename = 'firebase_imported_projects.pickle'
+    with open(filename, 'rb') as f:
+        imported_projects = pickle.load(f)
+
+    project_ids = [i[1] for i in imported_projects]
+    BaseFunctions.run_export('production', project_ids)
 
 
 if __name__ == '__main__':

tests/test_update.py

@@ -1,9 +1,16 @@
+import pickle
 from mapswipe_workers.basic import BaseFunctions
 
 
-def test_transfer_results():
-    BaseFunctions.run_update('production', [])
+def test_update():
+
+    filename = 'firebase_imported_projects.pickle'
+    with open(filename, 'rb') as f:
+        imported_projects = pickle.load(f)
+
+    project_ids = [i[1] for i in imported_projects]
+    BaseFunctions.run_update('production', project_ids)
 
 
 if __name__ == '__main__':
-    test_transfer_results()
+    test_update()