class TestPost(unittest.TestCase): def setUp(self): self.context = MagicMock() self.response_queue = MagicMock() self.endpoint = ["BL18I:XSPRESS3", "state", "value"] self.parameters = dict(arg1=5, arg2=True) self.post = Post(self.context, self.response_queue, self.endpoint, self.parameters) def test_init(self): self.assertEqual(self.context, self.post.context) self.assertEqual(self.response_queue, self.post.response_queue) self.assertEqual(self.endpoint, self.post.endpoint) self.assertEqual(self.parameters, self.post.parameters) self.assertEqual("malcolm:core/Post:1.0", self.post.typeid) def test_setters(self): self.post.set_endpoint(["BL18I:XSPRESS3", "state", "value2"]) self.assertEquals(["BL18I:XSPRESS3", "state", "value2"], self.post.endpoint) self.post.set_parameters(dict(arg1=2, arg2=False)) self.assertEquals(dict(arg1=2, arg2=False), self.post.parameters)