Add more checkes to get_span.
Co-authored-by: p-pautov <37922380+p-pautov@users.noreply.github.com>
This commit is contained in:
parent
a78bd259bd
commit
41bf12e56e
1 changed files with 4 additions and 1 deletions
|
|
@ -23,7 +23,10 @@ class TraceService(trace_service_pb2_grpc.TraceServiceServicer):
|
||||||
return self.batches.pop()
|
return self.batches.pop()
|
||||||
|
|
||||||
def get_span(self):
|
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")
|
@pytest.fixture(scope="module")
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue