class DataTableTest(unittest.TestCase):
def setUp(self):
self.addCleanup(self.my_cleanup, ('cleanup executado'))
self.table = DataTable('A')
def my_cleanup(self, msg):
print(msg)
def test_get_name(self):
self.assertEqual("A", self.table._get_name())
def test_set_name(self):
self.table._set_name("aurelio")
self.assertEqual("aurelio", self.table._get_name())
def test_delete_name(self):
self.assertRaises(AttributeError, self.table._del_name)
def test_add_column(self):
#table = DataTable("Empreendimento")
self.assertEqual(0, len(self.table._columns))
self.table.add_column('BId', 'bigint')
self.assertEqual(1, len(self.table._columns))
self.table.add_column('value', 'numeric')
self.assertEqual(2, len(self.table._columns))
self.table.add_column('desc', 'varchar')
self.assertEqual(3, len(self.table._columns))
def test_add_column_invalid_type(self):
a_table = DataTable('A')
self.assertRaises(Exception, a_table.add_column, ('col', 'invalid'))
def test_add_column_invalid_type_fail(self):
#a_table = DataTable("Empreendimento")
error = False
try:
self.table.add_column('col', 'invalid')
except:
error = True
if not error:
self.fail("Chamada não gerou erro, mas deveria")
