def test_fetchone(self):
"""Test for fetching one record."""
spatialite_path = sg_diagrams_database
db_manager = DatabaseManager(spatialite_path)
query = "SELECT province FROM provinces WHERE "
query += "Within(GeomFromText('POINT(25 -30)'), Geometry)"
result = db_manager.fetch_one(query)
expected_result = ('Free State',)
message = 'Expected %s, got %s' % (expected_result, result)
self.assertEqual(result, expected_result, message)
query = "SELECT province FROM provinces WHERE "
query += "Within(GeomFromText('POINT(100 100)'), Geometry)"
result = db_manager.fetch_one(query)
expected_result = None
message = 'Expected %s, got %s' % (expected_result, result)
self.assertEqual(result, expected_result, message)
db_manager.close()
def test_connection(self):
spatialite_path = sg_diagrams_database
db_manager = DatabaseManager(spatialite_path)
query = 'SELECT count(*) FROM provinces'
result = db_manager.fetch_one(query)
expected_result = (7,)
message = 'Expected %s, got %s' % (expected_result, result)
self.assertEqual(result, expected_result, message)
db_manager.close()
spatialite_path = sg_diagrams_database + 'zero'
db_manager = DatabaseManager(spatialite_path)
query = 'SELECT count(*) FROM provinces'
self.assertRaises(DatabaseException, db_manager.fetch_one, query)
db_manager.close()
if os.path.exists(spatialite_path):
os.remove(spatialite_path)
