tests

Author: lzchen

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
 

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()