Added test_write_points_udp

aviau · aviau · commit bc90e455a88a · 2014-06-06T16:44:26.000-04:00
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -4,6 +4,7 @@
 """
 import json
 import requests
+import socket
 from nose.tools import raises
 from mock import patch
 
@@ -36,7 +37,7 @@ def request(*args, **kwargs):
                 assert isinstance(data, str)
 
                 # Data must be a JSON string
-                assert c == json.loads(data)
+                assert c == json.loads(data, strict=True)
 
                 c = data
 
@@ -91,6 +92,27 @@ def test_write_points(self):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             assert cli.write_points(data) is True
 
+    def test_write_points_udp(self):
+        data = [
+            {
+                "points": [
+                    ["1", 1, 1.0],
+                    ["2", 2, 2.0]
+                ],
+                "name": "foo",
+                "columns": ["column_one", "column_two", "column_three"]
+            }
+        ]
+
+        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        s.bind(('0.0.0.0', 4444))
+
+        cli = InfluxDBClient('localhost', 8086, 'root', 'root', 'test', use_udp=True, udp_port=4444)
+        cli.write_points(data)
+
+        received_data, addr = s.recvfrom(1024)
+        assert data == json.loads(received_data.decode(), strict=True)
+
     @raises(Exception)
     def test_write_points_fails(self):
         with _mocked_session('post', 500) as mocked:
@@ -273,4 +295,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', [])
