Fixed flake8 + added test for permissions

aviau · aviau · commit beecccc12df4 · 2014-10-30T12:11:20.000-04:00
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -624,7 +624,9 @@ def add_database_user(self, new_username, new_password, permissions=None):
             try:
                 data['readFrom'], data['writeTo'] = permissions
             except (ValueError, TypeError):
-                raise TypeError("'permissions' must be (readFrom, writeTo) tuple")
+                raise TypeError(
+                    "'permissions' must be (readFrom, writeTo) tuple"
+                )
 
         self.request(
             url=url,
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -326,6 +326,21 @@ def test_get_database_user(self):
     def test_add_database_user(self):
         pass
 
+    def test_add_database_user_bad_permissions(self):
+        cli = InfluxDBClient()
+
+        with self.assertRaises(Exception) as ex:
+            cli.add_database_user(
+                new_password='paul',
+                new_username='paul',
+                permissions=('hello', 'hello', 'hello')
+            )
+
+        self.assertEqual(
+            str(ex.exception),
+            "'permissions' must be (readFrom, writeTo) tuple"
+        )
+
     def test_update_database_user_password(self):
         pass
 
