Update unit tests

smly · smly · commit 5ead1d8c4583 · 2013-11-08T10:50:08.000+09:00
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -518,7 +518,7 @@ def delete_database_user(self, username):
     # update the user by POSTing to db/site_dev/users/paul
 
     # todo
-    def update_permission(self, json_body):
+    def update_permission(self, username, json_body):
         """
         Update read/write permission
         """
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -1,4 +1,7 @@
 # -*- coding: utf-8 -*-
+"""
+unit tests
+"""
 import requests
 from nose.tools import raises
 from mock import patch
@@ -25,6 +28,77 @@ def test_switch_user(self):
         assert cli._username == 'another_username'
         assert cli._password == 'another_password'
 
+    def test_write_points(self):
+        data = [
+            {
+                "points": [
+                    ["1", 1, 1.0],
+                    ["2", 2, 2.0]
+                ],
+                "name": "foo",
+                "columns": ["column_one", "column_two", "column_three"]
+            }
+        ]
+
+        with patch.object(requests, 'post') as mocked_post:
+            mocked_post.return_value = _build_response_object(status_code=200)
+            cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+            assert cli.write_points(data) is True
+
+    @raises(Exception)
+    def test_write_points_fails(self):
+        with patch.object(requests, 'post') as mocked_post:
+            mocked_post.return_value = _build_response_object(status_code=500)
+            cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+            cli.write_points([])
+
+    def test_write_points_with_precision(self):
+        data = [
+            {
+                "points": [
+                    ["1", 1, 1.0],
+                    ["2", 2, 2.0]
+                ],
+                "name": "foo",
+                "columns": ["column_one", "column_two", "column_three"]
+            }
+        ]
+
+        with patch.object(requests, 'post') as mocked_post:
+            mocked_post.return_value = _build_response_object(status_code=200)
+            cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+            assert cli.write_points_with_precision(data) is True
+
+    @raises(Exception)
+    def test_write_points_with_precision_fails(self):
+        with patch.object(requests, 'post') as mocked_post:
+            mocked_post.return_value = _build_response_object(status_code=500)
+            cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+            cli.write_points_with_precision([])
+
+    @raises(NotImplementedError)
+    def test_delete_points(self):
+        cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+        cli.delete_points([])
+
+    @raises(NotImplementedError)
+    def test_create_scheduled_delete(self):
+        cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+        cli.create_scheduled_delete([])
+
+    @raises(NotImplementedError)
+    def test_get_list_scheduled_delete(self):
+        cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+        cli.get_list_scheduled_delete()
+
+    @raises(NotImplementedError)
+    def test_remove_scheduled_delete(self):
+        cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+        cli.remove_scheduled_delete(1)
+
+    def test_query(self):
+        pass
+
     def test_create_database(self):
         with patch.object(requests, 'post') as mocked_post:
             mocked_post.return_value = _build_response_object(status_code=201)
@@ -51,4 +125,57 @@ def test_delete_database_fails(self):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             cli.delete_database('old_db')
 
+    def test_get_list_cluster_admins(self):
+        pass
+
+    def test_add_cluster_admin(self):
+        pass
+
+    def test_update_cluster_admin_password(self):
+        pass
+
+    def test_delete_cluster_admin(self):
+        pass
+
+    def test_set_database_admin(self):
+        pass
+
+    def test_unset_database_admin(self):
+        pass
+
+    @raises(NotImplementedError)
+    def test_get_list_database_admins(self):
+        cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+        cli.get_list_database_admins()
+
+    @raises(NotImplementedError)
+    def test_add_database_admin(self):
+        cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+        cli.add_database_admin('admin', 'admin_secret_password')
+
+    @raises(NotImplementedError)
+    def test_update_database_admin_password(self):
+        cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+        cli.update_database_admin_password('admin', 'admin_secret_password')
+
+    @raises(NotImplementedError)
+    def test_delete_database_admin(self):
+        cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+        cli.delete_database_admin('admin')
+
+    def test_get_database_user(self):
+        pass
+
+    def test_add_database_user(self):
+        pass
+
+    def test_update_database_user_password(self):
+        pass
+
+    def test_delete_database_user(self):
+        pass
 
+    @raises(NotImplementedError)
+    def test_update_permission(self):
+        cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+        cli.update_permission('admin', [])
