def test_stop_service(self):
with mock.patch.object(openeogeotrellis.service_registry, 'KazooClient') as KazooClient:
reg = ZooKeeperServiceRegistry()
server = mock.Mock()
reg.register(
SecondaryService(service_metadata=dummy_service_metadata, host='oeo.net', port=5678, server=server)
)
reg.stop_service(dummy_service_metadata.id)
client = KazooClient.return_value
# print(client.mock_calls)
client.delete.assert_called_with('/openeo/services/s1234')
server.stop.assert_called_once()
def test_stop_service(self):
with mock.patch.object(openeogeotrellis.service_registry, 'KazooClient') as KazooClient:
reg = ZooKeeperServiceRegistry()
server = mock.Mock()
reg.register(dummy_service_metadata.id, SecondaryService(host='oeo.net', port=5678, server=server))
reg.persist(user_id='u9876', metadata=dummy_service_metadata, api_version="0.4.0")
reg.stop_service('u9876', dummy_service_metadata.id)
client = KazooClient.return_value
# print(client.mock_calls)
client.delete.assert_any_call('/openeo/services/u9876/s1234')
server.stop.assert_called_once()