Added unit tests

Author: tsv2013

.vscode/settings.json

@@ -0,0 +1,11 @@
+{
+    "python.testing.unittestArgs": [
+        "-v",
+        "-s",
+        "./tests",
+        "-p",
+        "*_test.py"
+    ],
+    "python.testing.pytestEnabled": false,
+    "python.testing.unittestEnabled": true
+}

tests/app_test.py

@@ -0,0 +1,138 @@
+import unittest
+import json
+from app import app
+
+class TestApp(unittest.TestCase):
+    def setUp(self):
+        app.testing = True
+        self.client = app.test_client()
+        self.session = {}
+        
+        app.config['SESSION_TYPE'] = 'null'
+        app.config['TESTING'] = True
+
+    def test_get_active_surveys(self):
+        with self.client:
+            response = self.client.get('/api/getActive')
+            data = json.loads(response.data)
+            
+            self.assertEqual(response.status_code, 200)
+            self.assertGreater(len(data), 0)
+            self.assertIn('id', data[0])
+            self.assertIn('name', data[0])
+
+    def test_get_survey_exists(self):
+        with self.client:
+            get_active = self.client.get('/api/getActive')
+            first_id = json.loads(get_active.data)[0]['id']
+            
+            response = self.client.get(f'/api/getSurvey?surveyId={first_id}')
+            data = json.loads(response.data)
+            
+            self.assertEqual(response.status_code, 200)
+            self.assertEqual(data['id'], first_id)
+
+    def test_get_survey_not_exists(self):
+        with self.client:
+            response = self.client.get('/api/getSurvey?surveyId=9999')
+            self.assertEqual(response.status_code, 200)
+            self.assertIsNone(json.loads(response.data))
+
+    def test_change_name(self):
+        with self.client:
+            create_resp = self.client.get('/api/create?name=Test%20Survey')
+            survey = json.loads(create_resp.data)
+            
+            new_name = "New Name"
+            response = self.client.get(
+                f'/api/changeName?id={survey["id"]}&name={new_name}'
+            )
+            updated = json.loads(response.data)
+            
+            self.assertEqual(updated['name'], new_name)
+            self.assertEqual(updated['id'], survey['id'])
+
+    def test_create_survey(self):
+        with self.client:
+            response = self.client.get('/api/create?name=Test%20Create')
+            data = json.loads(response.data)
+            
+            self.assertEqual(response.status_code, 200)
+            self.assertEqual(data['name'], 'Test Create')
+            self.assertIsNotNone(data['id'])
+
+    def test_change_json(self):
+        with self.client:
+            create_resp = self.client.get('/api/create')
+            survey = json.loads(create_resp.data)
+            
+            test_json = {'question': 'Test?'}
+            response = self.client.post(
+                '/api/changeJson',
+                json={'id': survey['id'], 'json': test_json}
+            )
+            updated = json.loads(response.data)
+            
+            self.assertEqual(updated['json'], test_json)
+
+    def test_post_results(self):
+        with self.client:
+            post_id = '123'
+            test_data = {'answer': 42}
+            
+            response = self.client.post(
+                '/api/post',
+                json={'postId': post_id, 'surveyResult': test_data}
+            )
+            
+            self.assertEqual(response.status_code, 200)
+            self.assertEqual(json.loads(response.data), {})
+
+    def test_delete_survey(self):
+        with self.client:
+            create_resp = self.client.get('/api/create')
+            survey = json.loads(create_resp.data)
+            
+            response = self.client.get(f'/api/delete?id={survey["id"]}')
+            data = json.loads(response.data)
+            
+            self.assertEqual(data['id'], survey['id'])
+
+    def test_get_results(self):
+        with self.client:
+            post_id = 'results_test'
+            test_data = {'data': 'test'}
+            
+            self.client.post(
+                '/api/post',
+                json={'postId': post_id, 'surveyResult': test_data}
+            )
+            
+            response = self.client.get(f'/api/results?postId={post_id}')
+            data = json.loads(response.data)
+            
+            self.assertEqual(len(data['data']), 1)
+            self.assertEqual(data['data'][0], test_data)
+
+    @unittest.skip("it doesn't work properly yet")
+    def test_static_routes(self):
+        routes = [
+            '/',
+            '/about',
+            '/run/test',
+            '/edit/123',
+            '/results/456'
+        ]
+        
+        for route in routes:
+            with self.subTest(route=route):
+                response = self.client.get(route)
+                self.assertEqual(response.status_code, 200)
+                self.assertIn(b'index.html', response.data)
+
+    def test_invalid_route(self):
+        response = self.client.get('/invalid-route')
+        self.assertEqual(response.status_code, 404)
+
+if __name__ == '__main__':
+    unittest.main()

tests/inmemorydbadapter_test.py

@@ -0,0 +1,116 @@
+import unittest
+from inmemorydbadapter import InMemoryDBAdapter
+from demo_surveys import demo_data
+
+class TestInMemoryDBAdapter(unittest.TestCase):
+    def setUp(self):
+        self.session = {}
+        self.adapter = InMemoryDBAdapter(self.session)
+    
+    def test_initialization(self):
+        self.assertEqual(self.adapter.current_id, len(demo_data["surveys"]) + 1)
+    
+    def test_get_table_existing(self):
+        self.session["surveys"] = [{"test": "data"}]
+        table = self.adapter.get_table("surveys")
+        self.assertEqual(table, [{"test": "data"}])
+    
+    def test_get_table_new(self):
+        table = self.adapter.get_table("new_table")
+        self.assertEqual(table, [])
+        self.assertIn("new_table", self.session)
+    
+    def test_find_by_id_found(self):
+        objects = [{"id": "1"}, {"id": "2"}]
+        result = self.adapter.find_by_id(objects, "1")
+        self.assertEqual(result, {"id": "1"})
+    
+    def test_find_by_id_not_found(self):
+        objects = [{"id": "1"}]
+        result = self.adapter.find_by_id(objects, "2")
+        self.assertIsNone(result)
+    
+    def test_add_survey_with_name(self):
+        new_survey = self.adapter.add_survey("Test Survey")
+        self.assertEqual(new_survey["name"], "Test Survey")
+        self.assertEqual(new_survey["id"], str(len(demo_data["surveys"]) + 1))
+        self.assertEqual(len(self.adapter.get_table("surveys")), 1)
+    
+    def test_add_survey_without_name(self):
+        new_survey = self.adapter.add_survey(None)
+        expected_name = f"{demo_data['default_name']} {self.adapter.current_id - 1}"
+        self.assertEqual(new_survey["name"], expected_name)
+    
+    def test_post_results_new_post(self):
+        post_id = "123"
+        test_data = {"answer": 42}
+        self.adapter.post_results(post_id, test_data)
+        results = self.adapter.get_results(post_id)
+        self.assertEqual(len(results["data"]), 1)
+        self.assertEqual(results["data"][0], test_data)
+    
+    def test_post_results_existing_post(self):
+        post_id = "123"
+        self.adapter.post_results(post_id, {"data": 1})
+        self.adapter.post_results(post_id, {"data": 2})
+        results = self.adapter.get_results(post_id)
+        self.assertEqual(len(results["data"]), 2)
+    
+    def test_get_results_not_exists(self):
+        results = self.adapter.get_results("non_existent")
+        self.assertIsNone(results)
+    
+    def test_delete_survey_exists(self):
+        survey = {"id": "1", "name": "Test"}
+        self.adapter.get_table("surveys").append(survey)
+        result = self.adapter.delete_survey("1")
+        self.assertEqual(result, survey)
+        self.assertEqual(len(self.adapter.get_table("surveys")), 0)
+    
+    def test_delete_survey_not_exists(self):
+        result = self.adapter.delete_survey("999")
+        self.assertIsNone(result)
+    
+    def test_store_survey_update(self):
+        survey = {"id": "1", "name": "Old", "json": "{}"}
+        self.adapter.get_table("surveys").append(survey)
+        updated = self.adapter.store_survey("1", "New", '{"new": "data"}')
+        self.assertEqual(updated["json"], '{"new": "data"}')
+    
+    def test_store_survey_create(self):
+        survey = self.adapter.store_survey("100", "New Survey", "{}")
+        self.assertEqual(survey["name"], "New Survey")
+        self.assertIn(survey, self.adapter.get_table("surveys"))
+    
+    def test_change_name_exists(self):
+        survey = {"id": "1", "name": "Old"}
+        self.adapter.get_table("surveys").append(survey)
+        self.adapter.change_name("1", "New")
+        self.assertEqual(survey["name"], "New")
+    
+    def test_change_name_not_exists(self):
+        result = self.adapter.change_name("999", "New")
+        self.assertIsNone(result)
+    
+    def test_get_surveys_initial_load(self):
+        surveys = self.adapter.get_surveys()
+        self.assertEqual(len(surveys), len(demo_data["surveys"]))
+        self.assertEqual(len(self.session["results"]), len(demo_data["results"]))
+    
+    def test_get_surveys_subsequent_calls(self):
+        self.adapter.get_surveys()
+        new_survey = self.adapter.add_survey("Test")
+        surveys = self.adapter.get_surveys()
+        self.assertIn(new_survey, surveys)
+    
+    def test_get_survey_exists(self):
+        self.adapter.get_surveys()
+        survey = self.adapter.get_survey("1")
+        self.assertEqual(survey["id"], "1")
+    
+    def test_get_survey_not_exists(self):
+        survey = self.adapter.get_survey("999")
+        self.assertIsNone(survey)
+
+if __name__ == '__main__':
+    unittest.main()