diff --git a/servicex/servicex_adapter.py b/servicex/servicex_adapter.py index 52b48368..cbb51b27 100644 --- a/servicex/servicex_adapter.py +++ b/servicex/servicex_adapter.py @@ -171,7 +171,7 @@ async def get_dataset(self, dataset_id=None) -> CachedDataset: dataset = CachedDataset(**result) return dataset - async def delete_dataset(self, dataset_id=None): + async def delete_dataset(self, dataset_id=None) -> bool: headers = await self._get_authorization() path_template = '/servicex/datasets/{dataset_id}' url = self.url + path_template.format(dataset_id=dataset_id) @@ -189,6 +189,8 @@ async def delete_dataset(self, dataset_id=None): elif r.status != 200: msg = await _extract_message(r) raise RuntimeError(f"Failed to delete dataset {dataset_id} - {msg}") + result = await r.json() + return result['stale'] async def submit_transform(self, transform_request: TransformRequest) -> str: headers = await self._get_authorization() diff --git a/tests/test_servicex_adapter.py b/tests/test_servicex_adapter.py index 208c3063..34c7d699 100644 --- a/tests/test_servicex_adapter.py +++ b/tests/test_servicex_adapter.py @@ -247,11 +247,12 @@ async def test_delete_dataset(delete, servicex): } delete.return_value.__aenter__.return_value.status = 200 - await servicex.delete_dataset(123) + r = await servicex.delete_dataset(123) delete.assert_called_with( url='https://servicex.org/servicex/datasets/123', headers={} ) + assert r @pytest.mark.asyncio