Skip to content

Commit 7139fd2

Browse files
minsungohbednar
authored andcommitted
feat: flush test on batch mode
1 parent 390f190 commit 7139fd2

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

tests/test_WriteApiBatching.py

+24
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
from influxdb_client.client.exceptions import InfluxDBError
1818
from influxdb_client.client.write.point import Point
1919
from influxdb_client.client.write_api import WriteOptions, WriteApi, PointSettings
20+
from tests.base_test import BaseTest
2021

2122

2223
class BatchingWriteTest(unittest.TestCase):
@@ -736,6 +737,29 @@ def __call__(self, conf: (str, str, str), data: str, error: InfluxDBError):
736737
self.assertIsInstance(callback.error, InfluxDBError)
737738
self.assertEqual(429, callback.error.response.status)
738739

740+
class BatchingWriteFlushTest(BaseTest):
741+
742+
def setUp(self):
743+
return super().setUp()
744+
745+
def test_flush(self):
746+
write_client = self.client.write_api()
747+
748+
bucket = self.create_test_bucket()
749+
750+
write_client.write(bucket.name, self.org, "h2o_feet,location=coyote_creek level\\ water_level=1 1")
751+
752+
write_client.flush()
753+
754+
time.sleep(1)
755+
756+
query = 'from(bucket:"' + bucket.name + '") |> range(start: 1970-01-01T00:00:00.000000001Z)'
757+
flux_result = self.client.query_api().query(query)
758+
759+
self.assertEqual(1, len(flux_result))
760+
761+
self.delete_test_bucket(bucket)
762+
739763

740764
if __name__ == '__main__':
741765
unittest.main()

0 commit comments

Comments
 (0)