Add test to check sqlite3 sends service_info

Author: bmermet

tests/contrib/httplib/test_httplib.py

@@ -12,9 +12,8 @@
 from ddtrace.contrib.httplib.patch import should_skip_request
 from ddtrace.pin import Pin
 
-from .utils import override_global_tracer
 from ...test_tracer import get_dummy_tracer
-from ...util import assert_dict_issuperset
+from ...util import assert_dict_issuperset, override_global_tracer
 
 
 if PY2:

tests/contrib/httplib/utils.py

@@ -6,6 +6,7 @@
 from nose.tools import eq_
 
 # project
+import ddtrace
 from ddtrace import Pin
 from ddtrace.contrib.sqlite3 import connection_factory
 from ddtrace.contrib.sqlite3.patch import patch, unpatch
@@ -31,6 +32,22 @@ def setUp(self):
     def tearDown(self):
         unpatch()
 
+    def test_service_info(self):
+        tracer = get_dummy_tracer()
+        backup_tracer = ddtrace.tracer
+        ddtrace.tracer = tracer
+
+        db = sqlite3.connect(":memory:")
+
+        services = tracer.writer.pop_services()
+        eq_(len(services), 1)
+        expected = {
+            'sqlite': {'app': 'sqlite', 'app_type': 'db'}
+        }
+        eq_(expected, services)
+
+        ddtrace.tracer = backup_tracer
+
     def test_sqlite(self):
         tracer = get_dummy_tracer()
         writer = tracer.writer

tests/contrib/sqlite3/test_sqlite3.py

@@ -1,4 +1,6 @@
+import ddtrace
 import mock
+from contextlib import contextmanager
 from nose.tools import ok_
 
 class FakeTime(object):
@@ -38,3 +40,15 @@ def assert_dict_issuperset(a, b):
 def assert_list_issuperset(a, b):
     ok_(set(a).issuperset(set(b)),
             msg="{a} is not a superset of {b}".format(a=a, b=b))
+
+@contextmanager
+def override_global_tracer(tracer):
+    """Helper functions that overrides the global tracer available in the
+    `ddtrace` package. This is required because in some `httplib` tests we
+    can't get easily the PIN object attached to the `HTTPConnection` to
+    replace the used tracer with a dummy tracer.
+    """
+    original_tracer = ddtrace.tracer
+    ddtrace.tracer = tracer
+    yield
+    ddtrace.tracer = original_tracer