def testTableMustContainMoreThanNumberOfRows(self): dbDriver = DBDriverMock() sut = Pydblibrary() sut.connect_to_database(dbDriver, 'someDbName', 'someUsername', 'somePassword', 'someHost', '7777') responses = {"select * from employee": [(0, 'John', 'Doe'), (1, 'Jane', 'Doe')]} dbDriver.connection.cursor().setQueryResponses(responses) sut.table_must_contain_more_than_number_of_rows('employee', 1)
def testTableMustContainMoreThanNumberOfRows(self): dbDriver = DBDriverMock() sut = Pydblibrary() sut.connect_to_database(dbDriver, 'someDbName', 'someUsername', 'somePassword', 'someHost', '7777') responses = { "select * from employee": [(0, 'John', 'Doe'), (1, 'Jane', 'Doe')] } dbDriver.connection.cursor().setQueryResponses(responses) sut.table_must_contain_more_than_number_of_rows('employee', 1)