From 864a87f0b4aae53ced3d171e8491e99045547b0e Mon Sep 17 00:00:00 2001
From: NguyenHoangSon96 <hoangsonnguyen4321@gmail.com>
Date: Thu, 10 Apr 2025 15:50:17 +0700
Subject: [PATCH 1/2] fix: flaky test

---
 auto_test.sh         | 17 +++++++++++++++++
 tests/test_polars.py | 15 +++++++++++----
 2 files changed, 28 insertions(+), 4 deletions(-)
 create mode 100644 auto_test.sh

diff --git a/auto_test.sh b/auto_test.sh
new file mode 100644
index 0000000..00c7a42
--- /dev/null
+++ b/auto_test.sh
@@ -0,0 +1,17 @@
+#!/bin/bash
+
+total_count=0
+error_count=0
+
+for (( i = 0; i < 100; i++ )); do
+  result=$(pytest tests/test_polars.py::TestWritePolars::test_write_polars_batching)
+  if [[ "$result" == *"expected call not found"* ]]; then
+    echo "$result"
+    ((error_count++))
+  fi
+
+  ((total_count++))
+done
+
+echo "Total run $total_count"
+echo "Error count $error_count"
diff --git a/tests/test_polars.py b/tests/test_polars.py
index b1934bd..435b4a2 100644
--- a/tests/test_polars.py
+++ b/tests/test_polars.py
@@ -1,6 +1,7 @@
 import importlib.util
+import time
 import unittest
-from unittest.mock import Mock
+from unittest.mock import Mock, ANY
 
 from influxdb_client_3 import PointSettings, InfluxDBClient3, write_client_options, WriteOptions
 from influxdb_client_3.write_client import WriteService
@@ -86,6 +87,12 @@ def test_write_polars_batching(self):
             data_frame_timestamp_column="time",
         )
 
-        actual = self.client._write_api._write_service.post_write.call_args[1]['body']
-        self.assertEqual(b'measurement temperature=22.4 1722470400000000000\n'
-                         b'measurement temperature=21.8 1722474000000000000', actual)
+        time.sleep(0.5)
+        self.client._write_api._write_service.post_write.assert_called_once_with(
+            org=ANY,
+            bucket=ANY,
+            precision=ANY,
+            async_req=ANY,
+            content_type=ANY,
+            urlopen_kw=ANY,
+            body=b'measurement temperature=22.4 1722470400000000000\nmeasurement temperature=21.8 1722474000000000000')

From 981ec4ebbbd00ab976f4d13917388626a28381c9 Mon Sep 17 00:00:00 2001
From: NguyenHoangSon96 <hoangsonnguyen4321@gmail.com>
Date: Thu, 10 Apr 2025 16:04:17 +0700
Subject: [PATCH 2/2] fix: flaky test

---
 auto_test.sh | 17 -----------------
 1 file changed, 17 deletions(-)
 delete mode 100644 auto_test.sh

diff --git a/auto_test.sh b/auto_test.sh
deleted file mode 100644
index 00c7a42..0000000
--- a/auto_test.sh
+++ /dev/null
@@ -1,17 +0,0 @@
-#!/bin/bash
-
-total_count=0
-error_count=0
-
-for (( i = 0; i < 100; i++ )); do
-  result=$(pytest tests/test_polars.py::TestWritePolars::test_write_polars_batching)
-  if [[ "$result" == *"expected call not found"* ]]; then
-    echo "$result"
-    ((error_count++))
-  fi
-
-  ((total_count++))
-done
-
-echo "Total run $total_count"
-echo "Error count $error_count"