14
14
import unittest
15
15
16
16
from requests_gssapi import REQUIRED
17
+ from requests_gssapi import SPNEGO
17
18
18
19
# Note: we're not using the @mock.patch decorator:
19
20
# > My only word of warning is that in the past, the patch decorator hides
@@ -110,7 +111,7 @@ def test_generate_request_header(self):
110
111
b64_negotiate_response )
111
112
fake_init .assert_called_with (
112
113
name = gssapi_sname (
"[email protected] " ),
113
- creds = None , mech = None , flags = gssflags , usage = "initiate" )
114
+ creds = None , mech = SPNEGO , flags = gssflags , usage = "initiate" )
114
115
fake_resp .assert_called_with (b"token" )
115
116
116
117
def test_generate_request_header_init_error (self ):
@@ -125,7 +126,7 @@ def test_generate_request_header_init_error(self):
125
126
auth .generate_request_header , response , host )
126
127
fake_init .assert_called_with (
127
128
name = gssapi_sname (
"[email protected] " ),
128
- usage = "initiate" , flags = gssflags , creds = None , mech = None )
129
+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
129
130
130
131
def test_generate_request_header_step_error (self ):
131
132
with patch .multiple ("gssapi.SecurityContext" , __init__ = fake_init ,
@@ -139,7 +140,7 @@ def test_generate_request_header_step_error(self):
139
140
auth .generate_request_header , response , host )
140
141
fake_init .assert_called_with (
141
142
name = gssapi_sname (
"[email protected] " ),
142
- usage = "initiate" , flags = gssflags , creds = None , mech = None )
143
+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
143
144
fail_resp .assert_called_with (b"token" )
144
145
145
146
def test_authenticate_user (self ):
@@ -176,7 +177,7 @@ def test_authenticate_user(self):
176
177
raw .release_conn .assert_called_with ()
177
178
fake_init .assert_called_with (
178
179
name = gssapi_sname (
"[email protected] " ),
179
- flags = gssflags , usage = "initiate" , creds = None , mech = None )
180
+ flags = gssflags , usage = "initiate" , creds = None , mech = SPNEGO )
180
181
fake_resp .assert_called_with (b"token" )
181
182
182
183
def test_handle_401 (self ):
@@ -213,7 +214,7 @@ def test_handle_401(self):
213
214
raw .release_conn .assert_called_with ()
214
215
fake_init .assert_called_with (
215
216
name = gssapi_sname (
"[email protected] " ),
216
- creds = None , mech = None , flags = gssflags , usage = "initiate" )
217
+ creds = None , mech = SPNEGO , flags = gssflags , usage = "initiate" )
217
218
fake_resp .assert_called_with (b"token" )
218
219
219
220
def test_authenticate_server (self ):
@@ -452,7 +453,7 @@ def test_handle_response_401(self):
452
453
raw .release_conn .assert_called_with ()
453
454
fake_init .assert_called_with (
454
455
name = gssapi_sname (
"[email protected] " ),
455
- usage = "initiate" , flags = gssflags , creds = None , mech = None )
456
+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
456
457
fake_resp .assert_called_with (b"token" )
457
458
458
459
def test_handle_response_401_rejected (self ):
@@ -495,7 +496,7 @@ def connection_send(self, *args, **kwargs):
495
496
raw .release_conn .assert_called_with ()
496
497
fake_init .assert_called_with (
497
498
name = gssapi_sname (
"[email protected] " ),
498
- usage = "initiate" , flags = gssflags , creds = None , mech = None )
499
+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
499
500
fake_resp .assert_called_with (b"token" )
500
501
501
502
def test_generate_request_header_custom_service (self ):
@@ -509,7 +510,7 @@ def test_generate_request_header_custom_service(self):
509
510
auth .generate_request_header (response , host ),
510
511
fake_init .assert_called_with (
511
512
name = gssapi_sname (
"[email protected] " ),
512
- usage = "initiate" , flags = gssflags , creds = None , mech = None )
513
+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
513
514
fake_resp .assert_called_with (b"token" )
514
515
515
516
def test_delegation (self ):
@@ -547,7 +548,7 @@ def test_delegation(self):
547
548
raw .release_conn .assert_called_with ()
548
549
fake_init .assert_called_with (
549
550
name = gssapi_sname (
"[email protected] " ),
550
- usage = "initiate" , flags = gssdelegflags , creds = None , mech = None )
551
+ usage = "initiate" , flags = gssdelegflags , creds = None , mech = SPNEGO )
551
552
fake_resp .assert_called_with (b"token" )
552
553
553
554
def test_principal_override (self ):
@@ -566,7 +567,7 @@ def test_principal_override(self):
566
567
fake_init .assert_called_with (
567
568
name = gssapi_sname (
"[email protected] " ),
568
569
usage = "initiate" , flags = gssflags ,
569
- creds = b"fake creds" , mech = None )
570
+ creds = b"fake creds" , mech = SPNEGO )
570
571
571
572
def test_realm_override (self ):
572
573
with patch .multiple ("gssapi.SecurityContext" , __init__ = fake_init ,
@@ -580,7 +581,7 @@ def test_realm_override(self):
580
581
auth .generate_request_header (response , host )
581
582
fake_init .assert_called_with (
582
583
name = gssapi_sname (
"[email protected] " ),
583
- usage = "initiate" , flags = gssflags , creds = None , mech = None )
584
+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
584
585
fake_resp .assert_called_with (b"token" )
585
586
586
587
def test_opportunistic_auth (self ):
@@ -610,7 +611,7 @@ def test_explicit_creds(self):
610
611
fake_init .assert_called_with (
611
612
name = gssapi_sname (
"[email protected] " ),
612
613
usage = "initiate" , flags = gssflags ,
613
- creds = b"fake creds" , mech = None )
614
+ creds = b"fake creds" , mech = SPNEGO )
614
615
fake_resp .assert_called_with (b"token" )
615
616
616
617
def test_explicit_mech (self ):
@@ -642,7 +643,7 @@ def test_target_name(self):
642
643
auth .generate_request_header (response , host )
643
644
fake_init .assert_called_with (
644
645
name = gssapi_sname (
"[email protected] " ),
645
- usage = "initiate" , flags = gssflags , creds = None , mech = None )
646
+ usage = "initiate" , flags = gssflags , creds = None , mech = SPNEGO )
646
647
fake_resp .assert_called_with (b"token" )
647
648
648
649
0 commit comments