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)
