def test_given_valid_attr_then_return(self): param_dict = dict(one=7, two=23) post = Request.Post(self.context, self.response_queue, [""], parameters=param_dict) self.assertEqual(param_dict, post.parameters)
def test_Post(self, request_mock): endpoint = ["BL18I:XSPRESS3", "configure"] post = Request.Post(self.context, self.response_queue, endpoint) request_mock.assert_called_once_with(self.context, self.response_queue, type_="Post")
def test_given_invalid_attr_then_raise_error(self): param_dict = dict(one=7, two=23) post = Request.Post(self.context, self.response_queue, [""], parameters=param_dict) with self.assertRaises(KeyError): post.null
def call_server_method(self, method_name, args): """Call method_name on the server Args: method_name (str): Name of the method args (dict): Map of arguments to be called with """ request = Request.Post(None, self.q, [self.block.name, method_name], args) self.client_comms.q.put(request) response = self.q.get() assert response.type_ == response.RETURN, \ "Expected Return, got %s" % response.type_ return response.value
def test_hello_controller_with_process(self): sync_factory = SyncFactory("sched") process = Process("proc", sync_factory) block = Block("hello") HelloController(block) process.add_block(block) process.start() q = sync_factory.create_queue() req = Request.Post(response_queue=q, context="ClientConnection", endpoint=["hello", "say_hello"], parameters=dict(name="thing")) req.set_id(44) process.q.put(req) resp = q.get(timeout=1) self.assertEqual(resp.id_, 44) self.assertEqual(resp.context, "ClientConnection") self.assertEqual(resp.type_, "Return") self.assertEqual(resp.value, dict(greeting="Hello thing"))