From 46b519eb64be80cfc8fd915e300ccac735c7a74a Mon Sep 17 00:00:00 2001 From: Leighton Chen Date: Thu, 21 Nov 2024 16:05:17 -0800 Subject: [PATCH] tests --- contrib/opencensus-ext-azure/CHANGELOG.md | 2 +- contrib/opencensus-ext-azure/tests/test_transport_mixin.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/contrib/opencensus-ext-azure/CHANGELOG.md b/contrib/opencensus-ext-azure/CHANGELOG.md index d64901ca9..ad8e8d689 100644 --- a/contrib/opencensus-ext-azure/CHANGELOG.md +++ b/contrib/opencensus-ext-azure/CHANGELOG.md @@ -3,7 +3,7 @@ ## Unreleased - Remove status code `206` from retry code + only count batch level for statsbeat -([#1239](https://github.com/census-instrumentation/opencensus-python/pull/1239)) +([#1247](https://github.com/census-instrumentation/opencensus-python/pull/1247)) ## 1.1.13 diff --git a/contrib/opencensus-ext-azure/tests/test_transport_mixin.py b/contrib/opencensus-ext-azure/tests/test_transport_mixin.py index 3a2c5d6aa..f955adfc5 100644 --- a/contrib/opencensus-ext-azure/tests/test_transport_mixin.py +++ b/contrib/opencensus-ext-azure/tests/test_transport_mixin.py @@ -454,9 +454,10 @@ def test_statsbeat_206_partial_retry(self): ], })) result = mixin._transmit([1, 2, 3]) - self.assertEqual(len(_requests_map), 3) + # We do not record any network statsbeat for 206 status code + self.assertEqual(len(_requests_map), 2) self.assertIsNotNone(_requests_map['duration']) - self.assertEqual(_requests_map['retry'][500], 1) + self.assertIsNone(_requests_map.get('retry')) self.assertEqual(_requests_map['count'], 1) self.assertEqual(result, TransportStatusCode.DROP) storage_mock.put.assert_called_once()