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)
