def test_get_metadata_invalid_service_id(self):
with mock.patch.object(openeogeotrellis.service_registry, 'KazooClient') as KazooClient:
reg = ZooKeeperServiceRegistry()
client = KazooClient.return_value
client.get.side_effect = NoNodeError
with pytest.raises(ServiceNotFoundException, match="Service 'foobar' does not exist."):
reg.get_metadata('foobar')
def test_get_metadata(self):
with mock.patch.object(openeogeotrellis.service_registry, 'KazooClient') as KazooClient:
reg = ZooKeeperServiceRegistry()
reg.persist(user_id='u9876', metadata=dummy_service_metadata, api_version="0.4.0")
# Extract "created" data
client = KazooClient.return_value
assert client.create.call_count == 1
path, raw = client.create.call_args_list[0][0]
# Set up return value for zookeeper "get"
assert isinstance(raw, bytes)
storage = {path: raw}
client.get.side_effect = lambda p: (storage[p], "dummy")
metadata = reg.get_metadata('u9876', 's1234')
assert metadata == dummy_service_metadata
def test_get_metadata(self):
with mock.patch.object(openeogeotrellis.service_registry, 'KazooClient') as KazooClient:
reg = ZooKeeperServiceRegistry()
reg.register(
SecondaryService(service_metadata=dummy_service_metadata, host='oeo.net', port=5678, server=mock.Mock)
)
# Extract "created" data
client = KazooClient.return_value
assert client.create.call_count == 1
path, raw = client.create.call_args_list[0][0]
# Set up return value for zookeeper "get"
assert isinstance(raw, bytes)
storage = {path: raw}
client.get.side_effect = lambda p: (storage[p], "dummy")
metadata = reg.get_metadata('s1234')
assert metadata == dummy_service_metadata