Fix emitter unit tests

miermans · Jul 5, 2022 · 2431abb · 2431abb
1 parent 9f876a9
commit 2431abb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/aio_snowplow_tracker/test/unit/test_emitters.py b/aio_snowplow_tracker/test/unit/test_emitters.py
@@ -126,7 +126,7 @@ async def test_input_no_flush(self, mok_flush: Any) -> None:
         self.assertFalse(e.reached_limit())
         mok_flush.assert_not_called()
 
-    @async_patch('aio_snowplow_tracker.Emitter.flush')
+    @async_patch('aio_snowplow_tracker.Emitter._flush_unsafe')
     async def test_input_flush_byte_limit(self, mok_flush: Any) -> None:
         mok_flush.side_effect = mocked_flush
 
@@ -139,7 +139,7 @@ async def test_input_flush_byte_limit(self, mok_flush: Any) -> None:
         self.assertTrue(e.reached_limit())
         self.assertEqual(mok_flush.call_count, 1)
 
-    @async_patch('aio_snowplow_tracker.Emitter.flush')
+    @async_patch('aio_snowplow_tracker.Emitter._flush_unsafe')
     async def test_input_flush_buffer(self, mok_flush: Any) -> None:
         mok_flush.side_effect = mocked_flush
 
@@ -158,7 +158,7 @@ async def test_input_flush_buffer(self, mok_flush: Any) -> None:
         self.assertTrue(e.reached_limit())
         self.assertEqual(mok_flush.call_count, 1)
 
-    @async_patch('aio_snowplow_tracker.Emitter.flush')
+    @async_patch('aio_snowplow_tracker.Emitter._flush_unsafe')
     async def test_input_bytes_queued(self, mok_flush: Any) -> None:
         mok_flush.side_effect = mocked_flush
 
@@ -172,7 +172,7 @@ async def test_input_bytes_queued(self, mok_flush: Any) -> None:
         await e.input(nvPairs)
         self.assertEqual(e.bytes_queued, 48)
 
-    @async_patch('aio_snowplow_tracker.Emitter.flush')
+    @async_patch('aio_snowplow_tracker.Emitter._flush_unsafe')
     async def test_input_bytes_post(self, mok_flush: Any) -> None:
         mok_flush.side_effect = mocked_flush