Add tests for create_user method.

geodimm · geodimm · commit 8aa05eaa8942 · 2015-04-22T21:59:52.000+01:00
diff --git a/tests/influxdb/client_test_with_server.py b/tests/influxdb/client_test_with_server.py
@@ -323,6 +323,39 @@ def test_query_fail(self):
         self.assertIn('{"results":[{"error":"database not found: db',
                       ctx.exception.content)
 
+    def test_create_user(self):
+        self.cli.create_user('test_user', 'secret_password')
+        rsp = list(self.cli.query("SHOW USERS")['results'])
+        self.assertIn({'user': 'test_user', 'admin': False},
+                      rsp)
+
+    def test_create_user_blank_password(self):
+        self.cli.create_user('test_user', '')
+        rsp = list(self.cli.query("SHOW USERS")['results'])
+        self.assertIn({'user': 'test_user', 'admin': False},
+                      rsp)
+
+    def test_create_user_blank_username(self):
+        with self.assertRaises(InfluxDBClientError) as ctx:
+            self.cli.create_user('', 'secret_password')
+        self.assertEqual(400, ctx.exception.code)
+        self.assertIn('{"error":"error parsing query: '
+                      'found WITH, expected identifier',
+                      ctx.exception.content)
+        rsp = list(self.cli.query("SHOW USERS")['results'])
+        self.assertEqual(rsp, [])
+
+    def test_create_user_invalid_username(self):
+        with self.assertRaises(InfluxDBClientError) as ctx:
+            self.cli.create_user('very invalid', 'secret_password')
+        self.assertEqual(400, ctx.exception.code)
+        self.assertIn('{"error":"error parsing query: '
+                      'found invalid, expected WITH',
+                      ctx.exception.content)
+        rsp = list(self.cli.query("SHOW USERS")['results'])
+        self.assertEqual(rsp, [])
+
+
 ############################################################################
 
 
