Add test with multiple responses.

Author: michaelsafyan

instrumentation-genai/opentelemetry-instrumentation-google-genai/tests/common/otel_mocker.py

@@ -179,6 +179,16 @@ def get_event_named(self, event_name):
                 return event
         return None
 
+    def get_events_named(self, event_name):
+        result = []
+        for event in self.get_finished_logs():
+            event_name_attr = event.attributes.get("event.name")
+            if event_name_attr is None:
+                continue
+            if event_name_attr == event_name:
+                result.append(event)
+        return result
+
     def assert_has_event_named(self, name):
         event = self.get_event_named(name)
         finished_logs = self.get_finished_logs()

instrumentation-genai/opentelemetry-instrumentation-google-genai/tests/common/requests_mocker.py

@@ -141,11 +141,11 @@ def combined_generator(args):
             if response.status_code != 200:
                 continue
             response_json = response.json()
-            response_json_str = json.dumps(response_json_str)
+            response_json_str = json.dumps(response_json)
             contents.append(f'data: {response_json_str}')
         contents_str = '\r\n'.join(contents)
         full_contents = f'{contents_str}\r\n\r\n'
-        result.raw = io.StringIO(full_contents)
+        result.raw = io.BytesIO(full_contents.encode())
         return result
     return combined_generator
 

instrumentation-genai/opentelemetry-instrumentation-google-genai/tests/generate_content/streaming_base.py

@@ -51,3 +51,15 @@ def test_instrumentation_does_not_break_core_functionality(self):
         self.assertEqual(len(responses), 1)
         response = responses[0]
         self.assertEqual(response.text, "Yep, it works!")
+
+    def test_handles_multiple_ressponses(self):
+        self.configure_valid_response(response_text="First response")
+        self.configure_valid_response(response_text="Second response")
+        responses = self.generate_content(
+            model="gemini-2.0-flash", contents="Does this work?"
+        )
+        self.assertEqual(len(responses), 2)
+        self.assertEqual(responses[0].text, "First response")
+        self.assertEqual(responses[1].text, "Second response")
+        choice_events = self.otel.get_events_named("gen_ai.choice")
+        self.assertEqual(len(choice_events), 2)