def testCloseConnection(self): dbDriver = DBDriverMock() sut = Pydblibrary() sut.connect_to_database(dbDriver, 'someDbName', 'someUsername', 'somePassword', 'someHost', '7777') sut.disconnect_from_database() self.assertFalse(dbDriver.connection.isConnected, 'Connected to database but should be disconnected.')