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