revert test removal

DataDog · Jan 16, 2025 · bc8efa5 · bc8efa5
1 parent 837afa9
commit bc8efa5
Showing 1 changed file with 35 additions and 0 deletions.
diff --git a/tests/tracer/test_filters.py b/tests/tracer/test_filters.py
@@ -1,8 +1,43 @@
+from unittest import TestCase
+
 import pytest
 
+from ddtrace._trace.filters import FilterRequestsOnUrl
+from ddtrace._trace.span import Span
+from ddtrace.ext.http import URL
 from ddtrace.trace import TraceFilter
 
 
+class FilterRequestOnUrlTests(TestCase):
+    def test_is_match(self):
+        span = Span(name="Name")
+        span.set_tag(URL, r"http://example.com")
+        filtr = FilterRequestsOnUrl("http://examp.*.com")
+        trace = filtr.process_trace([span])
+        self.assertIsNone(trace)
+
+    def test_is_not_match(self):
+        span = Span(name="Name")
+        span.set_tag(URL, r"http://anotherexample.com")
+        filtr = FilterRequestsOnUrl("http://examp.*.com")
+        trace = filtr.process_trace([span])
+        self.assertIsNotNone(trace)
+
+    def test_list_match(self):
+        span = Span(name="Name")
+        span.set_tag(URL, r"http://anotherdomain.example.com")
+        filtr = FilterRequestsOnUrl([r"http://domain\.example\.com", r"http://anotherdomain\.example\.com"])
+        trace = filtr.process_trace([span])
+        self.assertIsNone(trace)
+
+    def test_list_no_match(self):
+        span = Span(name="Name")
+        span.set_tag(URL, r"http://cooldomain.example.com")
+        filtr = FilterRequestsOnUrl([r"http://domain\.example\.com", r"http://anotherdomain\.example\.com"])
+        trace = filtr.process_trace([span])
+        self.assertIsNotNone(trace)
+
+
 def test_not_implemented_trace_filter():
     class Filter(TraceFilter):
         pass