Added test for add_database_user

aviau · aviau · commit a52f419ff61d · 2014-10-30T19:58:46.000-04:00
diff --git a/test-requirements.txt b/test-requirements.txt
@@ -1,2 +1,3 @@
 nose
 mock
+httpretty
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -6,6 +6,7 @@
 import requests
 import socket
 import unittest
+import httpretty
 from nose.tools import raises
 from mock import patch
 
@@ -331,8 +332,29 @@ def test_delete_database_admin(self):
     def test_get_database_user(self):
         pass
 
+    @httpretty.activate
     def test_add_database_user(self):
-        pass
+        httpretty.register_uri(
+            httpretty.POST,
+            "http://localhost:8086/db/db/users"
+        )
+
+        cli = InfluxDBClient(database='db')
+        cli.add_database_user(
+            new_username='paul',
+            new_password='laup',
+            permissions=('.*', '.*')
+        )
+
+        self.assertEqual(
+            httpretty.last_request().parsed_body,
+            {
+                u'writeTo': u'.*',
+                u'password': u'laup',
+                u'readFrom': u'.*',
+                u'name': u'paul'
+            }
+        )
 
     def test_add_database_user_bad_permissions(self):
         cli = InfluxDBClient()
