xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-04 21:00:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
mrT23
2025-06-02 18:04:44 +03:00
parent b6787fa6de
commit ecb39856ee
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/unittest/test_aws_secrets_manager_provider.py
View File

@ -67,7 +67,7 @@ class TestAWSSecretsManagerProvider:
mock_client.update_secret.return_value = {} mock_client.update_secret.return_value = {}
provider.store_secret('test-secret', 'test-value') provider.store_secret('test-secret', 'test-value')
mock_client.update_secret.assert_called_once_with( mock_client.put_secret_value.assert_called_once_with(
SecretId='test-secret', SecretId='test-secret',
SecretString='test-value' SecretString='test-value'
) )
@ -83,7 +83,7 @@ class TestAWSSecretsManagerProvider:
def test_store_secret_failure(self): def test_store_secret_failure(self):
provider, mock_client = self._provider() provider, mock_client = self._provider()
mock_client.update_secret.side_effect = Exception("AWS error") mock_client.put_secret_value.side_effect = Exception("AWS error")
with pytest.raises(Exception): with pytest.raises(Exception):
provider.store_secret('test-secret', 'test-value') provider.store_secret('test-secret', 'test-value')

8
tests/unittest/test_config_loader_secrets.py
View File

@ -7,7 +7,7 @@ from pr_agent.config_loader import apply_secrets_manager_config, apply_secrets_t
class TestConfigLoaderSecrets: class TestConfigLoaderSecrets:
def test_apply_secrets_manager_config_success(self): def test_apply_secrets_manager_config_success(self):
with patch('pr_agent.config_loader.get_secret_provider') as mock_get_provider, \ with patch('pr_agent.secret_providers.get_secret_provider') as mock_get_provider, \
patch('pr_agent.config_loader.apply_secrets_to_config') as mock_apply_secrets, \ patch('pr_agent.config_loader.apply_secrets_to_config') as mock_apply_secrets, \
patch('pr_agent.config_loader.get_settings') as mock_get_settings: patch('pr_agent.config_loader.get_settings') as mock_get_settings:
@ -26,14 +26,14 @@ class TestConfigLoaderSecrets:
mock_apply_secrets.assert_called_once_with({'openai.key': 'sk-test'}) mock_apply_secrets.assert_called_once_with({'openai.key': 'sk-test'})
def test_apply_secrets_manager_config_no_provider(self): def test_apply_secrets_manager_config_no_provider(self):
with patch('pr_agent.config_loader.get_secret_provider') as mock_get_provider: with patch('pr_agent.secret_providers.get_secret_provider') as mock_get_provider:
mock_get_provider.return_value = None mock_get_provider.return_value = None
# Confirm no exception is raised # Confirm no exception is raised
apply_secrets_manager_config() apply_secrets_manager_config()
def test_apply_secrets_manager_config_not_aws(self): def test_apply_secrets_manager_config_not_aws(self):
with patch('pr_agent.config_loader.get_secret_provider') as mock_get_provider, \ with patch('pr_agent.secret_providers.get_secret_provider') as mock_get_provider, \
patch('pr_agent.config_loader.get_settings') as mock_get_settings: patch('pr_agent.config_loader.get_settings') as mock_get_settings:
# Mock Google Cloud Storage provider # Mock Google Cloud Storage provider
@ -113,7 +113,7 @@ class TestConfigLoaderSecrets:
settings.set.assert_not_called() settings.set.assert_not_called()
def test_apply_secrets_manager_config_exception_handling(self): def test_apply_secrets_manager_config_exception_handling(self):
with patch('pr_agent.config_loader.get_secret_provider') as mock_get_provider: with patch('pr_agent.secret_providers.get_secret_provider') as mock_get_provider:
mock_get_provider.side_effect = Exception("Provider error") mock_get_provider.side_effect = Exception("Provider error")
# Confirm processing continues even when exception occurs # Confirm processing continues even when exception occurs