Skip to content

Commit 875a22e

Browse files
escattoneescattone
committed
add status_code to response Mocks
1 parent 061b888 commit 875a22e

File tree

1 file changed

+42
-42
lines changed

1 file changed

+42
-42
lines changed

‎tests/test_auth.py

Lines changed: 42 additions & 42 deletions
Original file line numberDiff line numberDiff line change
@@ -55,13 +55,13 @@ def test_invalid_token(self, request_mock, token_mock):
5555
auth_request.session = {}
5656

5757
token_mock.return_value = None
58-
get_json_mock = Mock()
58+
get_json_mock = Mock(status_code=200)
5959
get_json_mock.json.return_value = {
6060
"nickname": "username",
6161
"email": "email@example.com",
6262
}
6363
request_mock.get.return_value = get_json_mock
64-
post_json_mock = Mock()
64+
post_json_mock = Mock(status_code=200)
6565
post_json_mock.json.return_value = {
6666
"id_token": "id_token",
6767
"accesss_token": "access_token",
@@ -233,13 +233,13 @@ def test_successful_authentication_existing_user_namespaced(
233233
username="a_username", email="email@example.com"
234234
)
235235
token_mock.return_value = True
236-
get_json_mock = Mock()
236+
get_json_mock = Mock(status_code=200)
237237
get_json_mock.json.return_value = {
238238
"nickname": "a_username",
239239
"email": "email@example.com",
240240
}
241241
request_mock.get.return_value = get_json_mock
242-
post_json_mock = Mock()
242+
post_json_mock = Mock(status_code=200)
243243
post_json_mock.json.return_value = {
244244
"id_token": "id_token",
245245
"access_token": "access_granted",
@@ -282,13 +282,13 @@ def test_successful_authentication_existing_user(self, token_mock, request_mock)
282282
username="a_username", email="email@example.com"
283283
)
284284
token_mock.return_value = True
285-
get_json_mock = Mock()
285+
get_json_mock = Mock(status_code=200)
286286
get_json_mock.json.return_value = {
287287
"nickname": "a_username",
288288
"email": "email@example.com",
289289
}
290290
request_mock.get.return_value = get_json_mock
291-
post_json_mock = Mock()
291+
post_json_mock = Mock(status_code=200)
292292
post_json_mock.json.return_value = {
293293
"id_token": "id_token",
294294
"access_token": "access_granted",
@@ -335,13 +335,13 @@ def test_successful_authentication_existing_user_upper_case(
335335
username="a_username", email="EMAIL@EXAMPLE.COM"
336336
)
337337
token_mock.return_value = True
338-
get_json_mock = Mock()
338+
get_json_mock = Mock(status_code=200)
339339
get_json_mock.json.return_value = {
340340
"nickname": "a_username",
341341
"email": "email@example.com",
342342
}
343343
request_mock.get.return_value = get_json_mock
344-
post_json_mock = Mock()
344+
post_json_mock = Mock(status_code=200)
345345
post_json_mock.json.return_value = {
346346
"id_token": "id_token",
347347
"access_token": "access_granted",
@@ -390,11 +390,11 @@ def test_failed_authentication_verify_claims(
390390
User.objects.create_user(username="a_username", email="email@example.com")
391391
token_mock.return_value = True
392392
claims_mock.return_value = False
393-
get_json_mock = Mock()
393+
get_json_mock = Mock(status_code=200)
394394
claims_response = {"nickname": "a_username", "email": "email@example.com"}
395395
get_json_mock.json.return_value = claims_response
396396
request_mock.get.return_value = get_json_mock
397-
post_json_mock = Mock()
397+
post_json_mock = Mock(status_code=200)
398398
post_json_mock.json.return_value = {
399399
"id_token": "id_token",
400400
"access_token": "access_granted",
@@ -439,13 +439,13 @@ def test_successful_authentication_new_user(
439439

440440
algo_mock.return_value = "username_algo"
441441
token_mock.return_value = True
442-
get_json_mock = Mock()
442+
get_json_mock = Mock(status_code=200)
443443
get_json_mock.json.return_value = {
444444
"nickname": "a_username",
445445
"email": "email@example.com",
446446
}
447447
request_mock.get.return_value = get_json_mock
448-
post_json_mock = Mock()
448+
post_json_mock = Mock(status_code=200)
449449
post_json_mock.json.return_value = {
450450
"id_token": "id_token",
451451
"access_token": "access_granted",
@@ -499,13 +499,13 @@ def test_successful_authentication_basic_auth_token(self, token_mock, request_mo
499499
username="a_username", email="EMAIL@EXAMPLE.COM"
500500
)
501501
token_mock.return_value = True
502-
get_json_mock = Mock()
502+
get_json_mock = Mock(status_code=200)
503503
get_json_mock.json.return_value = {
504504
"nickname": "a_username",
505505
"email": "email@example.com",
506506
}
507507
request_mock.get.return_value = get_json_mock
508-
post_json_mock = Mock()
508+
post_json_mock = Mock(status_code=200)
509509
post_json_mock.json.return_value = {
510510
"id_token": "id_token",
511511
"access_token": "access_granted",
@@ -570,13 +570,13 @@ def test_jwt_decode_params(self, request_mock, jws_mock):
570570
auth_request.session = {}
571571

572572
jws_mock.return_value = json.dumps({"aud": "audience"}).encode("utf-8")
573-
get_json_mock = Mock()
573+
get_json_mock = Mock(status_code=200)
574574
get_json_mock.json.return_value = {
575575
"nickname": "username",
576576
"email": "email@example.com",
577577
}
578578
request_mock.get.return_value = get_json_mock
579-
post_json_mock = Mock()
579+
post_json_mock = Mock(status_code=200)
580580
post_json_mock.json.return_value = {
581581
"id_token": "token",
582582
"access_token": "access_token",
@@ -596,13 +596,13 @@ def test_jwt_decode_params_verify_false(self, request_mock, jws_mock):
596596
auth_request.session = {}
597597

598598
jws_mock.return_value = json.dumps({"aud": "audience"}).encode("utf-8")
599-
get_json_mock = Mock()
599+
get_json_mock = Mock(status_code=200)
600600
get_json_mock.json.return_value = {
601601
"nickname": "username",
602602
"email": "email@example.com",
603603
}
604604
request_mock.get.return_value = get_json_mock
605-
post_json_mock = Mock()
605+
post_json_mock = Mock(status_code=200)
606606
post_json_mock.json.return_value = {
607607
"id_token": "token",
608608
"access_token": "access_token",
@@ -636,13 +636,13 @@ def test_create_user_disabled(self, request_mock, jws_mock):
636636
auth_request.session = {}
637637

638638
jws_mock.return_value = json.dumps({"nonce": "nonce"}).encode("utf-8")
639-
get_json_mock = Mock()
639+
get_json_mock = Mock(status_code=200)
640640
get_json_mock.json.return_value = {
641641
"nickname": "a_username",
642642
"email": "email@example.com",
643643
}
644644
request_mock.get.return_value = get_json_mock
645-
post_json_mock = Mock()
645+
post_json_mock = Mock(status_code=200)
646646
post_json_mock.json.return_value = {
647647
"id_token": "id_token",
648648
"access_token": "access_granted",
@@ -660,13 +660,13 @@ def test_create_user_enabled(self, request_mock, jws_mock):
660660

661661
self.assertEqual(User.objects.filter(email="email@example.com").exists(), False)
662662
jws_mock.return_value = json.dumps({"nonce": "nonce"}).encode("utf-8")
663-
get_json_mock = Mock()
663+
get_json_mock = Mock(status_code=200)
664664
get_json_mock.json.return_value = {
665665
"nickname": "a_username",
666666
"email": "email@example.com",
667667
}
668668
request_mock.get.return_value = get_json_mock
669-
post_json_mock = Mock()
669+
post_json_mock = Mock(status_code=200)
670670
post_json_mock.json.return_value = {
671671
"id_token": "id_token",
672672
"access_token": "access_granted",
@@ -689,13 +689,13 @@ def test_custom_username_algo(self, request_mock, jws_mock, algo_mock):
689689
self.assertEqual(User.objects.filter(email="email@example.com").exists(), False)
690690
algo_mock.return_value = "username_algo"
691691
jws_mock.return_value = json.dumps({"nonce": "nonce"}).encode("utf-8")
692-
get_json_mock = Mock()
692+
get_json_mock = Mock(status_code=200)
693693
get_json_mock.json.return_value = {
694694
"nickname": "a_username",
695695
"email": "email@example.com",
696696
}
697697
request_mock.get.return_value = get_json_mock
698-
post_json_mock = Mock()
698+
post_json_mock = Mock(status_code=200)
699699
post_json_mock.json.return_value = {
700700
"id_token": "id_token",
701701
"access_token": "access_granted",
@@ -719,13 +719,13 @@ def test_custom_username_algo_dotted_path(self, request_mock, jws_mock):
719719

720720
self.assertEqual(User.objects.filter(email="email@example.com").exists(), False)
721721
jws_mock.return_value = json.dumps({"nonce": "nonce"}).encode("utf-8")
722-
get_json_mock = Mock()
722+
get_json_mock = Mock(status_code=200)
723723
get_json_mock.json.return_value = {
724724
"nickname": "a_username",
725725
"email": "email@example.com",
726726
}
727727
request_mock.get.return_value = get_json_mock
728-
post_json_mock = Mock()
728+
post_json_mock = Mock(status_code=200)
729729
post_json_mock.json.return_value = {
730730
"id_token": "id_token",
731731
"access_token": "access_granted",
@@ -747,13 +747,13 @@ def test_duplicate_emails_exact(self, request_mock, jws_mock):
747747
User.objects.create(username="user1", email="email@example.com")
748748
User.objects.create(username="user2", email="email@example.com")
749749
jws_mock.return_value = json.dumps({"nonce": "nonce"}).encode("utf-8")
750-
get_json_mock = Mock()
750+
get_json_mock = Mock(status_code=200)
751751
get_json_mock.json.return_value = {
752752
"nickname": "a_username",
753753
"email": "email@example.com",
754754
}
755755
request_mock.get.return_value = get_json_mock
756-
post_json_mock = Mock()
756+
post_json_mock = Mock(status_code=200)
757757
post_json_mock.json.return_value = {
758758
"id_token": "id_token",
759759
"access_token": "access_granted",
@@ -772,13 +772,13 @@ def test_duplicate_emails_case_mismatch(self, request_mock, jws_mock):
772772
User.objects.create(username="user1", email="email@example.com")
773773
User.objects.create(username="user2", email="eMaIl@ExAmPlE.cOm")
774774
jws_mock.return_value = json.dumps({"nonce": "nonce"}).encode("utf-8")
775-
get_json_mock = Mock()
775+
get_json_mock = Mock(status_code=200)
776776
get_json_mock.json.return_value = {
777777
"nickname": "a_username",
778778
"email": "email@example.com",
779779
}
780780
request_mock.get.return_value = get_json_mock
781-
post_json_mock = Mock()
781+
post_json_mock = Mock(status_code=200)
782782
post_json_mock.json.return_value = {
783783
"id_token": "id_token",
784784
"access_token": "access_granted",
@@ -806,13 +806,13 @@ def update_user(user, claims):
806806
update_user_mock.side_effect = update_user
807807

808808
jws_mock.return_value = json.dumps({"nonce": "nonce"}).encode("utf-8")
809-
get_json_mock = Mock()
809+
get_json_mock = Mock(status_code=200)
810810
get_json_mock.json.return_value = {
811811
"nickname": "a_username",
812812
"email": "email@example.com",
813813
}
814814
request_mock.get.return_value = get_json_mock
815-
post_json_mock = Mock()
815+
post_json_mock = Mock(status_code=200)
816816
post_json_mock.json.return_value = {
817817
"id_token": "id_token",
818818
"access_token": "access_granted",
@@ -844,13 +844,13 @@ def test_jwt_verify_sign_key(self, request_mock, jws_mock):
844844
auth_request.session = {}
845845

846846
jws_mock.return_value = json.dumps({"aud": "audience"}).encode("utf-8")
847-
get_json_mock = Mock()
847+
get_json_mock = Mock(status_code=200)
848848
get_json_mock.json.return_value = {
849849
"nickname": "username",
850850
"email": "email@example.com",
851851
}
852852
request_mock.get.return_value = get_json_mock
853-
post_json_mock = Mock()
853+
post_json_mock = Mock(status_code=200)
854854
post_json_mock.json.return_value = {
855855
"id_token": "token",
856856
"access_token": "access_token",
@@ -893,13 +893,13 @@ def test_jwt_verify_sign_key_calls(self, request_mock, jwk_mock, jws_mock):
893893
jwk_mock.return_value = jwk_mock_ret
894894

895895
jws_mock.return_value = json.dumps({"aud": "audience"}).encode("utf-8")
896-
get_json_mock = Mock()
896+
get_json_mock = Mock(status_code=200)
897897
get_json_mock.json.return_value = {
898898
"nickname": "username",
899899
"email": "email@example.com",
900900
}
901901
request_mock.get.return_value = get_json_mock
902-
post_json_mock = Mock()
902+
post_json_mock = Mock(status_code=200)
903903
post_json_mock.json.return_value = {
904904
"id_token": "token",
905905
"access_token": "access_token",
@@ -913,7 +913,7 @@ def test_jwt_verify_sign_key_calls(self, request_mock, jwk_mock, jws_mock):
913913
def test_retrieve_matching_jwk(self, mock_requests):
914914
"""Test retrieving valid jwk"""
915915

916-
get_json_mock = Mock()
916+
get_json_mock = Mock(status_code=200)
917917
get_json_mock.json.return_value = {
918918
"keys": [
919919
{
@@ -955,7 +955,7 @@ def test_retrieve_matching_jwk(self, mock_requests):
955955
def test_retrieve_matching_jwk_same_kid(self, mock_requests):
956956
"""Test retrieving valid jwk from a list of keys with the same kid"""
957957

958-
get_json_mock = Mock()
958+
get_json_mock = Mock(status_code=200)
959959
get_json_mock.json.return_value = {
960960
"keys": [
961961
{
@@ -1001,7 +1001,7 @@ def test_retrieve_matching_jwk_same_kid(self, mock_requests):
10011001
def test_retrieve_mismatcing_jwk_alg(self, mock_requests):
10021002
"""Test retrieving mismatching jwk alg"""
10031003

1004-
get_json_mock = Mock()
1004+
get_json_mock = Mock(status_code=200)
10051005
get_json_mock.json.return_value = {
10061006
"keys": [
10071007
{
@@ -1039,7 +1039,7 @@ def test_retrieve_mismatcing_jwk_alg(self, mock_requests):
10391039
def test_retrieve_mismatcing_jwk_kid(self, mock_requests):
10401040
"""Test retrieving mismatching jwk kid"""
10411041

1042-
get_json_mock = Mock()
1042+
get_json_mock = Mock(status_code=200)
10431043
get_json_mock.json.return_value = {
10441044
"keys": [
10451045
{
@@ -1077,7 +1077,7 @@ def test_retrieve_mismatcing_jwk_kid(self, mock_requests):
10771077
def test_retrieve_jwk_optional_alg(self, mock_requests):
10781078
"""Test retrieving jwk with optional alg"""
10791079

1080-
get_json_mock = Mock()
1080+
get_json_mock = Mock(status_code=200)
10811081
get_json_mock.json.return_value = {
10821082
"keys": [
10831083
{
@@ -1112,7 +1112,7 @@ def test_retrieve_jwk_optional_alg(self, mock_requests):
11121112
def test_retrieve_not_existing_jwk(self, mock_requests):
11131113
"""Test retrieving jwk that doesn't exist."""
11141114

1115-
get_json_mock = Mock()
1115+
get_json_mock = Mock(status_code=200)
11161116
get_json_mock.json.return_value = {"keys": [{"alg": "RS256", "kid": "kid"}]}
11171117
mock_requests.get.return_value = get_json_mock
11181118

0 commit comments

Comments
 (0)