def test_publish_add_column(self): """Publish should add a geometry column to tables that don't have one""" # Create a table self.test_table.create(self.engine, checkfirst=True) # Spatialize it ds = Datastored("test_already", "lat", "lng") ds.publish() # Check that the geometry column exists sql = "select count(column_name) from information_schema.columns where table_name='test_already' and column_name='geometry'" conn = self.engine.connect() res = conn.execute(sql) self.assertEqual(res.rowcount, 1) for row in res: self.assertEqual(row["count"], 1) conn.close()
def test_publish_already_published(self): """If a resource has already been spatialized, should not fail""" # Create a table self.test_table.create(self.engine, checkfirst=True) # Give it a geometry column sql = "select AddGeometryColumn('public', 'test_already', 'geometry', 4326, 'GEOMETRY', 2)" conn = self.engine.connect() try: t = conn.begin() conn.execute(sql) t.commit() except Exception as ex: t.rollback() conn.close() # Now try and "spatialize" it ds = Datastored("test_already", "lat", "lng") self.assertTrue(ds.publish())