Add tests for grant_admin_privileges()

wwentland · wwentland · commit df4ee82642b8 · 2016-06-13T19:34:26.000+02:00
These tests simply check for correct query generation and exception
handling if used incorrectly by passing in an empty string as username.
diff --git a/influxdb/tests/client_test.py b/influxdb/tests/client_test.py
@@ -710,6 +710,28 @@ def test_get_list_users_empty(self):
 
             self.assertListEqual(self.cli.get_list_users(), [])
 
+    def test_grant_admin_privileges(self):
+        example_response = '{"results":[{}]}'
+
+        with requests_mock.Mocker() as m:
+            m.register_uri(
+                requests_mock.GET,
+                "http://localhost:8086/query",
+                text=example_response
+            )
+            self.cli.grant_admin_privileges('test')
+
+            self.assertEqual(
+                m.last_request.qs['q'][0],
+                'grant all privileges to test'
+            )
+
+    @raises(Exception)
+    def test_grant_admin_privileges_invalid(self):
+        cli = InfluxDBClient('host', 8086, 'username', 'password')
+        with _mocked_session(cli, 'get', 400):
+            self.cli.grant_admin_privileges('')
+
     def test_revoke_admin_privileges(self):
         example_response = '{"results":[{}]}'
 
