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)
