Skip to content

Commit 310143e

Browse files
Remove reference from test case for client
1 parent 8312c9f commit 310143e

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

tests/test_client.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -92,26 +92,26 @@ def static():
9292
def classm(cls):
9393
cls.called_classm = True
9494

95-
@mock.patch.object(glm, "TextServiceClient", DummyClient)
95+
@mock.patch.object(glm, "GenerativeServiceClient", DummyClient)
9696
def test_default_metadata(self):
9797
# The metadata wrapper injects this argument.
9898
metadata = [("hello", "world")]
9999
client.configure(default_metadata=metadata)
100100

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()
103103

104-
self.assertEqual(metadata, text_client.metadata)
104+
self.assertEqual(metadata, generative_client.metadata)
105105

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)
107107

108108
# 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)
111111

112-
text_client.static()
112+
generative_client.static()
113113

114-
text_client.classm()
114+
generative_client.classm()
115115
self.assertTrue(ClientTests.DummyClient.called_classm)
116116

117117
def test_same_config(self):

0 commit comments

Comments
 (0)