Add tests for grant_admin_privileges()

These tests simply check for correct query generation and exception handling if used incorrectly by passing in an empty string as username.

Author: wwentland

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":[{}]}'