@@ -92,26 +92,26 @@ def static():
92
92
def classm (cls ):
93
93
cls .called_classm = True
94
94
95
- @mock .patch .object (glm , "TextServiceClient " , DummyClient )
95
+ @mock .patch .object (glm , "GenerativeServiceClient " , DummyClient )
96
96
def test_default_metadata (self ):
97
97
# The metadata wrapper injects this argument.
98
98
metadata = [("hello" , "world" )]
99
99
client .configure (default_metadata = metadata )
100
100
101
- text_client = client .get_default_text_client ()
102
- text_client . generate_text ()
101
+ generative_client = client .get_default_text_client ()
102
+ generative_client . generate_content ()
103
103
104
- self .assertEqual (metadata , text_client .metadata )
104
+ self .assertEqual (metadata , generative_client .metadata )
105
105
106
- self .assertEqual (text_client .not_a_function , ClientTests .DummyClient .not_a_function )
106
+ self .assertEqual (generative_client .not_a_function , ClientTests .DummyClient .not_a_function )
107
107
108
108
# Since these don't have a metadata arg, they'll fail if the wrapper is applied.
109
- text_client ._hidden ()
110
- self .assertTrue (text_client .called_hidden )
109
+ generative_client ._hidden ()
110
+ self .assertTrue (generative_client .called_hidden )
111
111
112
- text_client .static ()
112
+ generative_client .static ()
113
113
114
- text_client .classm ()
114
+ generative_client .classm ()
115
115
self .assertTrue (ClientTests .DummyClient .called_classm )
116
116
117
117
def test_same_config (self ):
0 commit comments