def test_get_metadata_all(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_children.side_effect = lambda up: [] if up.endswith("/_anonymous") else [path.split('/')[-1]] client.get.side_effect = lambda p: (storage[p], "dummy") metadata_all = reg.get_metadata_all('u9876') assert metadata_all == {'s1234': dummy_service_metadata}
def test_get_metadata_all(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_children.side_effect = lambda r: [path.split('/')[-1]] client.get.side_effect = lambda p: (storage[p], "dummy") metadata_all = reg.get_metadata_all() assert metadata_all == {'s1234': dummy_service_metadata}