class TestReplayContext(unittest.TestCase):
def setUp(self):
self.config = load_config()
self.config['name'] = local_name
self.config['persistent_store'] = mock.Mock()
self.replay_context = ReplayContext(**self.config)
self.request_context = zmq.Context()
self.request_socket = self.request_context.socket(zmq.REQ)
self.request_socket.connect(
self.config['replay_endpoints'][local_name])
def tearDown(self):
self.request_socket.close()
self.replay_context.publisher.close()
@requires_network
def test_get_replay(self):
# Setup the store to return what we ask.
answer = [{'foo': 'bar'}]
self.config['persistent_store'].get = mock.Mock(side_effect=[answer])
# Doesn't matter what we send as long as it is legit JSON,
# since the store is mocked
self.request_socket.send(u'{"id": 1}'.encode('utf-8'))
self.replay_context._req_rep_cycle()
rep = self.request_socket.recv_multipart()
print(rep)
print(answer)
assert len(answer) == len(rep)
for r, a in zip(rep, answer):
self.assertDictEqual(json.loads(r.decode('utf-8')), a)
@requires_network
def test_get_error(self):
# Setup the store to return what we ask.
answer = ValueError('No luck!')
self.config['persistent_store'].get = mock.Mock(side_effect=[answer])
# Doesn't matter what we send as long as it is legit JSON,
# since the store is mocked
self.request_socket.send(u'{"id": 1}'.encode('utf-8'))
self.replay_context._req_rep_cycle()
rep = self.request_socket.recv_multipart()
print(rep)
assert len(rep) == 1 and "error: 'No luck!'" == rep[0].decode('utf-8')
class TestReplayContext(unittest.TestCase):
def setUp(self):
self.config = load_config()
self.config['name'] = local_name
self.config['persistent_store'] = mock.Mock()
self.replay_context = ReplayContext(**self.config)
self.request_context = zmq.Context()
self.request_socket = self.request_context.socket(zmq.REQ)
self.request_socket.connect(
self.config['replay_endpoints'][local_name])
def tearDown(self):
self.request_socket.close()
self.replay_context.publisher.close()
@requires_network
def test_get_replay(self):
# Setup the store to return what we ask.
answer = [{'foo': 'bar'}]
self.config['persistent_store'].get = mock.Mock(side_effect=[answer])
# Doesn't matter what we send as long as it is legit JSON,
# since the store is mocked
self.request_socket.send(u'{"id": 1}'.encode('utf-8'))
self.replay_context._req_rep_cycle()
rep = self.request_socket.recv_multipart()
print(rep)
print(answer)
assert len(answer) == len(rep)
for r, a in zip(rep, answer):
self.assertDictEqual(json.loads(r.decode('utf-8')), a)
@requires_network
def test_get_error(self):
# Setup the store to return what we ask.
answer = ValueError('No luck!')
self.config['persistent_store'].get = mock.Mock(side_effect=[answer])
# Doesn't matter what we send as long as it is legit JSON,
# since the store is mocked
self.request_socket.send(u'{"id": 1}'.encode('utf-8'))
self.replay_context._req_rep_cycle()
rep = self.request_socket.recv_multipart()
print(rep)
assert len(rep) == 1 and "error: 'No luck!'" == rep[0].decode('utf-8')
