fmt: black celery tests (#1314)

* celery: black fmt tests

* celery: use self.assert_is_measured

Co-authored-by: Julien Danjou <[email protected]>

Author: Kyle-Verhoog

pyproject.toml

@@ -105,7 +105,6 @@ exclude = '''
       | botocore
       | bottle
       | cassandra
-      | celery
       | config.py
       | consul
       | dbapi

tests/contrib/celery/autopatch.py

@@ -1,9 +1,9 @@
 from ddtrace import Pin
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     # have to import celery in order to have the post-import hooks run
     import celery
 
     # now celery.Celery should be patched and should have a pin
     assert Pin.get_from(celery.Celery)
-    print('Test success')
+    print("Test success")

tests/contrib/celery/base.py

@@ -8,9 +8,9 @@
 from ...base import BaseTracerTestCase
 
 
-REDIS_URL = 'redis://127.0.0.1:{port}'.format(port=REDIS_CONFIG['port'])
-BROKER_URL = '{redis}/{db}'.format(redis=REDIS_URL, db=0)
-BACKEND_URL = '{redis}/{db}'.format(redis=REDIS_URL, db=1)
+REDIS_URL = "redis://127.0.0.1:{port}".format(port=REDIS_CONFIG["port"])
+BROKER_URL = "{redis}/{db}".format(redis=REDIS_URL, db=0)
+BACKEND_URL = "{redis}/{db}".format(redis=REDIS_URL, db=1)
 
 
 class CeleryBaseTestCase(BaseTracerTestCase):
@@ -23,8 +23,8 @@ def setUp(self):
 
         # instrument Celery and create an app with Broker and Result backends
         patch()
-        self.pin = Pin(service='celery-unittest', tracer=self.tracer)
-        self.app = Celery('celery.test_app', broker=BROKER_URL, backend=BACKEND_URL)
+        self.pin = Pin(service="celery-unittest", tracer=self.tracer)
+        self.app = Celery("celery.test_app", broker=BROKER_URL, backend=BACKEND_URL)
         # override pins to use our Dummy Tracer
         Pin.override(self.app, tracer=self.tracer)
 

tests/contrib/celery/test_autopatch.py

@@ -7,7 +7,5 @@ class DdtraceRunTest(unittest.TestCase):
     """Test that celery is patched successfully if run with ddtrace-run."""
 
     def test_autopatch(self):
-        out = subprocess.check_output(
-            ['ddtrace-run', 'python', 'tests/contrib/celery/autopatch.py']
-        )
-        assert out.startswith(b'Test success')
+        out = subprocess.check_output(["ddtrace-run", "python", "tests/contrib/celery/autopatch.py"])
+        assert out.startswith(b"Test success")