Skip to content

Commit cfeca26

Browse files
committed
test case for retry runtime error
1 parent 45a0ee8 commit cfeca26

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

tests/test_servicex_adapter.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -199,8 +199,15 @@ async def test_get_transform_status_auth_error(get, servicex):
199199

200200

201201
@pytest.mark.asyncio
202-
async def test_get_tranform_status_retry_error(servicex):
203-
with pytest.raises(Exception) as err:
202+
@patch('servicex.servicex_adapter.TransformStatus', side_effect=RuntimeError)
203+
@patch('servicex.servicex_adapter.RetryClient.get')
204+
async def test_get_tranform_status_retry_error(get,
205+
mock_transform_status,
206+
servicex,
207+
transform_status_response):
208+
with pytest.raises(RuntimeError) as err:
209+
get.return_value.__aenter__.return_value.json.return_value = transform_status_response['requests'][0] # NOQA: E501
210+
get.return_value.__aenter__.return_value.status = 200
204211
await servicex.get_transform_status("b8c508d0-ccf2-4deb-a1f7-65c839eebabf")
205212
assert "ServiceX WebAPI Error while getting transform status:" in str(err.value)
206213

0 commit comments

Comments
 (0)