Skip to content

Commit d2bbcb6

Browse files
committed
fix unit test coverage
1 parent c881fd7 commit d2bbcb6

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/test_aws_batch_unsampled_span_processor.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@ def test_on_end_sampled(self, mock_span_class):
2121

2222
self.processor.on_start(mock_span)
2323
self.processor.on_end(mock_span)
24-
self.processor.shutdown()
2524

2625
self.assertEqual(len(self.processor.queue), 0)
2726
mock_span.set_attribute.assert_not_called()
@@ -44,10 +43,10 @@ def test_on_end_not_sampled(self, mock_span_class):
4443
self.assertEqual(len(self.processor.queue), 1)
4544
self.assertIn(AWS_TRACE_FLAG_UNSAMPLED, mock_span1.set_attribute.call_args_list[0][0][0])
4645

46+
self.processor.shutdown()
4747
mock_span2 = mock_span_class.return_value
4848
mock_span2.context.trace_flags = trace_flags
4949
self.processor.on_start(mock_span2)
5050
self.processor.on_end(mock_span2)
5151

52-
self.assertEqual(len(self.processor.queue), 1)
53-
self.assertIn(AWS_TRACE_FLAG_UNSAMPLED, mock_span2.set_attribute.call_args_list[0][0][0])
52+
self.assertEqual(len(self.processor.queue), 0)

0 commit comments

Comments
 (0)