Add tests for drop_user method.

geodimm · geodimm · commit b3c04e1bc0cf · 2015-04-22T22:16:12.000+01:00
diff --git a/tests/influxdb/client_test_with_server.py b/tests/influxdb/client_test_with_server.py
@@ -355,6 +355,27 @@ def test_create_user_invalid_username(self):
         rsp = list(self.cli.query("SHOW USERS")['results'])
         self.assertEqual(rsp, [])
 
+    def test_drop_user(self):
+        self.cli.query("CREATE USER test WITH PASSWORD 'test'")
+        self.cli.drop_user('test')
+        users = list(self.cli.query("SHOW USERS")['results'])
+        self.assertEqual(users, [])
+
+    def test_drop_user_nonexisting(self):
+        with self.assertRaises(InfluxDBClientError) as ctx:
+            self.cli.drop_user('test')
+        self.assertEqual(500, ctx.exception.code)
+        self.assertIn('{"results":[{"error":"user not found"}]}',
+                      ctx.exception.content)
+
+    def test_drop_user_invalid(self):
+        with self.assertRaises(InfluxDBClientError) as ctx:
+            self.cli.drop_user('very invalid')
+        self.assertEqual(400, ctx.exception.code)
+        self.assertIn('{"error":"error parsing query: '
+                      'found invalid, expected',
+                      ctx.exception.content)
+
 
 ############################################################################
 
