Update test generator script

1 files changed, 8 insertions, 5 deletions

diff --git a/tools/test-generator.py b/tools/test-generator.py
index 681d1755d..8d7377c2c 100755
--- a/tools/test-generator.py
+++ b/tools/test-generator.py
@@ -3,11 +3,11 @@
 import sys
 import math
 import urllib
-import urllib2
+import httplib
 import time
 
 id = '123456789012345'
-server = 'http://localhost:5055'
+server = 'localhost:5055'
 period = 1
 step = 0.001
 
@@ -32,9 +32,10 @@ for i in range(0, len(waypoints)):
         lon = lon1 + (lon2 - lon1) * j / count
         points.append((lat, lon))
 
-def send(lat, lon, course):
+def send(conn, lat, lon, course):
     params = (('id', id), ('timestamp', int(time.time())), ('lat', lat), ('lon', lon), ('bearing', course))
-    urllib2.urlopen(server + '?' + urllib.urlencode(params)).read()
+    conn.request('GET', '?' + urllib.urlencode(params))
+    conn.getresponse()
 
 def course(lat1, lon1, lat2, lon2):
     lat1 = lat1 * math.pi / 180
@@ -47,9 +48,11 @@ def course(lat1, lon1, lat2, lon2):
 
 index = 0
 
+conn = httplib.HTTPConnection(server)
+
 while True:
     (lat1, lon1) = points[index % len(points)]
     (lat2, lon2) = points[(index + 1) % len(points)]
-    send(lat1, lon1, course(lat1, lon1, lat2, lon2))
+    send(conn, lat1, lon1, course(lat1, lon1, lat2, lon2))
     time.sleep(period)
     index += 1