39
39
b64_negotiate_server = "negotiate " + b64encode (b"servertoken" ).decode ()
40
40
41
41
42
- def gssapi_name (s ):
42
+ def gssapi_sname (s ):
43
43
return gssapi .Name (s , gssapi .NameType .hostbased_service )
44
44
45
45
46
+ def gssapi_uname (s ):
47
+ return gssapi .Name (s , gssapi .NameType .user )
48
+
49
+
46
50
class GSSAPITestCase (unittest .TestCase ):
47
51
def setUp (self ):
48
52
"""Setup."""
@@ -105,7 +109,7 @@ def test_generate_request_header(self):
105
109
auth .generate_request_header (response , host ),
106
110
b64_negotiate_response )
107
111
fake_init .assert_called_with (
108
- name = gssapi_name (
"[email protected] " ),
112
+ name = gssapi_sname (
"[email protected] " ),
109
113
creds = None , mech = None , flags = gssflags , usage = "initiate" )
110
114
fake_resp .assert_called_with (b"token" )
111
115
@@ -120,7 +124,7 @@ def test_generate_request_header_init_error(self):
120
124
self .assertRaises (requests_gssapi .exceptions .SPNEGOExchangeError ,
121
125
auth .generate_request_header , response , host )
122
126
fake_init .assert_called_with (
123
- name = gssapi_name (
"[email protected] " ),
127
+ name = gssapi_sname (
"[email protected] " ),
124
128
usage = "initiate" , flags = gssflags , creds = None , mech = None )
125
129
126
130
def test_generate_request_header_step_error (self ):
@@ -134,7 +138,7 @@ def test_generate_request_header_step_error(self):
134
138
self .assertRaises (requests_gssapi .exceptions .SPNEGOExchangeError ,
135
139
auth .generate_request_header , response , host )
136
140
fake_init .assert_called_with (
137
- name = gssapi_name (
"[email protected] " ),
141
+ name = gssapi_sname (
"[email protected] " ),
138
142
usage = "initiate" , flags = gssflags , creds = None , mech = None )
139
143
fail_resp .assert_called_with (b"token" )
140
144
@@ -171,7 +175,7 @@ def test_authenticate_user(self):
171
175
connection .send .assert_called_with (request )
172
176
raw .release_conn .assert_called_with ()
173
177
fake_init .assert_called_with (
174
- name = gssapi_name (
"[email protected] " ),
178
+ name = gssapi_sname (
"[email protected] " ),
175
179
flags = gssflags , usage = "initiate" , creds = None , mech = None )
176
180
fake_resp .assert_called_with (b"token" )
177
181
@@ -208,7 +212,7 @@ def test_handle_401(self):
208
212
connection .send .assert_called_with (request )
209
213
raw .release_conn .assert_called_with ()
210
214
fake_init .assert_called_with (
211
- name = gssapi_name (
"[email protected] " ),
215
+ name = gssapi_sname (
"[email protected] " ),
212
216
creds = None , mech = None , flags = gssflags , usage = "initiate" )
213
217
fake_resp .assert_called_with (b"token" )
214
218
@@ -447,7 +451,7 @@ def test_handle_response_401(self):
447
451
connection .send .assert_called_with (request )
448
452
raw .release_conn .assert_called_with ()
449
453
fake_init .assert_called_with (
450
- name = gssapi_name (
"[email protected] " ),
454
+ name = gssapi_sname (
"[email protected] " ),
451
455
usage = "initiate" , flags = gssflags , creds = None , mech = None )
452
456
fake_resp .assert_called_with (b"token" )
453
457
@@ -490,7 +494,7 @@ def connection_send(self, *args, **kwargs):
490
494
connection .send .assert_called_with (request )
491
495
raw .release_conn .assert_called_with ()
492
496
fake_init .assert_called_with (
493
- name = gssapi_name (
"[email protected] " ),
497
+ name = gssapi_sname (
"[email protected] " ),
494
498
usage = "initiate" , flags = gssflags , creds = None , mech = None )
495
499
fake_resp .assert_called_with (b"token" )
496
500
@@ -504,7 +508,7 @@ def test_generate_request_header_custom_service(self):
504
508
auth = requests_gssapi .HTTPKerberosAuth (service = "barfoo" )
505
509
auth .generate_request_header (response , host ),
506
510
fake_init .assert_called_with (
507
- name = gssapi_name (
"[email protected] " ),
511
+ name = gssapi_sname (
"[email protected] " ),
508
512
usage = "initiate" , flags = gssflags , creds = None , mech = None )
509
513
fake_resp .assert_called_with (b"token" )
510
514
@@ -542,7 +546,7 @@ def test_delegation(self):
542
546
connection .send .assert_called_with (request )
543
547
raw .release_conn .assert_called_with ()
544
548
fake_init .assert_called_with (
545
- name = gssapi_name (
"[email protected] " ),
549
+ name = gssapi_sname (
"[email protected] " ),
546
550
usage = "initiate" , flags = gssdelegflags , creds = None , mech = None )
547
551
fake_resp .assert_called_with (b"token" )
548
552
@@ -558,9 +562,9 @@ def test_principal_override(self):
558
562
auth .generate_request_header (response , host )
559
563
fake_creds .assert_called_with (gssapi .creds .Credentials ,
560
564
usage = "initiate" ,
561
- name = gssapi_name ("user@REALM" ))
565
+ name = gssapi_uname ("user@REALM" , ))
562
566
fake_init .assert_called_with (
563
- name = gssapi_name (
"[email protected] " ),
567
+ name = gssapi_sname (
"[email protected] " ),
564
568
usage = "initiate" , flags = gssflags ,
565
569
creds = b"fake creds" , mech = None )
566
570
@@ -575,7 +579,7 @@ def test_realm_override(self):
575
579
hostname_override = "otherhost.otherdomain.org" )
576
580
auth .generate_request_header (response , host )
577
581
fake_init .assert_called_with (
578
- name = gssapi_name (
"[email protected] " ),
582
+ name = gssapi_sname (
"[email protected] " ),
579
583
usage = "initiate" , flags = gssflags , creds = None , mech = None )
580
584
fake_resp .assert_called_with (b"token" )
581
585
@@ -604,7 +608,7 @@ def test_explicit_creds(self):
604
608
auth = requests_gssapi .HTTPSPNEGOAuth (creds = creds )
605
609
auth .generate_request_header (response , host )
606
610
fake_init .assert_called_with (
607
- name = gssapi_name (
"[email protected] " ),
611
+ name = gssapi_sname (
"[email protected] " ),
608
612
usage = "initiate" , flags = gssflags ,
609
613
creds = b"fake creds" , mech = None )
610
614
fake_resp .assert_called_with (b"token" )
@@ -621,7 +625,7 @@ def test_explicit_mech(self):
621
625
auth = requests_gssapi .HTTPSPNEGOAuth (mech = fake_mech )
622
626
auth .generate_request_header (response , host )
623
627
fake_init .assert_called_with (
624
- name = gssapi_name (
"[email protected] " ),
628
+ name = gssapi_sname (
"[email protected] " ),
625
629
usage = "initiate" , flags = gssflags ,
626
630
creds = None , mech = b'fake mech' )
627
631
fake_resp .assert_called_with (b"token" )
@@ -637,7 +641,7 @@ def test_target_name(self):
637
641
638
642
auth .generate_request_header (response , host )
639
643
fake_init .assert_called_with (
640
- name = gssapi_name (
"[email protected] " ),
644
+ name = gssapi_sname (
"[email protected] " ),
641
645
usage = "initiate" , flags = gssflags , creds = None , mech = None )
642
646
fake_resp .assert_called_with (b"token" )
643
647
0 commit comments