@@ -483,17 +483,18 @@ def create_perf_variants():
483483def create_aws_auth_variants ():
484484 variants = []
485485
486- for host_name in ["ubuntu22" , "win64" , "macos" ]:
486+ for host_name in ["ubuntu20" , " ubuntu22" , "win64" , "macos" ]:
487487 expansions = dict ()
488- tasks = [".auth-aws" ]
489488 tags = []
490489 if host_name == "macos" :
491490 tasks = [
492491 ".auth-aws !.auth-aws-web-identity !.auth-aws-ecs !.auth-aws-ec2 !.auth-aws-eks"
493492 ]
494493 tags = ["pr" ]
495- elif host_name == "win64" :
494+ elif host_name in [ "win64" , "ubuntu20" ] :
496495 tasks = [".auth-aws !.auth-aws-ecs !.auth-aws-eks" ]
496+ elif host_name == "ubuntu22" :
497+ tasks = [".auth-aws !.server-4.4 !.server-5.0" ]
497498 host = HOSTS [host_name ]
498499 variant = create_variant (
499500 tasks ,
@@ -743,34 +744,27 @@ def create_aws_tasks():
743744 "env-creds" ,
744745 "session-creds" ,
745746 "web-identity" ,
747+ "web-identity-session-name" ,
746748 ]
747749 base_tags = ["auth-aws" ]
748750 assume_func = FunctionCall (func = "assume ec2 role" )
749751
750752 for version , test_type , python in zip_cycle (get_versions_from ("4.4" ), aws_test_types , CPYTHONS ):
751753 base_name = f"test-auth-aws-{ version } "
752- tags = [* base_tags , f"auth-aws-{ test_type } " ]
753754 name = get_task_name (f"{ base_name } -{ test_type } " , python = python )
755+ orig_test_type = test_type
756+ if test_type == "web-identity-session-name" :
757+ test_type = "web-identity" # noqa:PLW2901
758+ tags = [* base_tags , f"auth-aws-{ test_type } " ]
754759 test_vars = dict (TEST_NAME = "auth_aws" , SUB_TEST_NAME = test_type , PYTHON_VERSION = python )
760+ if orig_test_type == "web-identity-session-name" :
761+ test_vars ["AWS_ROLE_SESSION_NAME" ] = "test"
755762 test_func = FunctionCall (func = "run tests" , vars = test_vars )
756763 server_vars = dict (AUTH_AWS = "1" , VERSION = version )
757764 server_func = FunctionCall (func = "run server" , vars = server_vars )
758765 funcs = [server_func , assume_func , test_func ]
759766 tasks .append (EvgTask (name = name , tags = tags , commands = funcs ))
760767
761- if test_type == "web-identity" :
762- tags = [* base_tags , "auth-aws-web-identity" ]
763- name = get_task_name (f"{ base_name } -web-identity-session-name" , python = python )
764- test_vars = dict (
765- TEST_NAME = "auth_aws" ,
766- SUB_TEST_NAME = "web-identity" ,
767- AWS_ROLE_SESSION_NAME = "test" ,
768- PYTHON_VERSION = python ,
769- )
770- test_func = FunctionCall (func = "run tests" , vars = test_vars )
771- funcs = [server_func , assume_func , test_func ]
772- tasks .append (EvgTask (name = name , tags = tags , commands = funcs ))
773-
774768 for test_type in ["eks" , "ecs" ]:
775769 name = get_task_name (f"test-auth-aws-{ test_type } " )
776770 tags = [* base_tags , f"auth-aws-{ test_type } " ]
0 commit comments