Skip to content

Commit 0fc618e

Browse files
committed
mock correctly
1 parent 70d640a commit 0fc618e

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

tests/test_03_key_bundle.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -517,13 +517,14 @@ def mocked_jwks_response():
517517
def test_httpc_params_1():
518518
source = 'https://login.salesforce.com/id/keys' # From test_jwks_url()
519519
# Mock response
520-
responses.add(method=responses.GET, url=source, json=JWKS_DICT, status=200)
521-
httpc_params = {'timeout': (2, 2)} # connect, read timeouts in seconds
522-
kb = KeyBundle(source=source, httpc=requests.request,
523-
httpc_params=httpc_params)
524-
assert kb.do_remote()
525-
520+
with responses.RequestsMock() as rsps:
521+
rsps.add(method=responses.GET, url=source, json=JWKS_DICT, status=200)
522+
httpc_params = {'timeout': (2, 2)} # connect, read timeouts in seconds
523+
kb = KeyBundle(source=source, httpc=requests.request,
524+
httpc_params=httpc_params)
525+
assert kb.do_remote()
526526

527+
@pytest.mark.network
527528
def test_httpc_params_2():
528529
httpc_params = {'timeout': 0}
529530
kb = KeyBundle(source='https://login.salesforce.com/id/keys',

0 commit comments

Comments
 (0)