|
14 | 14 | import google.protobuf.json_format |
15 | 15 | import opentelemetry.proto.trace.v1.trace_pb2 |
16 | 16 |
|
| 17 | +from dnsdisttests import pickAvailablePort |
17 | 18 | import test_Protobuf |
18 | 19 |
|
19 | 20 |
|
@@ -66,12 +67,8 @@ def sendQueryAndGetProtobuf( |
66 | 67 | self.assertTrue(receivedResponse) |
67 | 68 | self.assertEqual(response, receivedResponse) |
68 | 69 |
|
69 | | - if self._protobufQueue.empty(): |
70 | | - # let the protobuf messages the time to get there |
71 | | - time.sleep(1) |
72 | | - |
73 | 70 | # check the protobuf message corresponding to the UDP query |
74 | | - return self.getFirstProtobufMessage() |
| 71 | + return self.getFirstProtobufMessage(timeout=1) |
75 | 72 |
|
76 | 73 | def checkOTData( |
77 | 74 | self, |
@@ -719,6 +716,11 @@ def servfailOnTraceParent(request: dns.message.Message): |
719 | 716 |
|
720 | 717 |
|
721 | 718 | class TestOpenTelemetryTracingStripIncomingTraceParent(DNSDistOpenTelemetryProtobufTest): |
| 719 | + # this test suite uses a different responder port |
| 720 | + # because it uses a different responder logic so we |
| 721 | + # need to make sure we are not hitting the backend |
| 722 | + # from a different test |
| 723 | + _testServerPort = pickAvailablePort() |
722 | 724 | _yaml_config_params = [ |
723 | 725 | "_testServerPort", |
724 | 726 | ] |
@@ -821,6 +823,11 @@ def verifyTraceparentInQuery(request: dns.message.Message): |
821 | 823 |
|
822 | 824 |
|
823 | 825 | class TestOpenTelemetryTracingSendTraceparentDownstream(DNSDistOpenTelemetryProtobufTest): |
| 826 | + # this test suite uses a different responder port |
| 827 | + # because it uses a different responder logic so we |
| 828 | + # need to make sure we are not hitting the backend |
| 829 | + # from a different test |
| 830 | + _testServerPort = pickAvailablePort() |
824 | 831 | _yaml_config_params = [ |
825 | 832 | "_testServerPort", |
826 | 833 | ] |
|
0 commit comments