From 492277bf416cad7a531f5d4144c32f2a44be33db Mon Sep 17 00:00:00 2001 From: Claudiu Belu Date: Fri, 24 Apr 2026 11:19:37 +0000 Subject: [PATCH] Fix getting the endpoint storage default Currently, trying to get the endpoint storage default ends with the following error: ``` Traceback (most recent call last): File "/home/ubuntu/workdir/python-coriolisclient/v1/endpoint_storage.py", line 48, in get_default TypeError: Resource.get() takes 1 positional argument but 2 were given ``` --- coriolisclient/tests/v1/test_endpoint_storage.py | 3 ++- coriolisclient/v1/endpoint_storage.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/coriolisclient/tests/v1/test_endpoint_storage.py b/coriolisclient/tests/v1/test_endpoint_storage.py index 23d7acc..f845b31 100644 --- a/coriolisclient/tests/v1/test_endpoint_storage.py +++ b/coriolisclient/tests/v1/test_endpoint_storage.py @@ -46,7 +46,8 @@ def test_get_default( ): mock_endpoint = mock.Mock() mock_endpoint.uuid = '53773ab8-1474-4cf7-bf0c-a496a6595ecb' - mock_get.return_value = {"config_default": "mock_default"} + mock_get.return_value = endpoint_storage.EndpointStorage( + mock.Mock(), {"config_default": "mock_default"}) result = self.endpoint.get_default( mock_endpoint, diff --git a/coriolisclient/v1/endpoint_storage.py b/coriolisclient/v1/endpoint_storage.py index 0c8ee8c..36d81d2 100644 --- a/coriolisclient/v1/endpoint_storage.py +++ b/coriolisclient/v1/endpoint_storage.py @@ -43,4 +43,4 @@ def get_default(self, endpoint, environment=None): encoded_env = common.encode_base64_param(environment, is_json=True) url = '%s?env=%s' % (url, encoded_env) - return self._get(url, 'storage').get('config_default') + return self._get(url, 'storage').to_dict().get('config_default')