def testSetCurrentDatabase(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') self.assertTrue(id(sut._connectionCache.current.connection) == id(dbDriver2.connection), 'Current database is db1 but should be db2.') sut.set_current_database('db1') self.assertTrue(id(sut._connectionCache.current.connection) == id(dbDriver1.connection), 'Current database is db2 but should be db1.')
def testSetCurrentDatabase(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') self.assertTrue( id(sut._connectionCache.current.connection) == id( dbDriver2.connection), 'Current database is db1 but should be db2.') sut.set_current_database('db1') self.assertTrue( id(sut._connectionCache.current.connection) == id( dbDriver1.connection), 'Current database is db2 but should be db1.')