class DBHandlerTester(unittest.TestCase):
def setUp(self):
self.handler = DBHandler()
def tearDown(self):
self.handler.closeCursor()
def testCreateNewTableAndInsertRow(self):
self.handler.createNewTable(['c1', 'c2', 'c3', 'c4'])
row1 = ['r1c1', 'r1c2', 'r1c3', 'r1c4']
self.handler.insertRow(row1)
row2 = ['r2c1', 'r2c2', 'r2c3', 'r3c4']
self.handler.insertRow(row2)
results = self.handler.executeQuery('select * from xmldata')
# each returned row is a tuple, so has to be converted to list for comparison
assert list(results[0]) == row1 and list(results[1]) == row2, results
def testGetColumnNames(self):
columns = ['c1', 'c2', 'c3', 'c4']
self.handler.createNewTable(columns)
results = self.handler.getColumnNames()
assert results == columns, results
