def testConnectToDevice(self): flexmock(self._dao).should_receive('getObjectById').and_return( Device("test", "qfx5100-48s-6q", "root", "abcd1234", "leaf", "", "192.168.48.182", None)) dataCollector = DeviceDataCollectorNetconf('1234', {}, InMemoryDao) dataCollector._dao = self._dao dataCollector.manualInit() dataCollector.connectToDevice()
def setUp(self): self._dao = InMemoryDao.getInstance() self.dataCollector = DeviceDataCollectorNetconf( '1234', {}, InMemoryDao)