diff --git a/test/test_database.py b/test/test_database.py index bf1d8a825fc4387b887c626f16944892a056dfdf..37239aad9b254ff14e02f6b9d1956c8789577a36 100644 --- a/test/test_database.py +++ b/test/test_database.py @@ -105,16 +105,25 @@ class DatabaseTest(unittest.TestCase): def test_get_codebook(self): """Assert function get_codebook works as expected.""" - + # Get test-data from json file_path = os.path.join(TESTDATA_PATH, "codebook.json") with open(file_path, "r") as file: - data = json.load(file) + expected = json.load(file) file.close() - stmt = 'SELECT ALL FROM CodeBook' - queried = self.database.conn.execute(stmt).fetchall() + # Get data from db + stmt = 'SELECT * FROM CodeBook' + actual = self.database.conn.execute(stmt).fetchall() + + # Check same lenght + self.assertEqual(len(expected), len(actual)) - + # Check total equality + for i in range(len(expected)): + self.assertEqual(expected[i]['word'], actual[i][0]) + self.assertEqual(expected[i]['category'], actual[i][1]) + self.assertEqual(expected[i]['type'], actual[i][2]) + self.assertEqual(expected[i]['code'], actual[i][3]) if __name__ == '__main__':