1414import unittest
1515
1616from requests_gssapi import REQUIRED
17+ from requests_gssapi import SPNEGO
1718
1819# Note: we're not using the @mock.patch decorator:
1920# > My only word of warning is that in the past, the patch decorator hides
@@ -110,7 +111,7 @@ def test_generate_request_header(self):
110111 b64_negotiate_response )
111112 fake_init .assert_called_with (
112113 name = gssapi_sname (
"[email protected] " ),
113- creds = None , mech = None , flags = gssflags , usage = "initiate" )
114+ creds = None , mech = SPNEGO , flags = gssflags , usage = "initiate" )
114115 fake_resp .assert_called_with (b"token" )
115116
116117 def test_generate_request_header_init_error (self ):
@@ -125,7 +126,7 @@ def test_generate_request_header_init_error(self):
125126 auth .generate_request_header , response , host )
126127 fake_init .assert_called_with (
127128 name = gssapi_sname (
"[email protected] " ),
128- usage = "initiate" , flags = gssflags , creds = None , mech = None )
129+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
129130
130131 def test_generate_request_header_step_error (self ):
131132 with patch .multiple ("gssapi.SecurityContext" , __init__ = fake_init ,
@@ -139,7 +140,7 @@ def test_generate_request_header_step_error(self):
139140 auth .generate_request_header , response , host )
140141 fake_init .assert_called_with (
141142 name = gssapi_sname (
"[email protected] " ),
142- usage = "initiate" , flags = gssflags , creds = None , mech = None )
143+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
143144 fail_resp .assert_called_with (b"token" )
144145
145146 def test_authenticate_user (self ):
@@ -176,7 +177,7 @@ def test_authenticate_user(self):
176177 raw .release_conn .assert_called_with ()
177178 fake_init .assert_called_with (
178179 name = gssapi_sname (
"[email protected] " ),
179- flags = gssflags , usage = "initiate" , creds = None , mech = None )
180+ flags = gssflags , usage = "initiate" , creds = None , mech = SPNEGO )
180181 fake_resp .assert_called_with (b"token" )
181182
182183 def test_handle_401 (self ):
@@ -213,7 +214,7 @@ def test_handle_401(self):
213214 raw .release_conn .assert_called_with ()
214215 fake_init .assert_called_with (
215216 name = gssapi_sname (
"[email protected] " ),
216- creds = None , mech = None , flags = gssflags , usage = "initiate" )
217+ creds = None , mech = SPNEGO , flags = gssflags , usage = "initiate" )
217218 fake_resp .assert_called_with (b"token" )
218219
219220 def test_authenticate_server (self ):
@@ -452,7 +453,7 @@ def test_handle_response_401(self):
452453 raw .release_conn .assert_called_with ()
453454 fake_init .assert_called_with (
454455 name = gssapi_sname (
"[email protected] " ),
455- usage = "initiate" , flags = gssflags , creds = None , mech = None )
456+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
456457 fake_resp .assert_called_with (b"token" )
457458
458459 def test_handle_response_401_rejected (self ):
@@ -495,7 +496,7 @@ def connection_send(self, *args, **kwargs):
495496 raw .release_conn .assert_called_with ()
496497 fake_init .assert_called_with (
497498 name = gssapi_sname (
"[email protected] " ),
498- usage = "initiate" , flags = gssflags , creds = None , mech = None )
499+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
499500 fake_resp .assert_called_with (b"token" )
500501
501502 def test_generate_request_header_custom_service (self ):
@@ -509,7 +510,7 @@ def test_generate_request_header_custom_service(self):
509510 auth .generate_request_header (response , host ),
510511 fake_init .assert_called_with (
511512 name = gssapi_sname (
"[email protected] " ),
512- usage = "initiate" , flags = gssflags , creds = None , mech = None )
513+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
513514 fake_resp .assert_called_with (b"token" )
514515
515516 def test_delegation (self ):
@@ -547,7 +548,7 @@ def test_delegation(self):
547548 raw .release_conn .assert_called_with ()
548549 fake_init .assert_called_with (
549550 name = gssapi_sname (
"[email protected] " ),
550- usage = "initiate" , flags = gssdelegflags , creds = None , mech = None )
551+ usage = "initiate" , flags = gssdelegflags , creds = None , mech = SPNEGO )
551552 fake_resp .assert_called_with (b"token" )
552553
553554 def test_principal_override (self ):
@@ -566,7 +567,7 @@ def test_principal_override(self):
566567 fake_init .assert_called_with (
567568 name = gssapi_sname (
"[email protected] " ),
568569 usage = "initiate" , flags = gssflags ,
569- creds = b"fake creds" , mech = None )
570+ creds = b"fake creds" , mech = SPNEGO )
570571
571572 def test_realm_override (self ):
572573 with patch .multiple ("gssapi.SecurityContext" , __init__ = fake_init ,
@@ -580,7 +581,7 @@ def test_realm_override(self):
580581 auth .generate_request_header (response , host )
581582 fake_init .assert_called_with (
582583 name = gssapi_sname (
"[email protected] " ),
583- usage = "initiate" , flags = gssflags , creds = None , mech = None )
584+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
584585 fake_resp .assert_called_with (b"token" )
585586
586587 def test_opportunistic_auth (self ):
@@ -610,7 +611,7 @@ def test_explicit_creds(self):
610611 fake_init .assert_called_with (
611612 name = gssapi_sname (
"[email protected] " ),
612613 usage = "initiate" , flags = gssflags ,
613- creds = b"fake creds" , mech = None )
614+ creds = b"fake creds" , mech = SPNEGO )
614615 fake_resp .assert_called_with (b"token" )
615616
616617 def test_explicit_mech (self ):
@@ -642,7 +643,7 @@ def test_target_name(self):
642643 auth .generate_request_header (response , host )
643644 fake_init .assert_called_with (
644645 name = gssapi_sname (
"[email protected] " ),
645- usage = "initiate" , flags = gssflags , creds = None , mech = None )
646+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
646647 fake_resp .assert_called_with (b"token" )
647648
648649
0 commit comments