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)