def test_set(self):
server = RegistryServer(server_uri='tcp://127.0.0.1:11999')
client = RegistryClient(server_uri='tcp://127.0.0.1:11999')
client.set('a','b')
client.set('x.y','z')
client.commit()
assert client._RegistryClient__registry.get_values() == server.get_values()
client.set('a','c')
assert client._RegistryClient__registry.get_values() != server.get_values()
class TestRegistryServer:
def setUp(self):
self.server_socket_mock = Mock()
self.msg_thread_mock = Mock()
self.patcher = patch('src.registry_server.ServerSocket', self.server_socket_mock)
self.msg_thread_patcher = patch('src.registry_server.MsgThread', self.msg_thread_mock)
self.patcher.start()
self.msg_thread_patcher.start()
self.s = RegistryServer({'a':'b'})
def tearDown(self):
self.patcher.stop()
self.msg_thread_patcher.stop()
def test_get(self):
value = self.s.get('a')
assert value == 'b'
def test_set(self):
self.s.set('a','c')
value = self.s.get('a')
assert value == 'c'
def test_remove(self):
self.s.set('a','c')
self.s.remove('a')
assert not self.s.get('a')
def test_commit(self):
self.s.commit({'a':'b','c':'d'})
assert self.s._RegistryServer__registry.get_values()[1] == {'a':'b','c':'d'}
def test_get_values(self):
self.s.commit({'a':'b'})
assert self.s.get_values()[1] == {'a':'b'}
def test_listens(self):
assert self.s.socket
def test_msg_thread_started(self):
assert self.msg_thread_mock.called
assert self.msg_thread_mock.call_args[0][0] == self.s
assert self.msg_thread_mock.call_args[0][1] == self.s.socket
assert self.msg_thread_mock().start.called
def test_calls_callback_when_changed(self):
refresh_func_mock = Mock()
c = RegistryServer({}, refresh_callback=refresh_func_mock)
c.set('a','b')
assert refresh_func_mock.called
assert refresh_func_mock.call_args == call(c)
refresh_func_mock.reset_mock()
c = RegistryServer({}, refresh_callback=refresh_func_mock)
c.set('a','b')
c.remove('a')
assert refresh_func_mock.called
refresh_func_mock.reset_mock()
c = RegistryServer({}, refresh_callback=refresh_func_mock)
c.commit({})
assert refresh_func_mock.called