def test_fail(self):
# TODO: test logged message
context = api.Context("flow_id", "1.2.3.4", 5678)
error = RuntimeError("Expected failure")
with self.assertRaises(RuntimeError) as e:
run_with_vars(context, None, Logged().fail, error)
self.assertIs(e.exception, error)
def setUp(self):
self.uuid = API.VM.BLANK_UUID
self.cif = FakeClientIF()
self.vmParams = {
'vmId': self.uuid,
'vmName': 'TESTING',
'memSize': 8 * 1024,
'vmType': 'kvm',
'display': 'qxl',
'kvmEnable': 'true',
'smp': '1',
}
with MonkeyPatchScope([(API, 'clientIF', self.cif)]):
self.vm = API.VM(self.uuid)
# to make testing easier
threadlocal.vars.context = api.Context("flow_id", "1.2.3.4", 5678)
def handleMessage(self, data, flow_id):
if self._messageHandler is not None:
context = api.Context(flow_id, self._client_host,
self._client_port)
self._messageHandler(
(self._server, self.get_local_address(), context, data))
def test_success(self):
# TODO: test logged message
context = api.Context("flow_id", "1.2.3.4", 5678)
task = Task("task_id")
result = run_with_vars(context, task, Logged().succeed, "a", b=1)
self.assertEqual(result, (("a", ), {"b": 1}))