Merge pull request influxdata#58 from areski/fixpep8

jvshahid · jvshahid · commit 9f3d7302a565 · 2014-10-16T13:57:06.000-04:00
Fixed PEP8
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -43,15 +43,15 @@ def request(*args, **kwargs):
 
         # Anyway, Content must be a JSON string (or empty string)
         if not isinstance(c, str):
-            c =  json.dumps(c)
+            c = json.dumps(c)
 
         return _build_response_object(status_code=status_code, content=c)
 
     mocked = patch.object(
         session,
         'request',
-        side_effect = request
-        )
+        side_effect=request
+    )
 
     return mocked
 
@@ -88,7 +88,7 @@ def test_write_points(self):
             }
         ]
 
-        with _mocked_session('post', 200, data) as mocked:
+        with _mocked_session('post', 200, data):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             assert cli.write_points(data) is True
 
@@ -115,7 +115,7 @@ def test_write_points_udp(self):
 
     @raises(Exception)
     def test_write_points_fails(self):
-        with _mocked_session('post', 500) as mocked:
+        with _mocked_session('post', 500):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             cli.write_points([])
 
@@ -131,13 +131,13 @@ def test_write_points_with_precision(self):
             }
         ]
 
-        with _mocked_session('post', 200, data) as mocked:
+        with _mocked_session('post', 200, data):
             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 _mocked_session('post', 500) as mocked:
+        with _mocked_session('post', 500):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             cli.write_points_with_precision([])
 
@@ -154,7 +154,7 @@ def test_delete_points(self):
 
     @raises(Exception)
     def test_delete_points_with_wrong_name(self):
-        with _mocked_session('delete', 400) as mocked:
+        with _mocked_session('delete', 400):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             cli.delete_points("nonexist")
 
@@ -175,70 +175,71 @@ def test_remove_scheduled_delete(self):
 
     def test_query(self):
         data = [
-            {   "name":"foo",
+            {
+                "name": "foo",
                 "columns": ["time", "sequence_number", "column_one"],
                 "points": [
                     [1383876043, 16, "2"], [1383876043, 15, "1"],
                     [1383876035, 14, "2"], [1383876035, 13, "1"]
                 ]
             }
         ]
-        with _mocked_session('get', 200, data) as mocked:
+        with _mocked_session('get', 200, data):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             result = cli.query('select column_one from foo;')
             assert len(result[0]['points']) == 4
 
     @raises(Exception)
     def test_query_fail(self):
-        with _mocked_session('get', 401) as mocked:
+        with _mocked_session('get', 401):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             cli.query('select column_one from foo;')
 
     def test_create_database(self):
-        with _mocked_session('post', 201, {"name": "new_db"}) as mocked:
+        with _mocked_session('post', 201, {"name": "new_db"}):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             assert cli.create_database('new_db') is True
 
     @raises(Exception)
     def test_create_database_fails(self):
-        with _mocked_session('post', 401) as mocked:
+        with _mocked_session('post', 401):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             cli.create_database('new_db')
 
     def test_delete_database(self):
-        with _mocked_session('delete', 204) as mocked:
+        with _mocked_session('delete', 204):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             assert cli.delete_database('old_db') is True
 
     @raises(Exception)
     def test_delete_database_fails(self):
-        with _mocked_session('delete', 401) as mocked:
+        with _mocked_session('delete', 401):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             cli.delete_database('old_db')
 
     def test_get_database_list(self):
         data = [
             {"name": "a_db"}
         ]
-        with _mocked_session('get', 200, data) as mocked:
+        with _mocked_session('get', 200, data):
             cli = InfluxDBClient('host', 8086, 'username', 'password')
             assert len(cli.get_database_list()) == 1
             assert cli.get_database_list()[0]['name'] == 'a_db'
 
     @raises(Exception)
     def test_get_database_list_fails(self):
-        with _mocked_session('get', 401) as mocked:
+        with _mocked_session('get', 401):
             cli = InfluxDBClient('host', 8086, 'username', 'password')
             cli.get_database_list()
 
     def test_delete_series(self):
-        with _mocked_session('delete', 204) as mocked:
+        with _mocked_session('delete', 204):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             cli.delete_series('old_series')
 
     @raises(Exception)
     def test_delete_series_fails(self):
-        with _mocked_session('delete', 401) as mocked:
+        with _mocked_session('delete', 401):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             cli.delete_series('old_series')
 
@@ -295,4 +296,4 @@ def test_delete_database_user(self):
     @raises(NotImplementedError)
     def test_update_permission(self):
         cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
-        cli.update_permission('admin', [])
+        cli.update_permission('admin', [])
