@@ -501,14 +501,34 @@ def consensus_project_with_batch(
501
501
502
502
@pytest .fixture
503
503
def dataset (client , rand_gen ):
504
- dataset = client .create_dataset (name = rand_gen (str ))
504
+ # Handle invalid default IAM integrations in test environments gracefully
505
+ try :
506
+ dataset = client .create_dataset (name = rand_gen (str ))
507
+ except ValueError as e :
508
+ if "Integration is not valid" in str (e ):
509
+ # Fallback to creating dataset without IAM integration for tests
510
+ dataset = client .create_dataset (
511
+ name = rand_gen (str ), iam_integration = None
512
+ )
513
+ else :
514
+ raise e
505
515
yield dataset
506
516
dataset .delete ()
507
517
508
518
509
519
@pytest .fixture (scope = "function" )
510
520
def unique_dataset (client , rand_gen ):
511
- dataset = client .create_dataset (name = rand_gen (str ))
521
+ # Handle invalid default IAM integrations in test environments gracefully
522
+ try :
523
+ dataset = client .create_dataset (name = rand_gen (str ))
524
+ except ValueError as e :
525
+ if "Integration is not valid" in str (e ):
526
+ # Fallback to creating dataset without IAM integration for tests
527
+ dataset = client .create_dataset (
528
+ name = rand_gen (str ), iam_integration = None
529
+ )
530
+ else :
531
+ raise e
512
532
yield dataset
513
533
dataset .delete ()
514
534
@@ -857,15 +877,35 @@ def func(project):
857
877
858
878
@pytest .fixture
859
879
def initial_dataset (client , rand_gen ):
860
- dataset = client .create_dataset (name = rand_gen (str ))
880
+ # Handle invalid default IAM integrations in test environments gracefully
881
+ try :
882
+ dataset = client .create_dataset (name = rand_gen (str ))
883
+ except ValueError as e :
884
+ if "Integration is not valid" in str (e ):
885
+ # Fallback to creating dataset without IAM integration for tests
886
+ dataset = client .create_dataset (
887
+ name = rand_gen (str ), iam_integration = None
888
+ )
889
+ else :
890
+ raise e
861
891
yield dataset
862
892
863
893
dataset .delete ()
864
894
865
895
866
896
@pytest .fixture
867
897
def video_data (client , rand_gen , video_data_row , wait_for_data_row_processing ):
868
- dataset = client .create_dataset (name = rand_gen (str ))
898
+ # Handle invalid default IAM integrations in test environments gracefully
899
+ try :
900
+ dataset = client .create_dataset (name = rand_gen (str ))
901
+ except ValueError as e :
902
+ if "Integration is not valid" in str (e ):
903
+ # Fallback to creating dataset without IAM integration for tests
904
+ dataset = client .create_dataset (
905
+ name = rand_gen (str ), iam_integration = None
906
+ )
907
+ else :
908
+ raise e
869
909
data_row_ids = []
870
910
data_row = dataset .create_data_row (video_data_row )
871
911
data_row = wait_for_data_row_processing (client , data_row )
@@ -884,7 +924,17 @@ def create_video_data_row(rand_gen):
884
924
885
925
@pytest .fixture
886
926
def video_data_100_rows (client , rand_gen , wait_for_data_row_processing ):
887
- dataset = client .create_dataset (name = rand_gen (str ))
927
+ # Handle invalid default IAM integrations in test environments gracefully
928
+ try :
929
+ dataset = client .create_dataset (name = rand_gen (str ))
930
+ except ValueError as e :
931
+ if "Integration is not valid" in str (e ):
932
+ # Fallback to creating dataset without IAM integration for tests
933
+ dataset = client .create_dataset (
934
+ name = rand_gen (str ), iam_integration = None
935
+ )
936
+ else :
937
+ raise e
888
938
data_row_ids = []
889
939
for _ in range (100 ):
890
940
data_row = dataset .create_data_row (create_video_data_row (rand_gen ))
0 commit comments