fix(tests): new version of InfluxDB allows empty writes - influxdata/…

…influxdb#22575
influxdata · Oct 4, 2021 · a0779ab · a0779ab
1 parent 96d99de
commit a0779ab
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/tests/test_WriteApi.py b/tests/test_WriteApi.py
@@ -338,12 +338,15 @@ def test_use_default_org(self):
     def test_write_empty_data(self):
         bucket = self.create_test_bucket()
 
-        with self.assertRaises(ApiException) as cm:
-            self.write_client.write(bucket.name, record="")
-        exception = cm.exception
-
-        self.assertEqual(400, exception.status)
-        self.assertEqual("Bad Request", exception.reason)
+        from distutils.version import LooseVersion
+        version = self.client.health().version
+        if version != 'nightly' and LooseVersion(version) <= LooseVersion("2.0.8"):
+            with self.assertRaises(ApiException) as cm:
+                self.write_client.write(bucket.name, record="")
+            exception = cm.exception
+
+            self.assertEqual(400, exception.status)
+            self.assertEqual("Bad Request", exception.reason)
 
         result = self.query_api.query(
             "from(bucket:\"" + bucket.name + "\") |> range(start: 1970-01-01T00:00:00.000000001Z) |> last()", self.org)