def testDbElementsAreEqual(self): dbDriver1 = DBDriverMock() dbDriver2 = DBDriverMock() sut = Pydblibrary() sut.connect_to_database(dbDriver1, 'someDbName1', 'someUsername', 'somePassword', 'someHost', '7777', 'db1') sut.connect_to_database(dbDriver2, 'someDbName2', 'someUsername', 'somePassword', 'someHost', '7777', 'db2') responses = {'select * from employee': [(0, 'John', 'Doe'), (1, 'Jane', 'Doe')]} dbDriver1.connection.cursor().setQueryResponses(responses) dbDriver2.connection.cursor().setQueryResponses(responses) sut.db_elements_are_equal('select * from employee', 'db1', 'db2')
def testDbElementsAreEqual(self): dbDriver1 = DBDriverMock() dbDriver2 = DBDriverMock() sut = Pydblibrary() sut.connect_to_database(dbDriver1, 'someDbName1', 'someUsername', 'somePassword', 'someHost', '7777', 'db1') sut.connect_to_database(dbDriver2, 'someDbName2', 'someUsername', 'somePassword', 'someHost', '7777', 'db2') responses = { 'select * from employee': [(0, 'John', 'Doe'), (1, 'Jane', 'Doe')] } dbDriver1.connection.cursor().setQueryResponses(responses) dbDriver2.connection.cursor().setQueryResponses(responses) sut.db_elements_are_equal('select * from employee', 'db1', 'db2')