Skip to content

Commit a12a803

Browse files
committed
tests(street): add integration test for street project
1 parent c1f1a79 commit a12a803

File tree

1 file changed

+60
-0
lines changed

1 file changed

+60
-0
lines changed
Lines changed: 60 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,60 @@
1+
import unittest
2+
3+
from click.testing import CliRunner
4+
5+
from mapswipe_workers import auth, mapswipe_workers
6+
from mapswipe_workers.definitions import logger
7+
from mapswipe_workers.utils.create_directories import create_directories
8+
from tests.integration import set_up, tear_down
9+
10+
11+
class TestCreateFootprintProject(unittest.TestCase):
12+
def setUp(self):
13+
self.project_id = [
14+
set_up.create_test_project_draft("street", "street"),
15+
]
16+
create_directories()
17+
18+
def tearDown(self):
19+
for element in self.project_id:
20+
tear_down.delete_test_data(element)
21+
22+
def test_create_street_project(self):
23+
runner = CliRunner()
24+
result = runner.invoke(
25+
mapswipe_workers.run_create_projects, catch_exceptions=False
26+
)
27+
if result.exit_code != 0:
28+
raise result.exception
29+
pg_db = auth.postgresDB()
30+
for element in self.project_id:
31+
logger.info(f"Checking project {self.project_id}")
32+
query = "SELECT project_id FROM projects WHERE project_id = %s"
33+
result = pg_db.retr_query(query, [element])[0][0]
34+
self.assertEqual(result, element)
35+
36+
# check if usernames made it to postgres
37+
query = """
38+
SELECT count(*)
39+
FROM tasks
40+
WHERE project_id = %s
41+
"""
42+
result = pg_db.retr_query(query, [element])[0][0]
43+
self.assertGreater(result, 0)
44+
45+
fb_db = auth.firebaseDB()
46+
ref = fb_db.reference(f"/v2/projects/{element}")
47+
result = ref.get(shallow=True)
48+
self.assertIsNotNone(result)
49+
50+
ref = fb_db.reference(f"/v2/groups/{element}")
51+
result = ref.get(shallow=True)
52+
self.assertIsNotNone(result)
53+
54+
# Footprint projects have tasks in Firebase
55+
ref = fb_db.reference(f"/v2/tasks/{element}")
56+
result = ref.get(shallow=True)
57+
self.assertIsNotNone(result)
58+
59+
if __name__ == "__main__":
60+
unittest.main()

0 commit comments

Comments
 (0)