Skip to content

Commit a9ac2e7

Browse files
chore(test): use dummy tracer fixture in some tracer tests
1 parent bbe6f5b commit a9ac2e7

File tree

7 files changed

+78
-137
lines changed

7 files changed

+78
-137
lines changed

tests/conftest.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,9 @@ def tracer(use_global_tracer):
184184
if use_global_tracer:
185185
return ddtrace.tracer
186186
else:
187-
return DummyTracer()
187+
t = DummyTracer()
188+
yield t
189+
t.shutdown()
188190

189191

190192
@pytest.fixture

tests/tracer/test_gitmetadata.py

Lines changed: 9 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
import pytest
1111

1212
from tests.subprocesstest import run_in_subprocess
13-
from tests.utils import DummyTracer
1413
from tests.utils import TracerTestCase
1514

1615

@@ -42,9 +41,7 @@ class GitMetadataTestCase(TracerTestCase):
4241
DD_MAIN_PACKAGE="mypackage",
4342
)
4443
)
45-
def test_gitmetadata_from_package(self):
46-
tracer = DummyTracer()
47-
44+
def test_gitmetadata_from_package(self, tracer):
4845
with tracer.trace("span") as s:
4946
pass
5047

@@ -57,9 +54,7 @@ def test_gitmetadata_from_package(self):
5754
DD_TAGS="git.commit.sha:12345,git.repository_url:github.com/user/tag_repo",
5855
)
5956
)
60-
def test_gitmetadata_from_DD_TAGS(self):
61-
tracer = DummyTracer()
62-
57+
def test_gitmetadata_from_DD_TAGS(self, tracer):
6358
with tracer.trace("span") as s:
6459
pass
6560

@@ -78,9 +73,7 @@ def test_gitmetadata_from_DD_TAGS(self):
7873
DD_MAIN_PACKAGE="mypackage",
7974
)
8075
)
81-
def test_gitmetadata_from_ENV(self):
82-
tracer = DummyTracer()
83-
76+
def test_gitmetadata_from_ENV(self, tracer):
8477
with tracer.trace("span") as s:
8578
pass
8679

@@ -102,9 +95,7 @@ def test_gitmetadata_from_ENV(self):
10295
DD_TRACE_GIT_METADATA_ENABLED="false",
10396
)
10497
)
105-
def test_gitmetadata_disabled(self):
106-
tracer = DummyTracer()
107-
98+
def test_gitmetadata_disabled(self, tracer):
10899
with tracer.trace("span") as s:
109100
pass
110101

@@ -121,9 +112,7 @@ def test_gitmetadata_disabled(self):
121112
DD_MAIN_PACKAGE="pytest",
122113
)
123114
)
124-
def test_gitmetadata_package_without_metadata(self):
125-
tracer = DummyTracer()
126-
115+
def test_gitmetadata_package_without_metadata(self, tracer):
127116
with tracer.trace("span") as s:
128117
pass
129118

@@ -141,9 +130,7 @@ def test_gitmetadata_package_without_metadata(self):
141130
DD_MAIN_PACKAGE="mypackage",
142131
)
143132
)
144-
def test_gitmetadata_from_env_filtering_https(self):
145-
tracer = DummyTracer()
146-
133+
def test_gitmetadata_from_env_filtering_https(self, tracer):
147134
with tracer.trace("span") as s:
148135
pass
149136

@@ -161,9 +148,7 @@ def test_gitmetadata_from_env_filtering_https(self):
161148
DD_TAGS="git.commit.sha:12345,git.repository_url:https://username:[email protected]/user/tag_repo.git",
162149
)
163150
)
164-
def test_gitmetadata_from_ddtags_filtering_https(self):
165-
tracer = DummyTracer()
166-
151+
def test_gitmetadata_from_ddtags_filtering_https(self, tracer):
167152
with tracer.trace("span") as s:
168153
pass
169154

@@ -182,9 +167,7 @@ def test_gitmetadata_from_ddtags_filtering_https(self):
182167
DD_MAIN_PACKAGE="mypackage",
183168
)
184169
)
185-
def test_gitmetadata_from_env_filtering_ssh(self):
186-
tracer = DummyTracer()
187-
170+
def test_gitmetadata_from_env_filtering_ssh(self, tracer):
188171
with tracer.trace("span") as s:
189172
pass
190173

@@ -202,9 +185,7 @@ def test_gitmetadata_from_env_filtering_ssh(self):
202185
DD_TAGS="git.commit.sha:12345,git.repository_url:ssh://[email protected]/user/tag_repo.git",
203186
)
204187
)
205-
def test_gitmetadata_from_ddtags_filtering_ssh(self):
206-
tracer = DummyTracer()
207-
188+
def test_gitmetadata_from_ddtags_filtering_ssh(self, tracer):
208189
with tracer.trace("span") as s:
209190
pass
210191

tests/tracer/test_global_config.py

Lines changed: 12 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ class GlobalConfigTestCase(TestCase):
1717
def setUp(self):
1818
self.config = Config()
1919
self.config.web = IntegrationConfig(self.config, "web")
20-
self.tracer = DummyTracer()
2120

2221
def test_registration(self):
2322
# ensure an integration can register a new list of settings
@@ -112,7 +111,7 @@ def test_settings_merge_deep(self):
112111
assert self.config.requests["a"]["b"]["c"] is True
113112
assert self.config.requests["a"]["b"]["d"] is True
114113

115-
def test_settings_hook(self):
114+
def test_settings_hook(self, tracer):
116115
"""
117116
When calling `Hooks.emit()`
118117
When there is a hook registered
@@ -125,7 +124,7 @@ def on_web_request(span):
125124
span.set_tag("web.request", "/")
126125

127126
# Create our span
128-
with self.tracer.start_span("web.request") as span:
127+
with tracer.start_span("web.request") as span:
129128
assert "web.request" not in span.get_tags()
130129

131130
# Emit the span
@@ -134,7 +133,7 @@ def on_web_request(span):
134133
# Assert we updated the span as expected
135134
assert span.get_tag("web.request") == "/"
136135

137-
def test_settings_hook_args(self):
136+
def test_settings_hook_args(self, tracer):
138137
"""
139138
When calling `Hooks.emit()` with arguments
140139
When there is a hook registered
@@ -148,7 +147,7 @@ def on_web_request(span, request, response):
148147
span.set_tag("web.response", response)
149148

150149
# Create our span
151-
with self.tracer.start_span("web.request") as span:
150+
with tracer.start_span("web.request") as span:
152151
assert "web.request" not in span.get_tags()
153152

154153
# Emit the span
@@ -159,7 +158,7 @@ def on_web_request(span, request, response):
159158
assert span.get_tag("web.request") == "request"
160159
assert span.get_tag("web.response") == "response"
161160

162-
def test_settings_hook_args_failure(self):
161+
def test_settings_hook_args_failure(self, tracer):
163162
"""
164163
When calling `Hooks.emit()` with arguments
165164
When there is a hook registered that is missing parameters
@@ -173,7 +172,7 @@ def on_web_request(span, request):
173172
span.set_tag("web.request", request)
174173

175174
# Create our span
176-
with self.tracer.start_span("web.request") as span:
175+
with tracer.start_span("web.request") as span:
177176
assert "web.request" not in span.get_tags()
178177

179178
# Emit the span
@@ -183,7 +182,7 @@ def on_web_request(span, request):
183182
# Assert we did not update the span
184183
assert "web.request" not in span.get_tags()
185184

186-
def test_settings_multiple_hooks(self):
185+
def test_settings_multiple_hooks(self, tracer):
187186
"""
188187
When calling `Hooks.emit()`
189188
When there are multiple hooks registered
@@ -204,7 +203,7 @@ def on_web_request3(span):
204203
span.set_tag("web.method", "GET")
205204

206205
# Create our span
207-
with self.tracer.start_span("web.request") as span:
206+
with tracer.start_span("web.request") as span:
208207
assert "web.request" not in span.get_tags()
209208
assert "web.status" not in span.get_metrics()
210209
assert "web.method" not in span.get_tags()
@@ -217,7 +216,7 @@ def on_web_request3(span):
217216
assert span.get_metric("web.status") == 200
218217
assert span.get_tag("web.method") == "GET"
219218

220-
def test_settings_hook_failure(self):
219+
def test_settings_hook_failure(self, tracer):
221220
"""
222221
When calling `Hooks.emit()`
223222
When the hook raises an exception
@@ -228,20 +227,20 @@ def test_settings_hook_failure(self):
228227
self.config.web.hooks.register("request")(on_web_request)
229228

230229
# Create our span
231-
with self.tracer.start_span("web.request") as span:
230+
with tracer.start_span("web.request") as span:
232231
# Emit the span
233232
# DEV: This is the test, to ensure no exceptions are raised
234233
self.config.web.hooks.emit("request", span)
235234
on_web_request.assert_called()
236235

237-
def test_settings_no_hook(self):
236+
def test_settings_no_hook(self, tracer):
238237
"""
239238
When calling `Hooks.emit()`
240239
When no hook is registered
241240
we do not raise an exception
242241
"""
243242
# Create our span
244-
with self.tracer.start_span("web.request") as span:
243+
with tracer.start_span("web.request") as span:
245244
# Emit the span
246245
# DEV: This is the test, to ensure no exceptions are raised
247246
self.config.web.hooks.emit("request", span)

0 commit comments

Comments
 (0)