def test_select_returns_expected_data(self): SQLite3Mock.fetchall_returns = [ sqlite3.Error("Error"), [], [("a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l")], [("us", "united states")], ] ldb = LocationDatabase() # Country Result result = ldb.single_select("sql1", ("tuple1",), CountryEntity) self.assertIsInstance(result, list) self.assertEqual(1, len(result)) self.assertIsInstance(result[0], CountryEntity) self.assertEqual(result[0].abbreviation, "us") self.assertEqual(result[0].name, "united states") # City Result result = ldb.single_select("sql2", ("tuple2",), CityEntity) self.assertIsInstance(result, list) self.assertEqual(1, len(result)) self.assertIsInstance(result[0], CityEntity) self.assertEqual(result[0].country, "a") self.assertEqual(result[0].postal_code, "b") self.assertEqual(result[0].city, "c") self.assertEqual(result[0].state1, "d") self.assertEqual(result[0].state2, "e") self.assertEqual(result[0].province1, "f") self.assertEqual(result[0].province2, "g") self.assertEqual(result[0].community1, "h") self.assertEqual(result[0].community2, "i") self.assertEqual(result[0].latitude, "j") self.assertEqual(result[0].longitude, "k") self.assertEqual(result[0].coord_accuracy, "l") # No results result = ldb.single_select("sql3", ("tuple3",), CityEntity) self.assertIsNone(result) # Error while fetching result = ldb.single_select("sql4", ("tuple4",), CountryEntity) self.assertIsNone(result) self.assertEqual( SQLite3Mock.execute_calls, [ ("PRAGMA temp_store = 2", None), ("sql1", ("tuple1",)), ("PRAGMA temp_store = 2", None), ("sql2", ("tuple2",)), ("PRAGMA temp_store = 2", None), ("sql3", ("tuple3",)), ("PRAGMA temp_store = 2", None), ("sql4", ("tuple4",)), ], )
def test_select_returns_expected_data(self): SQLite3Mock.fetchall_returns = [ sqlite3.Error('Error'), [], [('a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l')], [('us', 'united states')] ] ldb = LocationDatabase() # Country Result result = ldb.single_select('sql1', ('tuple1', ), CountryEntity) self.assertIsInstance(result, list) self.assertEqual(1, len(result)) self.assertIsInstance(result[0], CountryEntity) self.assertEqual(result[0].abbreviation, 'us') self.assertEqual(result[0].name, 'united states') # City Result result = ldb.single_select('sql2', ('tuple2', ), CityEntity) self.assertIsInstance(result, list) self.assertEqual(1, len(result)) self.assertIsInstance(result[0], CityEntity) self.assertEqual(result[0].country, 'a') self.assertEqual(result[0].postal_code, 'b') self.assertEqual(result[0].city, 'c') self.assertEqual(result[0].state1, 'd') self.assertEqual(result[0].state2, 'e') self.assertEqual(result[0].province1, 'f') self.assertEqual(result[0].province2, 'g') self.assertEqual(result[0].community1, 'h') self.assertEqual(result[0].community2, 'i') self.assertEqual(result[0].latitude, 'j') self.assertEqual(result[0].longitude, 'k') self.assertEqual(result[0].coord_accuracy, 'l') # No results result = ldb.single_select('sql3', ('tuple3', ), CityEntity) self.assertIsNone(result) # Error while fetching result = ldb.single_select('sql4', ('tuple4', ), CountryEntity) self.assertIsNone(result) self.assertEqual(SQLite3Mock.execute_calls, [ ('PRAGMA temp_store = 2', None), ('sql1', ('tuple1', )), ('PRAGMA temp_store = 2', None), ('sql2', ('tuple2', )), ('PRAGMA temp_store = 2', None), ('sql3', ('tuple3', )), ('PRAGMA temp_store = 2', None), ('sql4', ('tuple4', )), ])
def test_select(self): ldb = LocationDatabase() with self.assertRaises(TypeError): ldb.single_select("sql", "burp", CityEntity)
def test_select(self): ldb = LocationDatabase() with self.assertRaises(TypeError): ldb.single_select('sql', 'burp', CityEntity)