chore(client_test): adding in an old test from legacy PR#315 (influxdata#752)

sebito91 · web-flow · commit 6baf7eee61e9 · 2019-09-04T08:24:19.000-05:00
* chore(client_test): adding in an old test from legacy PR#315

* chore(client_test): update from GET to POST

* chore(client_test): update to clear out flake8 issues
diff --git a/influxdb/tests/client_test.py b/influxdb/tests/client_test.py
@@ -706,6 +706,49 @@ def test_create_retention_policy(self):
                 '"db" duration 1d replication 4 shard duration 0s'
             )
 
+    def test_create_retention_policy_shard_duration(self):
+        """Test create retention policy with a custom shard duration."""
+        example_response = '{"results":[{}]}'
+
+        with requests_mock.Mocker() as m:
+            m.register_uri(
+                requests_mock.POST,
+                "http://localhost:8086/query",
+                text=example_response
+            )
+            self.cli.create_retention_policy(
+                'somename2', '1d', 4, database='db',
+                shard_duration='1h'
+            )
+
+            self.assertEqual(
+                m.last_request.qs['q'][0],
+                'create retention policy "somename2" on '
+                '"db" duration 1d replication 4 shard duration 1h'
+            )
+
+    def test_create_retention_policy_shard_duration_default(self):
+        """Test create retention policy with a default shard duration."""
+        example_response = '{"results":[{}]}'
+
+        with requests_mock.Mocker() as m:
+            m.register_uri(
+                requests_mock.POST,
+                "http://localhost:8086/query",
+                text=example_response
+            )
+            self.cli.create_retention_policy(
+                'somename3', '1d', 4, database='db',
+                shard_duration='1h', default=True
+            )
+
+            self.assertEqual(
+                m.last_request.qs['q'][0],
+                'create retention policy "somename3" on '
+                '"db" duration 1d replication 4 shard duration 1h '
+                'default'
+            )
+
     def test_alter_retention_policy(self):
         """Test alter retention policy for TestInfluxDBClient object."""
         example_response = '{"results":[{}]}'
