class TestGetAllTasks(unittest.TestCase): def setUp(self): self.sql_db = Database(TEST_DB_PATH) self.sql_db.init_sqlite_db() for i in range(0, 3): self.sql_db.add_task() i += 1 def test_get_all_tasks(self): expected_response = [{"task_id": i, "task_status": "Pending", "report_id": None} for i in range(1, 4)] resp = self.sql_db.get_all_tasks() for i in range(0, 3): self.assertDictEqual(expected_response[i], resp[i]) def tearDown(self): os.remove(TEST_DB_PATH)
class TestGetAllTasks(unittest.TestCase): def setUp(self): self.sql_db = Database(TEST_DB_PATH) self.sql_db.init_sqlite_db() for i in range(0, 3): self.sql_db.add_task() i += 1 def test_get_all_tasks(self): expected_response = [{ 'task_id': i, 'task_status': 'Pending', 'report_id': None } for i in range(1, 4)] resp = self.sql_db.get_all_tasks() for i in range(0, 3): self.assertDictEqual(expected_response[i], resp[i]) def tearDown(self): os.remove(TEST_DB_PATH)
class TestTaskManipulation(unittest.TestCase): def setUp(self): self.sql_db = Database(TEST_DB_PATH) self.sql_db.init_sqlite_db() self.sql_db.add_task(task_id=1, task_status='Pending', report_id=None) def test_add_second_task(self): resp = self.sql_db.add_task() self.assertEqual(resp, 2) def test_get_task(self): resp = self.sql_db.get_task(task_id=1) self.assertEqual(resp.task_id, 1) self.assertEqual(resp.task_status, 'Pending') self.assertEqual(resp.report_id, None) def test_update_task(self): resp = self.sql_db.update_task( task_id=1, task_status='Complete', report_id= '88d11f0ea5cc77a59b6e47deee859440f26d2d14440beb712dbac8550d35ef1f') self.assertDictEqual(resp, self.sql_db.get_task(1).to_dict()) self.assertDictEqual( resp, { 'task_id': 1, 'task_status': 'Complete', 'report_id': '88d11f0ea5cc77a59b6e47deee859440f26d2d14440beb712dbac8550d35ef1f' }) def test_delete_task(self): deleted = self.sql_db.delete_task(task_id=1) self.assertTrue(deleted) resp = self.sql_db.get_task(task_id=1) self.assertEqual(resp, None) def tearDown(self): os.remove(TEST_DB_PATH)
class TestTaskAdd(unittest.TestCase): def setUp(self): self.sql_db = Database(TEST_DB_PATH) self.sql_db.init_sqlite_db() def test_add_task(self): task_id = 1 resp = self.sql_db.add_task(task_id=task_id, task_status="Pending", report_id=None) self.assertEqual(resp, task_id) def tearDown(self): os.remove(TEST_DB_PATH)
class TestTaskManipulation(unittest.TestCase): def setUp(self): self.sql_db = Database(TEST_DB_PATH) self.sql_db.init_sqlite_db() self.sql_db.add_task(task_id=1, task_status="Pending", report_id=None) def test_add_second_task(self): resp = self.sql_db.add_task() self.assertEqual(resp, 2) def test_get_task(self): resp = self.sql_db.get_task(task_id=1) self.assertEqual(resp.task_id, 1) self.assertEqual(resp.task_status, "Pending") self.assertEqual(resp.report_id, None) def test_update_task(self): resp = self.sql_db.update_task( task_id=1, task_status="Complete", report_id="88d11f0ea5cc77a59b6e47deee859440f26d2d14440beb712dbac8550d35ef1f", ) self.assertDictEqual(resp, self.sql_db.get_task(1).to_dict()) self.assertDictEqual( resp, { "task_id": 1, "task_status": "Complete", "report_id": "88d11f0ea5cc77a59b6e47deee859440f26d2d14440beb712dbac8550d35ef1f", }, ) def test_delete_task(self): deleted = self.sql_db.delete_task(task_id=1) self.assertTrue(deleted) resp = self.sql_db.get_task(task_id=1) self.assertEqual(resp, None) def tearDown(self): os.remove(TEST_DB_PATH)
class TestTaskAdd(unittest.TestCase): def setUp(self): self.sql_db = Database(TEST_DB_PATH) self.sql_db.init_sqlite_db() def test_add_task(self): task_id = 1 resp = self.sql_db.add_task(task_id=task_id, task_status='Pending', report_id=None) self.assertEqual(resp, task_id) def tearDown(self): os.remove(TEST_DB_PATH)