def setUp(self): self.fakeGuestAgent = guestagent.GuestAgent(None, None, self.log, lambda: None, lambda: None) self.fakeGuestAgent.MAX_MESSAGE_SIZE = 100 self.maxMessageSize = self.fakeGuestAgent.MAX_MESSAGE_SIZE self.fakeGuestAgent._clearReadBuffer() # Guest agent must not be stopped self.fakeGuestAgent._stopped = False # Copy the defaults of the guest agent -> Not set information self.infoDefaults = self.fakeGuestAgent.guestInfo.copy()
def test_handleMessage(self): fakeGuestAgent = guestagent.GuestAgent(None, None, self.log, lambda: None, lambda: None) testCase = namedtuple('testCase', 'msgType, message, assertDict') for t in zip(_MSG_TYPES, _INPUTS, _OUTPUTS): t = testCase(*t) fakeGuestAgent._handleMessage(t.msgType, t.message) for (k, v) in six.iteritems(t.assertDict): self.assertEqual(fakeGuestAgent.guestInfo[k], v)
def test_guestinfo_encapsulation(self): fake_guest_agent = guestagent.GuestAgent(None, None, self.log, lambda: None, lambda: None) fake_guest_agent._handleMessage(_MSG_TYPES[0], _INPUTS[0]) with MonkeyPatchScope([(fake_guest_agent, 'isResponsive', lambda: True) ]): guest_info = fake_guest_agent.getGuestInfo() for k in _OUTPUTS[0]: value = guest_info[k] if isinstance(value, list): value.append('modified') elif isinstance(value, dict): value['modified'] = 'modified' else: value = 'modified' guest_info[k] = value guest_info = fake_guest_agent.getGuestInfo() for (k, v) in six.iteritems(_OUTPUTS[0]): assert guest_info[k] == v
def setUp(self): self.agent = guestagent.GuestAgent(None, None, None, lambda: None, lambda: None)