Skip to content

Commit 103c760

Browse files
authored
Merge pull request PowerDNS#17469 from rgacogne/ddist-pb-tests-timeout
dnsdist: Consistently handle slow protobuf messages in our tests
2 parents d609e61 + ca45be9 commit 103c760

2 files changed

Lines changed: 51 additions & 88 deletions

File tree

regression-tests.dnsdist/test_OpenTelemetryTracing.py

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
import google.protobuf.json_format
1515
import opentelemetry.proto.trace.v1.trace_pb2
1616

17+
from dnsdisttests import pickAvailablePort
1718
import test_Protobuf
1819

1920

@@ -66,12 +67,8 @@ def sendQueryAndGetProtobuf(
6667
self.assertTrue(receivedResponse)
6768
self.assertEqual(response, receivedResponse)
6869

69-
if self._protobufQueue.empty():
70-
# let the protobuf messages the time to get there
71-
time.sleep(1)
72-
7370
# check the protobuf message corresponding to the UDP query
74-
return self.getFirstProtobufMessage()
71+
return self.getFirstProtobufMessage(timeout=1)
7572

7673
def checkOTData(
7774
self,
@@ -719,6 +716,11 @@ def servfailOnTraceParent(request: dns.message.Message):
719716

720717

721718
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()
722724
_yaml_config_params = [
723725
"_testServerPort",
724726
]
@@ -821,6 +823,11 @@ def verifyTraceparentInQuery(request: dns.message.Message):
821823

822824

823825
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()
824831
_yaml_config_params = [
825832
"_testServerPort",
826833
]

0 commit comments

Comments
 (0)