Add more checkes to get_span.

Co-authored-by: p-pautov <37922380+p-pautov@users.noreply.github.com>
This commit is contained in:
Eugene 2024-12-19 17:00:37 -08:00 committed by GitHub
parent a78bd259bd
commit 41bf12e56e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -23,7 +23,10 @@ class TraceService(trace_service_pb2_grpc.TraceServiceServicer):
return self.batches.pop()
def get_span(self):
return self.get_batch()[0].scope_spans[0].spans.pop()
batch = self.get_batch()
assert len(batch.scope_spans) == 1
assert len(batch.scope_spans[0].spans) == 1
return batch.scope_spans[0].spans.pop()
@pytest.fixture(scope="module")