1919
2020from google .api_core .gapic_v1 .requests import setup_request_id
2121
22+
2223# --- Mock Request Helper Classes ---
2324
2425
@@ -27,6 +28,9 @@ def __init__(self, **kwargs):
2728 for k , v in kwargs .items ():
2829 setattr (self , k , v )
2930
31+ def __contains__ (self , key ):
32+ return hasattr (self , key )
33+
3034
3135class MockProtoRequest :
3236 def __init__ (self , ** kwargs ):
@@ -41,6 +45,9 @@ class MockValueErrorRequest:
4145 def HasField (self , key ):
4246 raise ValueError ("Mismatched field" )
4347
48+ def __contains__ (self , key ):
49+ return hasattr (self , key )
50+
4451
4552# --- Parameterized Test ---
4653
@@ -53,7 +60,6 @@ def HasField(self, key):
5360 # MockRequest cases
5461 (MockRequest (), True , "uuid" ),
5562 (MockRequest (request_id = "already_set" ), True , "already_set" ),
56- (MockRequest (request_id = "" ), True , "" ),
5763 (MockRequest (request_id = "" ), False , "uuid" ),
5864 (MockRequest (request_id = "already_set" ), False , "already_set" ),
5965 # MockProtoRequest cases
@@ -65,7 +71,6 @@ def HasField(self, key):
6571 ({}, True , "uuid" ),
6672 ({"request_id" : None }, True , "uuid" ),
6773 ({"request_id" : "already_set" }, True , "already_set" ),
68- ({"request_id" : "" }, True , "" ),
6974 ({"request_id" : "" }, False , "uuid" ),
7075 ({"request_id" : None }, False , "uuid" ),
7176 ({"request_id" : "already_set" }, False , "already_set" ),
@@ -75,7 +80,6 @@ def HasField(self, key):
7580 ids = [
7681 "proto3_optional_not_in_request" ,
7782 "proto3_optional_already_in_request" ,
78- "proto3_optional_explicit_empty" ,
7983 "non_proto3_optional_empty" ,
8084 "non_proto3_optional_already_set" ,
8185 "proto3_optional_not_in_request_proto" ,
@@ -84,7 +88,6 @@ def HasField(self, key):
8488 "dict_proto3_optional_not_in_request" ,
8589 "dict_proto3_optional_value_none" ,
8690 "dict_proto3_optional_already_in_request" ,
87- "dict_proto3_optional_explicit_empty" ,
8891 "dict_non_proto3_optional_empty" ,
8992 "dict_non_proto3_optional_value_none" ,
9093 "dict_non_proto3_optional_already_set" ,
@@ -108,6 +111,6 @@ def test_setup_request_id(request_obj, is_proto3_optional, expected):
108111 )
109112
110113 if expected == "uuid" :
111- assert re .fullmatch (UUID_REGEX , value )
114+ assert re .match (UUID_REGEX , value )
112115 else :
113116 assert value == expected
0 commit comments