Urllib3 instrumentation tests

Author: pglombardo

tests/test_urllib3.py

@@ -0,0 +1,62 @@
+from __future__ import absolute_import
+from nose.tools import assert_equals
+import opentracing
+import instana.instrumentation.urllib3
+import urllib3
+
+
+class TestUrllib3:
+    def setUp(self):
+        """ Clear all spans before a test run """
+        self.http = urllib3.PoolManager()
+        self.recorder = opentracing.global_tracer.recorder
+        self.recorder.clear_spans()
+
+    def tearDown(self):
+        """ Do nothing for now """
+        return None
+
+    def test_vanilla_requests(self):
+        r = self.http.request('GET', 'http://127.0.0.1:5000/')
+        assert_equals(r.status, 200)
+
+    def test_get_request(self):
+        span = opentracing.global_tracer.start_span("test")
+        r = self.http.request('GET', 'http://127.0.0.1:5000/')
+        span.finish()
+
+        spans = self.recorder.queued_spans()
+        first_span = spans[1]
+        second_span = spans[0]
+
+        assert(r)
+        assert_equals(200, r.status)
+        assert_equals(2, len(spans))
+        assert_equals("test", first_span.data.sdk.name)
+        assert_equals("urllib3", second_span.n)
+        assert_equals(200, second_span.data.http.status)
+        assert_equals("http://127.0.0.1:5000/", second_span.data.http.url)
+        assert_equals("GET", second_span.data.http.method)
+
+        assert_equals(None, second_span.error)
+        assert_equals(None, second_span.ec)
+
+    def test_put_request(self):
+        span = opentracing.global_tracer.start_span("test")
+        r = self.http.request('PUT', 'http://127.0.0.1:5000/notfound')
+        span.finish()
+
+        spans = self.recorder.queued_spans()
+        first_span = spans[1]
+        second_span = spans[0]
+
+        assert(r)
+        assert_equals(404, r.status)
+        assert_equals(2, len(spans))
+        assert_equals("test", first_span.data.sdk.name)
+        assert_equals("urllib3", second_span.n)
+        assert_equals(404, second_span.data.http.status)
+        assert_equals("http://127.0.0.1:5000/notfound", second_span.data.http.url)
+        assert_equals("PUT", second_span.data.http.method)
+        assert_equals(None, second_span.error)
+        assert_equals(None, second_span.ec)