diff --git a/sdk/cosmos/azure-cosmos/test/test_backwards_compatibility.py b/sdk/cosmos/azure-cosmos/test/test_backwards_compatibility.py index 43ef7ad99663..c3f117327dd5 100644 --- a/sdk/cosmos/azure-cosmos/test/test_backwards_compatibility.py +++ b/sdk/cosmos/azure-cosmos/test/test_backwards_compatibility.py @@ -68,29 +68,11 @@ def side_effect_populate_partition_key_range_statistics(self, *args, **kwargs): self.assertTrue(args[2][http_constants.HttpHeaders.PopulatePartitionKeyRangeStatistics] is True) raise StopIteration - def side_effect_populate_query_metrics(self, *args, **kwargs): - # Extract request headers from args - self.assertTrue(args[2][http_constants.HttpHeaders.PopulateQueryMetrics] is True) - raise StopIteration - def side_effect_populate_quota_info(self, *args, **kwargs): # Extract request headers from args self.assertTrue(args[2][http_constants.HttpHeaders.PopulateQuotaInfo] is True) raise StopIteration - def test_populate_query_metrics(self): - cosmos_client_connection = self.containerForTest.client_connection - cosmos_client_connection._CosmosClientConnection__Get = MagicMock( - side_effect=self.side_effect_populate_query_metrics) - try: - self.containerForTest.read(populate_query_metrics=True) - except StopIteration: - pass - try: - self.containerForTest.read(True) - except StopIteration: - pass - def test_populate_quota_info(self): cosmos_client_connection = self.containerForTest.client_connection cosmos_client_connection._CosmosClientConnection__Get = MagicMock( @@ -100,7 +82,7 @@ def test_populate_quota_info(self): except StopIteration: pass try: - self.containerForTest.read(False, True) + self.containerForTest.read(False, False, True) except StopIteration: pass @@ -113,7 +95,7 @@ def test_populate_partition_key_range_statistics(self): except StopIteration: pass try: - self.containerForTest.read(False, False, True) + self.containerForTest.read(False, True) except StopIteration: pass