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}))