def test_enable_postgis(self, mock_dump_cmd):
connector = PgDumpGisConnector()
connector.settings['ADMIN_USER'] = '******'
connector._enable_postgis()
self.assertIn('"CREATE EXTENSION IF NOT EXISTS postgis;"',
mock_dump_cmd.call_args[0][0])
self.assertIn('--username=foo', mock_dump_cmd.call_args[0][0])
def test_enable_postgis_port(self, mock_dump_cmd):
connector = PgDumpGisConnector()
connector.settings['ADMIN_USER'] = '******'
# Without
connector.settings.pop('PORT', None)
connector._enable_postgis()
self.assertNotIn(' --port=', mock_dump_cmd.call_args[0][0])
# With
connector.settings['PORT'] = 42
connector._enable_postgis()
self.assertIn(' --port=42', mock_dump_cmd.call_args[0][0])
def test_restore_dump(self, mock_dump_cmd, mock_restore_cmd):
connector = PgDumpGisConnector()
dump = connector.create_dump()
# Without ADMINUSER
connector.settings.pop('ADMIN_USER', None)
connector.restore_dump(dump)
self.assertTrue(mock_restore_cmd.called)
# With
connector.settings['ADMIN_USER'] = '******'
connector.restore_dump(dump)
self.assertTrue(mock_restore_cmd.called)
def setUp(self):
self.connector = PgDumpGisConnector()
self.connector.settings['HOST'] = 'hostname'
