Add sent_report_count to AsyncIntegrationTest

bugsnag · Aug 11, 2023 · 90b9ee7 · 90b9ee7
1 parent 48c287a
commit 90b9ee7
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/tests/async_utils.py b/tests/async_utils.py
@@ -11,10 +11,13 @@
 class AsyncIntegrationTest(IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
         self.server = FakeBugsnag()
-        bugsnag.configure(asynchronous=True,
-                          endpoint=self.server.events_url,
-                          session_endpoint=self.server.sessions_url,
-                          api_key='ffffffffffffffffff')
+        self.sent_report_count = 0
+        bugsnag.configure(
+            asynchronous=True,
+            endpoint=self.server.events_url,
+            session_endpoint=self.server.sessions_url,
+            api_key='ffffffffffffffffff'
+        )
 
     async def asyncTearDown(self):
         bugsnag.legacy.default_client.uninstall_sys_hook()
@@ -29,7 +32,9 @@ async def poll():
             while len(self.server.events_received) == 0:
                 await asyncio.sleep(0.1)
 
+            self.sent_report_count += 1
             return self.server.events_received.pop()
+
         try:
             return await asyncio.wait_for(poll(), timeout=1.0)
         except asyncio.TimeoutError: