Skip to content

Commit 91fd8e6

Browse files
authored
Merge pull request #139 from ssl-hep:gordonwatts/issue118
Cached queries should not talk to the internet
2 parents caad92a + 6cade06 commit 91fd8e6

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

servicex/servicex.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,8 @@ async def _get_files(self, selection_query: str, data_type: str,
283283
yield r
284284

285285
# Cache the final status
286-
await self._update_query_status(client, request_id)
286+
if cached_files is None:
287+
await self._update_query_status(client, request_id)
287288

288289
except ServiceXUnknownRequestID as e:
289290
self._cache.remove_query(query)

tests/test_servicex.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -778,8 +778,11 @@ async def test_download_cached_nonet(mocker):
778778
mock_cache = build_cache_mock(mocker, query_cache_return='123-455',
779779
files=[('f1', 'file1.root')])
780780
mock_logger = mocker.MagicMock(spec=log_adaptor)
781-
mock_transform_status = mocker.Mock(side_effect=[(0, 2, 0)])
782-
mock_servicex_adaptor = MockServiceXAdaptor(mocker, "123-456", mock_transform_status)
781+
mock_bomb = mocker.Mock(side_effect=RuntimeError('should not be called'))
782+
mock_servicex_adaptor = MockServiceXAdaptor(mocker, "XXX-XXX",
783+
mock_transform_status=mock_bomb,
784+
mock_query=mock_bomb,
785+
mock_transform_query_status=mock_bomb)
783786
mock_minio_adaptor = MockMinioAdaptor(mocker, files=['one_minio_entry', 'two_minio_entry'])
784787
data_adaptor = mocker.MagicMock(spec=DataConverterAdaptor)
785788

@@ -791,10 +794,6 @@ async def test_download_cached_nonet(mocker):
791794
local_log=mock_logger)
792795
await ds.get_data_rootfiles_async('(valid qastle string')
793796

794-
# Check the the number of times we called for a transform is good.
795-
mock_transform_status.submit_query.assert_not_called()
796-
mock_transform_status.transform_status.assert_not_called()
797-
798797

799798
@pytest.mark.asyncio
800799
async def test_download_write_to_inmem_cache(mocker):

0 commit comments

Comments
 (0)