@@ -743,19 +743,18 @@ def create_aws_tasks():
743743 "env-creds" ,
744744 "session-creds" ,
745745 "web-identity" ,
746- "ecs" ,
747- "eks" ,
748746 ]
747+ base_tags = ["auth-aws" ]
748+ assume_func = FunctionCall (func = "assume ec2 role" )
749+
749750 for version , test_type , python in zip_cycle (get_versions_from ("4.4" ), aws_test_types , CPYTHONS ):
750751 base_name = f"test-auth-aws-{ version } "
751- base_tags = ["auth-aws" ]
752- server_vars = dict (AUTH_AWS = "1" , VERSION = version )
753- server_func = FunctionCall (func = "run server" , vars = server_vars )
754- assume_func = FunctionCall (func = "assume ec2 role" )
755752 tags = [* base_tags , f"auth-aws-{ test_type } " ]
756753 name = get_task_name (f"{ base_name } -{ test_type } " , python = python )
757754 test_vars = dict (TEST_NAME = "auth_aws" , SUB_TEST_NAME = test_type , PYTHON_VERSION = python )
758755 test_func = FunctionCall (func = "run tests" , vars = test_vars )
756+ server_vars = dict (AUTH_AWS = "1" , VERSION = version )
757+ server_func = FunctionCall (func = "run server" , vars = server_vars )
759758 funcs = [server_func , assume_func , test_func ]
760759 tasks .append (EvgTask (name = name , tags = tags , commands = funcs ))
761760
@@ -772,6 +771,14 @@ def create_aws_tasks():
772771 funcs = [server_func , assume_func , test_func ]
773772 tasks .append (EvgTask (name = name , tags = tags , commands = funcs ))
774773
774+ for test_type in ["eks" , "ecs" ]:
775+ name = get_task_name (f"test-auth-aws-{ test_type } " )
776+ tags = [* base_tags , f"auth-aws-{ test_type } " ]
777+ test_vars = dict (TEST_NAME = "auth_aws" , SUB_TEST_NAME = test_type )
778+ test_func = FunctionCall (func = "run tests" , vars = test_vars )
779+ funcs = [assume_func , test_func ]
780+ tasks .append (EvgTask (name = name , tags = tags , commands = funcs ))
781+
775782 return tasks
776783
777784
0 commit comments