def setUp(self): super(self.__class__, self).setUp() self.outfile = tempfile.NamedTemporaryFile() self.writer = PostgresFileWriter(self.outfile)
class TestPostgresFileWriter(WithOutput): def setUp(self): super(self.__class__, self).setUp() self.outfile = tempfile.NamedTemporaryFile() self.writer = PostgresFileWriter(self.outfile) def tearDown(self): super(self.__class__, self).tearDown() self.writer.close() def test_truncate(self): self.writer.truncate(self.table1) def test_write_table(self): self.writer.write_table(self.table1) def test_write_indexes(self): self.writer.write_indexes(self.table1) def test_write_constraints(self): self.writer.write_constraints(self.table2) def test_write_contents(self): self.writer.write_contents(self.table1, self.reader)