Add tests for get_list_users method.

geodimm · geodimm · commit 398c94dd0622 · 2015-04-22T00:50:33.000+01:00
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -534,3 +534,34 @@ def connection_error(self, *args, **kwargs):
 
         with self.assertRaises(requests.exceptions.ConnectionError):
             cli.write_points(self.dummy_points)
+
+    def test_get_list_users(self):
+        example_response = (
+            '{"results":[{"series":[{"columns":["user","admin"],'
+            '"values":[["test",false]]}]}]}'
+        )
+
+        with requests_mock.Mocker() as m:
+            m.register_uri(
+                requests_mock.GET,
+                "http://localhost:8086/query",
+                text=example_response
+            )
+
+            self.assertListEqual(
+                self.cli.get_list_users(),
+                [{'user': 'test', 'admin': False}]
+            )
+
+    def test_get_list_users_empty(self):
+        example_response = (
+            '{"results":[{"series":[{"columns":["user","admin"]}]}]}'
+        )
+        with requests_mock.Mocker() as m:
+            m.register_uri(
+                requests_mock.GET,
+                "http://localhost:8086/query",
+                text=example_response
+            )
+
+            self.assertListEqual(self.cli.get_list_users(), [])
diff --git a/tests/influxdb/client_test_with_server.py b/tests/influxdb/client_test_with_server.py
@@ -536,6 +536,19 @@ def test_get_list_series_non_empty(self):
             rsp
         )
 
+    def test_get_list_users_empty(self):
+        rsp = self.cli.get_list_users()
+        self.assertEqual([], rsp)
+
+    def test_get_list_users_non_empty(self):
+        self.cli.query("CREATE USER test WITH PASSWORD 'test'")
+        rsp = self.cli.get_list_users()
+
+        self.assertEqual(
+            [{'user': 'test', 'admin': False}],
+            rsp
+        )
+
     def test_default_retention_policy(self):
         rsp = self.cli.get_list_retention_policies()
         self.assertEqual(
