Added tests for add and delete_cluster_admins

aviau · aviau · commit 9d980a8a5620 · 2014-10-31T15:45:45.000-04:00
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -293,7 +293,25 @@ def test_get_list_cluster_admins(self):
         pass
 
     def test_add_cluster_admin(self):
-        pass
+        with requests_mock.Mocker() as m:
+            m.register_uri(
+                requests_mock.POST,
+                "http://localhost:8086/cluster_admins"
+            )
+
+            cli = InfluxDBClient(database='db')
+            cli.add_cluster_admin(
+                new_username='paul',
+                new_password='laup'
+            )
+
+            self.assertDictEqual(
+                json.loads(m.last_request.body),
+                {
+                    'name': 'paul',
+                    'password': 'laup'
+                }
+            )
 
     def test_update_cluster_admin_password(self):
         with requests_mock.Mocker() as m:
@@ -314,7 +332,17 @@ def test_update_cluster_admin_password(self):
             )
 
     def test_delete_cluster_admin(self):
-        pass
+        with requests_mock.Mocker() as m:
+            m.register_uri(
+                requests_mock.DELETE,
+                "http://localhost:8086/cluster_admins/paul",
+                status_code=204,
+            )
+
+            cli = InfluxDBClient(database='db')
+            cli.delete_cluster_admin(username='paul')
+
+            self.assertIsNone(m.last_request.body)
 
     def test_set_database_admin(self):
         pass
