def setUp(self):
super(self.__class__, self).setUp()
self.writer = PostgresDbWriter(self.config.options['destination']['postgres'], True)
class TestPostgresDbWriter(WithOutput):
def setUp(self):
super(self.__class__, self).setUp()
self.writer = PostgresDbWriter(self.config.options['destination']['postgres'], True)
def tearDown(self):
super(self.__class__, self).tearDown()
self.writer.close()
def test_truncate(self):
self.writer.truncate(self.table1)
def test_write_table_indexes_and_constraints(self):
self.writer.write_table(table=self.table1)
self.writer.write_indexes(self.table1)
self.writer.write_constraints(self.table2)
def test_write_contents(self):
self.writer.write_contents(self.table1, self.reader)