@@ -112,10 +112,13 @@ def create_free_threaded_variants() -> list[BuildVariant]:
112112 # TODO: PYTHON-5027
113113 continue
114114 tasks = [".free-threading" ]
115+ tags = []
116+ if host_name == "rhel8" :
117+ tags .append ("pr" )
115118 host = HOSTS [host_name ]
116119 python = "3.13t"
117120 display_name = get_variant_name ("Free-threaded" , host , python = python )
118- variant = create_variant (tasks , display_name , python = python , host = host )
121+ variant = create_variant (tasks , display_name , tags = tags , python = python , host = host )
119122 variants .append (variant )
120123 return variants
121124
@@ -329,7 +332,7 @@ def create_atlas_data_lake_variants():
329332 tasks = [".test-no-orchestration" ]
330333 expansions = dict (TEST_NAME = "data_lake" )
331334 display_name = get_variant_name ("Atlas Data Lake" , host )
332- return [create_variant (tasks , display_name , host = host , expansions = expansions )]
335+ return [create_variant (tasks , display_name , tags = [ "pr" ], host = host , expansions = expansions )]
333336
334337
335338def create_mod_wsgi_variants ():
@@ -370,9 +373,9 @@ def create_oidc_auth_variants():
370373 variants = []
371374 for host_name in ["ubuntu22" , "macos" , "win64" ]:
372375 if host_name == "ubuntu22" :
373- tasks = [".auth_oidc " ]
376+ tasks = [".auth_oidc_remote " ]
374377 else :
375- tasks = [".auth_oidc !.auth_oidc_remote" ]
378+ tasks = ["!.auth_oidc_remote .auth_oidc " ]
376379 host = HOSTS [host_name ]
377380 variants .append (
378381 create_variant (
@@ -382,6 +385,18 @@ def create_oidc_auth_variants():
382385 batchtime = BATCHTIME_WEEK ,
383386 )
384387 )
388+ # Add a specific local test to run on PRs.
389+ if host_name == "ubuntu22" :
390+ tasks = ["!.auth_oidc_remote .auth_oidc" ]
391+ variants .append (
392+ create_variant (
393+ tasks ,
394+ get_variant_name ("Auth OIDC Local" , host ),
395+ tags = ["pr" ],
396+ host = host ,
397+ batchtime = BATCHTIME_WEEK ,
398+ )
399+ )
385400 return variants
386401
387402
@@ -406,6 +421,7 @@ def create_mockupdb_variants():
406421 [".test-no-orchestration" ],
407422 get_variant_name ("MockupDB" , host ),
408423 host = host ,
424+ tags = ["pr" ],
409425 expansions = expansions ,
410426 )
411427 ]
@@ -430,6 +446,7 @@ def create_atlas_connect_variants():
430446 create_variant (
431447 [".test-no-orchestration" ],
432448 get_variant_name ("Atlas connect" , host ),
449+ tags = ["pr" ],
433450 host = DEFAULT_HOST ,
434451 )
435452 ]
@@ -469,15 +486,18 @@ def create_aws_auth_variants():
469486 for host_name in ["ubuntu20" , "win64" , "macos" ]:
470487 expansions = dict ()
471488 tasks = [".auth-aws" ]
489+ tags = []
472490 if host_name == "macos" :
473491 tasks = [".auth-aws !.auth-aws-web-identity !.auth-aws-ecs !.auth-aws-ec2" ]
492+ tags = ["pr" ]
474493 elif host_name == "win64" :
475494 tasks = [".auth-aws !.auth-aws-ecs" ]
476495 host = HOSTS [host_name ]
477496 variant = create_variant (
478497 tasks ,
479498 get_variant_name ("Auth AWS" , host ),
480499 host = host ,
500+ tags = tags ,
481501 expansions = expansions ,
482502 )
483503 variants .append (variant )
@@ -487,7 +507,7 @@ def create_aws_auth_variants():
487507def create_no_server_variants ():
488508 host = HOSTS ["rhel8" ]
489509 name = get_variant_name ("No server" , host = host )
490- return [create_variant ([".test-no-orchestration" ], name , host = host )]
510+ return [create_variant ([".test-no-orchestration" ], name , host = host , tags = [ "pr" ] )]
491511
492512
493513def create_alternative_hosts_variants ():
@@ -512,14 +532,18 @@ def create_alternative_hosts_variants():
512532 expansions = dict (VERSION = "latest" )
513533 handle_c_ext (C_EXTS [0 ], expansions )
514534 host = HOSTS [host_name ]
535+ tags = []
515536 if "fips" in host_name .lower ():
516537 expansions ["REQUIRE_FIPS" ] = "1"
538+ if "amazon" in host_name .lower ():
539+ tags .append ("pr" )
517540 variants .append (
518541 create_variant (
519542 [".test-no-toolchain" ],
520543 display_name = get_variant_name ("Other hosts" , host , version = version ),
521544 batchtime = batchtime ,
522545 host = host ,
546+ tags = tags ,
523547 expansions = expansions ,
524548 )
525549 )
@@ -693,16 +717,18 @@ def create_kms_tasks():
693717 for success in [True , False ]:
694718 name = f"test-{ kms_type } kms"
695719 sub_test_name = kms_type
720+ tags = []
696721 if not success :
697722 name += "-fail"
698723 sub_test_name += "-fail"
724+ tags .append ("pr" )
699725 commands = []
700726 if not success :
701727 commands .append (FunctionCall (func = "run server" ))
702728 test_vars = dict (TEST_NAME = "kms" , SUB_TEST_NAME = sub_test_name )
703729 test_func = FunctionCall (func = "run tests" , vars = test_vars )
704730 commands .append (test_func )
705- tasks .append (EvgTask (name = name , commands = commands ))
731+ tasks .append (EvgTask (name = name , tags = tags , commands = commands ))
706732 return tasks
707733
708734
@@ -756,6 +782,7 @@ def create_oidc_tasks():
756782 if sub_test != "default" :
757783 tags .append ("auth_oidc_remote" )
758784 tasks .append (EvgTask (name = task_name , tags = tags , commands = [test_func ]))
785+
759786 return tasks
760787
761788
@@ -802,6 +829,8 @@ def _create_ocsp_tasks(algo, variant, server_type, base_task_name):
802829 tags = ["ocsp" , f"ocsp-{ algo } " , version ]
803830 if "disableStapling" not in variant :
804831 tags .append ("ocsp-staple" )
832+ if algo == "valid-cert-server-staples" and version == "latest" :
833+ tags .append ("pr" )
805834
806835 task_name = get_task_name (
807836 f"test-ocsp-{ algo } -{ base_task_name } " , python = python , version = version
0 commit comments