daroche/nginx-otel-nix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Address review notices.

Browse source
This commit is contained in:
Eugene Grebenschikov 2024-12-19 15:31:29 -08:00
parent e625db711a
commit 023ea7962f
2 changed files with 10 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/trace_service.py
View file

@ -14,18 +14,16 @@ class TraceService(trace_service_pb2_grpc.TraceServiceServicer):
self.batches.append(request.resource_spans)
return trace_service_pb2.ExportTracePartialSuccess()
def wait_one_batch(self):
def get_batch(self):
for _ in range(10):
if len(self.batches):
break
time.sleep(0.001)
assert len(self.batches) == 1, "No spans received"
assert len(self.batches) == 1
return self.batches.pop()
def get_span(self):
self.wait_one_batch()
span = self.batches[0][0].scope_spans[0].spans.pop()
self.batches.clear()
return span
return self.get_batch()[0].scope_spans[0].spans.pop()
@pytest.fixture(scope="module")