Add tests for get_list_privileges()

wwentland · wwentland · commit 3742a0ec68b0 · 2016-06-13T19:34:48.000+02:00
These test check for correct return value and exception handling in case
of incorrect usage.
diff --git a/influxdb/tests/client_test.py b/influxdb/tests/client_test.py
@@ -776,6 +776,31 @@ def test_revoke_privilege_invalid(self):
         with _mocked_session(cli, 'get', 400):
             self.cli.revoke_privilege('', 'testdb', 'test')
 
+    def test_get_list_privileges(self):
+        data = {'results': [
+            {'series': [
+                {'columns': ['database', 'privilege'],
+                 'values': [
+                     ['db1', 'READ'],
+                     ['db2', 'ALL PRIVILEGES'],
+                     ['db3', 'NO PRIVILEGES']]}
+            ]}
+        ]}
+
+        with _mocked_session(self.cli, 'get', 200, json.dumps(data)):
+            self.assertListEqual(
+                self.cli.get_list_privileges('test'),
+                [{'database': 'db1', 'privilege': 'READ'},
+                 {'database': 'db2', 'privilege': 'ALL PRIVILEGES'},
+                 {'database': 'db3', 'privilege': 'NO PRIVILEGES'}]
+            )
+
+    @raises(Exception)
+    def test_get_list_privileges_fails(self):
+        cli = InfluxDBClient('host', 8086, 'username', 'password')
+        with _mocked_session(cli, 'get', 401):
+            cli.get_list_privileges('test')
+
     def test_invalid_port_fails(self):
         with self.assertRaises(ValueError):
             InfluxDBClient('host', '80/redir', 'username', 'password')
