26
26
load_kube_config , list_kube_config_contexts ,
27
27
new_client_from_config )
28
28
29
+ BEARER_TOKEN_FORMAT = "Bearer %s"
30
+
29
31
NON_EXISTING_FILE = "zz_non_existing_file_472398324"
30
32
31
33
@@ -407,7 +409,8 @@ def test_no_user_context(self):
407
409
self .assertEqual (expected , actual )
408
410
409
411
def test_simple_token (self ):
410
- expected = FakeConfig (host = TEST_HOST , token = TEST_DATA_BASE64 )
412
+ expected = FakeConfig (host = TEST_HOST ,
413
+ token = BEARER_TOKEN_FORMAT % TEST_DATA_BASE64 )
411
414
actual = FakeConfig ()
412
415
KubeConfigLoader (
413
416
config_dict = self .TEST_KUBE_CONFIG ,
@@ -420,10 +423,12 @@ def test_load_user_token(self):
420
423
config_dict = self .TEST_KUBE_CONFIG ,
421
424
active_context = "simple_token" )
422
425
self .assertTrue (loader ._load_user_token ())
423
- self .assertEqual (TEST_DATA_BASE64 , loader .token )
426
+ self .assertEqual (BEARER_TOKEN_FORMAT % TEST_DATA_BASE64 , loader .token )
424
427
425
428
def test_gcp (self ):
426
- expected = FakeConfig (host = TEST_HOST , token = TEST_ANOTHER_DATA_BASE64 )
429
+ expected = FakeConfig (
430
+ host = TEST_HOST ,
431
+ token = BEARER_TOKEN_FORMAT % TEST_ANOTHER_DATA_BASE64 )
427
432
actual = FakeConfig ()
428
433
KubeConfigLoader (
429
434
config_dict = self .TEST_KUBE_CONFIG ,
@@ -439,7 +444,8 @@ def test_load_gcp_token(self):
439
444
active_context = "gcp" ,
440
445
get_google_credentials = lambda : TEST_ANOTHER_DATA_BASE64 )
441
446
self .assertTrue (loader ._load_gcp_token ())
442
- self .assertEqual (TEST_ANOTHER_DATA_BASE64 , loader .token )
447
+ self .assertEqual (BEARER_TOKEN_FORMAT % TEST_ANOTHER_DATA_BASE64 ,
448
+ loader .token )
443
449
444
450
def test_user_pass (self ):
445
451
expected = FakeConfig (host = TEST_HOST , token = TEST_BASIC_TOKEN )
@@ -458,13 +464,6 @@ def test_load_user_pass_token(self):
458
464
self .assertEqual (TEST_BASIC_TOKEN , loader .token )
459
465
460
466
def test_ssl_no_cert_files (self ):
461
- expected = FakeConfig (
462
- host = TEST_SSL_HOST ,
463
- token = TEST_DATA_BASE64 ,
464
- cert_file = TEST_CLIENT_CERT ,
465
- key_file = TEST_CLIENT_KEY ,
466
- ssl_ca_cert = TEST_CERTIFICATE_AUTH
467
- )
468
467
actual = FakeConfig ()
469
468
loader = KubeConfigLoader (
470
469
config_dict = self .TEST_KUBE_CONFIG ,
@@ -475,7 +474,7 @@ def test_ssl_no_cert_files(self):
475
474
def test_ssl (self ):
476
475
expected = FakeConfig (
477
476
host = TEST_SSL_HOST ,
478
- token = TEST_DATA_BASE64 ,
477
+ token = BEARER_TOKEN_FORMAT % TEST_DATA_BASE64 ,
479
478
cert_file = self ._create_temp_file (TEST_CLIENT_CERT ),
480
479
key_file = self ._create_temp_file (TEST_CLIENT_KEY ),
481
480
ssl_ca_cert = self ._create_temp_file (TEST_CERTIFICATE_AUTH )
@@ -513,7 +512,7 @@ def test_set_active_context(self):
513
512
def test_ssl_with_relative_ssl_files (self ):
514
513
expected = FakeConfig (
515
514
host = TEST_SSL_HOST ,
516
- token = TEST_DATA_BASE64 ,
515
+ token = BEARER_TOKEN_FORMAT % TEST_DATA_BASE64 ,
517
516
cert_file = self ._create_temp_file (TEST_CLIENT_CERT ),
518
517
key_file = self ._create_temp_file (TEST_CLIENT_KEY ),
519
518
ssl_ca_cert = self ._create_temp_file (TEST_CERTIFICATE_AUTH )
@@ -539,7 +538,8 @@ def test_ssl_with_relative_ssl_files(self):
539
538
shutil .rmtree (temp_dir )
540
539
541
540
def test_load_kube_config (self ):
542
- expected = FakeConfig (host = TEST_HOST , token = TEST_DATA_BASE64 )
541
+ expected = FakeConfig (host = TEST_HOST ,
542
+ token = BEARER_TOKEN_FORMAT % TEST_DATA_BASE64 )
543
543
config_file = self ._create_temp_file (yaml .dump (self .TEST_KUBE_CONFIG ))
544
544
actual = FakeConfig ()
545
545
load_kube_config (config_file = config_file ,context = "simple_token" ,
@@ -564,7 +564,7 @@ def test_new_client_from_config(self):
564
564
client = new_client_from_config (
565
565
config_file = config_file , context = "simple_token" )
566
566
self .assertEqual (TEST_HOST , client .config .host )
567
- self .assertEqual (TEST_DATA_BASE64 ,
567
+ self .assertEqual (BEARER_TOKEN_FORMAT % TEST_DATA_BASE64 ,
568
568
client .config .api_key ['authorization' ])
569
569
570
570
0 commit comments