Basic framework for testing flask server with one passing unit test. …

…It was honestly hard to find something we could pass :P

microsetta_private_api/api/tests/test_integration.py

@@ -0,0 +1,110 @@
+import pytest
+import microsetta_private_api.server
+from microsetta_private_api.repo.transaction import Transaction
+from microsetta_private_api.repo.account_repo import AccountRepo
+from microsetta_private_api.model.account import Account
+from microsetta_private_api.repo.source_repo import SourceRepo
+from microsetta_private_api.model.source import \
+    Source, HumanInfo, AnimalInfo, EnvironmentInfo
+import datetime
+import json
+
+ACCT_ID = "aaaaaaaa-bbbb-cccc-dddd-eeeeffffffff"
+HUMAN_ID = "b0b0b0b0-b0b0-b0b0-b0b0-b0b0b0b0b0b0"
+DOGGY_ID = "dddddddd-dddd-dddd-dddd-dddddddddddd"
+PLANTY_ID = "eeeeeeee-eeee-eeee-eeee-eeeeeeeeeeee"
+
+
+def json_converter(o):
+    if isinstance(o, datetime.datetime):
+        return str(o)
+    return o.__dict__
+
+
+def setup_test_data():
+    with Transaction() as t:
+        acct_repo = AccountRepo(t)
+        acct_repo.delete_account(ACCT_ID)
+        acc = Account(ACCT_ID,
+                      "foo@baz.com",
+                      "globus",
+                      "Dan",
+                      "H",
+                      '{"a":5, "b":7}',
+                      "USER")
+        acct_repo.create_account(acc)
+
+        source_repo = SourceRepo(t)
+        source_repo.delete_source(ACCT_ID, DOGGY_ID)
+        source_repo.delete_source(ACCT_ID, PLANTY_ID)
+        source_repo.delete_source(ACCT_ID, HUMAN_ID)
+
+        source_repo.create_source(Source.create_human(
+            HUMAN_ID,
+            ACCT_ID,
+            HumanInfo("Bo", "bo@bo.com", False, "Mr Bo", False, "Mrs Bo",
+                      False, datetime.datetime.utcnow(), "TODO,WutDis?")
+        ))
+        source_repo.create_source(Source.create_animal(
+            DOGGY_ID,
+            ACCT_ID,
+            AnimalInfo("Doggy")))
+        source_repo.create_source(Source.create_environment(
+            PLANTY_ID,
+            ACCT_ID,
+            EnvironmentInfo("Planty", "The green one")))
+
+        t.commit()
+
+
+def teardown_test_data():
+    with Transaction() as t:
+        acct_repo = AccountRepo(t)
+        source_repo = SourceRepo(t)
+        source_repo.delete_source(ACCT_ID, DOGGY_ID)
+        source_repo.delete_source(ACCT_ID, PLANTY_ID)
+        source_repo.delete_source(ACCT_ID, HUMAN_ID)
+        acct_repo.delete_account(ACCT_ID)
+
+        t.commit()
+
+
+@pytest.fixture
+def client():
+    setup_test_data()
+    app = microsetta_private_api.server.build_app()
+    with app.app.test_client() as client:
+        yield client
+    teardown_test_data()
+
+
+def test_get_sources(client):
+    resp = client.get('/api/accounts/%s/sources' % ACCT_ID).data
+    sources = json.loads(resp)
+    assert len([x for x in sources if x['source_name'] == 'Bo']) == 1
+    assert len([x for x in sources if x['source_name'] == 'Doggy']) == 1
+    assert len([x for x in sources if x['source_name'] == 'Planty']) == 1
+
+
+# BROKEN
+# def test_external_surveys(client):
+#     # I can't even find a test that we can pass for demo purposes!
+#     resp = client.get('/api/accounts/%s/sources' % ACCT_ID).data
+#     sources = json.loads(resp)
+#     bobo = [x for x in sources if x['source_name'] == 'Bo'][0]
+#     doggy = [x for x in sources if x['source_name'] == 'Doggy'][0]
+#     env = [x for x in sources if x['source_name'] == 'Planty'][0]
+#
+#     resp = client.get('/accounts/%s/sources/%s/survey_templates' %
+#                       (ACCT_ID, bobo.id))
+#     bobo_surveys = json.loads(resp.data)
+#     resp = client.get('/accounts/%s/sources/%s/survey_templates' %
+#                       (ACCT_ID, doggy.id))
+#     doggy_surveys = json.loads(resp.data)
+#     resp = client.get('/accounts/%s/sources/%s/survey_templates' %
+#                       (ACCT_ID, env.id))
+#     env_surveys = json.loads(resp.data)
+#
+#     assert bobo_surveys == [1, 3, 4, 5]
+#     assert doggy_surveys == [2]
+#     assert env_surveys == [3]

microsetta_private_api/server.py

@@ -10,8 +10,8 @@
 import connexion
 from microsetta_private_api.util.util import JsonifyDefaultEncoder
 
-# If we're running in stand alone mode, run the application
-if __name__ == '__main__':
+
+def build_app():
     # Create the application instance
     app = connexion.FlaskApp(__name__)
 
@@ -22,5 +22,10 @@
     # Note: app.app is the actual Flask application instance, so any Flask
     # settings have to be set there.
     app.app.json_encoder = JsonifyDefaultEncoder
+    return app
 
+
+# If we're running in stand alone mode, run the application
+if __name__ == '__main__':
+    app = build_app()
     app.run(port=8082, debug=True)

repo_test_scratch.py

@@ -145,4 +145,3 @@ def json_converter(o):
     print(survey_model == survey_model2)
 
     survey_answers_repo.delete_answered_survey(ACCT_ID, answer_id)
-