mongoengine: restore v3 interface

Author: clutchski

ddtrace/contrib/mongoengine/__init__.py

@@ -24,6 +24,8 @@
 
 with require_modules(required_modules) as missing_modules:
     if not missing_modules:
-        from .patch import patch
+        from .patch import patch, trace_mongoengine
+
+        __all__ = ['patch', 'trace_mongoengine']
+
 
-        __all__ = ['patch']

ddtrace/contrib/mongoengine/patch.py

@@ -1,6 +1,7 @@
 import mongoengine
 
 from .trace import WrappedConnect
+from ddtrace.util import deprecated
 
 # Original connect function
 _connect = mongoengine.connect
@@ -12,3 +13,7 @@ def patch():
 def unpatch():
     setattr(mongoengine, 'connect', _connect)
 
+@deprecated(message='Use patching instead (see the docs).', version='0.6.0')
+def trace_mongoengine(*args, **kwargs):
+    return _connect
+

tests/contrib/mongoengine/test_backwards.py

@@ -0,0 +1,27 @@
+"""
+ensure old interfaces exist and won't break things.
+"""
+
+
+import mongoengine
+
+from tests.test_tracer import get_dummy_tracer
+from tests.contrib import config
+
+class Singer(mongoengine.Document):
+    first_name = mongoengine.StringField(max_length=50)
+    last_name = mongoengine.StringField(max_length=50)
+
+
+def test_less_than_v04():
+    # interface from < v0.4
+    from ddtrace.contrib.mongoengine import trace_mongoengine
+    tracer = get_dummy_tracer()
+
+    connect = trace_mongoengine(tracer, service="my-mongo-db", patch=False)
+    connect(port=config.MONGO_CONFIG['port'])
+
+    lc = Singer()
+    lc.first_name = 'leonard'
+    lc.last_name = 'cohen'
+    lc.save()