def test__build_header(self, mockr, mockp):
mockr.return_value = {'one':1, 'two':2}
mockp.return_value = {'two':-2, 'three':3}
ep = ProcessRPCResponseEndpointUnit()
header = ep._build_header(sentinel.raw_msg)
self.assertEquals(header, {'one':1, 'two':-2, 'three':3})
def test__build_header(self, mockr, mockp):
mockr.return_value = {"one": 1, "two": 2}
mockp.return_value = {"two": -2, "three": 3}
ep = ProcessRPCResponseEndpointUnit()
header = ep._build_header(sentinel.raw_msg)
self.assertEquals(header, {"one": 1, "two": -2, "three": 3})
def test__message_received(self, mockmr):
procmock = Mock()
procmock.push_context = MagicMock()
ep = ProcessRPCResponseEndpointUnit(process=procmock)
ep._message_received(sentinel.msg, sentinel.headers)
procmock.push_context.assert_called_once_with(sentinel.headers)
procmock.push_context().__enter__.assert_called_once_with()
mockmr.assert_called_once_with(ep, sentinel.msg, sentinel.headers)
def test__message_received(self, mockmr):
procmock = Mock()
procmock.push_context = MagicMock()
ep = ProcessRPCResponseEndpointUnit(process=procmock)
ep._message_received(sentinel.msg, sentinel.headers)
procmock.push_context.assert_called_once_with(sentinel.headers)
procmock.push_context().__enter__.assert_called_once_with()
mockmr.assert_called_once_with(ep, sentinel.msg, sentinel.headers)
def test_init(self, mockr, mockp):
ep = ProcessRPCResponseEndpointUnit(process=sentinel.process, other=sentinel.other)
mockp.assert_called_once_with(ep, process=sentinel.process)
mockr.assert_called_once_with(ep, other=sentinel.other)