|
17 | 17 | from influxdb_client.client.exceptions import InfluxDBError |
18 | 18 | from influxdb_client.client.write.point import Point |
19 | 19 | from influxdb_client.client.write_api import WriteOptions, WriteApi, PointSettings |
| 20 | +from tests.base_test import BaseTest |
20 | 21 |
|
21 | 22 |
|
22 | 23 | class BatchingWriteTest(unittest.TestCase): |
@@ -736,6 +737,29 @@ def __call__(self, conf: (str, str, str), data: str, error: InfluxDBError): |
736 | 737 | self.assertIsInstance(callback.error, InfluxDBError) |
737 | 738 | self.assertEqual(429, callback.error.response.status) |
738 | 739 |
|
| 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月01日T00: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 | + |
739 | 763 |
|
740 | 764 | if __name__ == '__main__': |
741 | 765 | unittest.main() |
0 commit comments