diff --git a/tests/test_servicex_adapter.py b/tests/test_servicex_adapter.py index f410e47f..9c5810a8 100644 --- a/tests/test_servicex_adapter.py +++ b/tests/test_servicex_adapter.py @@ -92,15 +92,13 @@ async def test_get_transforms_auth_error(mock_get, servicex): @pytest.mark.asyncio -@patch("servicex.servicex_adapter.RetryClient.post") +@patch("servicex.servicex_adapter.AsyncClient.post") @patch("servicex.servicex_adapter.jwt.decode") async def test_get_transforms_wlcg_bearer_token( decode, post, servicex, transform_status_response ): - post.return_value.__aenter__.return_value.json.return_value = { - "access_token": "luckycharms" - } - post.return_value.__aenter__.return_value.status = 401 + post.return_value.json.return_value = {"access_token": "luckycharms"} + post.return_value.status_code = 401 token_file = tempfile.NamedTemporaryFile(mode="w+t", delete=False) token_file.write( """"