99
1010import pytest
1111
12- # from pytest import MonkeyPatch
13-
1412import ads
1513from ads .telemetry .telemetry import (
1614 EXTRA_USER_AGENT_INFO ,
@@ -37,8 +35,9 @@ def teardown_method(self):
3735 @patch ("oci.config.validate_config" )
3836 @patch ("oci.signer.Signer" )
3937 def test_user_agent_api_keys_using_test_profile (
40- self , mock_signer , mock_validate_config
38+ self , mock_signer , mock_validate_config , monkeypatch
4139 ):
40+ monkeypatch .delenv (EXTRA_USER_AGENT_INFO , raising = False )
4241 with patch ("oci.config.from_file" , return_value = self .test_config ):
4342 auth_info = ads .auth .api_keys ("test_path" , "TEST_PROFILE" )
4443 assert (
@@ -49,6 +48,7 @@ def test_user_agent_api_keys_using_test_profile(
4948 @patch ("oci.auth.signers.get_resource_principals_signer" )
5049 def test_user_agent_rp (self , mock_signer , monkeypatch ):
5150 monkeypatch .delenv ("OCI_RESOURCE_PRINCIPAL_VERSION" , raising = False )
51+ monkeypatch .delenv (EXTRA_USER_AGENT_INFO , raising = False )
5252 importlib .reload (ads .config )
5353 importlib .reload (ads .telemetry )
5454 auth_info = ads .auth .resource_principal ()
@@ -64,6 +64,7 @@ def test_user_agent_default_signer(
6464 ):
6565 # monkeypatch = MonkeyPatch()
6666 monkeypatch .delenv ("OCI_RESOURCE_PRINCIPAL_VERSION" , raising = False )
67+ monkeypatch .delenv (EXTRA_USER_AGENT_INFO , raising = False )
6768 importlib .reload (ads .config )
6869 with patch ("oci.config.from_file" , return_value = self .test_config ):
6970 auth_info = ads .auth .default_signer ()
@@ -121,6 +122,7 @@ def test_user_agent_default_signer_known_resources(
121122 # monkeypatch = MonkeyPatch()
122123 monkeypatch .setenv ("OCI_RESOURCE_PRINCIPAL_VERSION" , "1.1" )
123124 monkeypatch .setenv (INPUT_DATA ["RESOURCE_KEY" ], "1234" )
125+ monkeypatch .delenv (EXTRA_USER_AGENT_INFO , raising = False )
124126 if INPUT_DATA [EXTRA_USER_AGENT_INFO ] is not None :
125127 monkeypatch .setenv (EXTRA_USER_AGENT_INFO , INPUT_DATA [EXTRA_USER_AGENT_INFO ])
126128
@@ -148,6 +150,7 @@ def test_user_agent_default_signer_ociservice(
148150 monkeypatch ,
149151 ):
150152 monkeypatch .setenv ("OCI_RESOURCE_PRINCIPAL_VERSION" , "1.1" )
153+ monkeypatch .delenv (EXTRA_USER_AGENT_INFO , raising = False )
151154
152155 importlib .reload (ads .config )
153156 importlib .reload (ads .telemetry )
0 commit comments