chore(deps): update pytest to 9.xx and add pytest-asyncio to fix some warnings in tests (#2222)

even-even · web-flow · commit 53a1b3909e5c · 2026-02-21T23:43:32.000+02:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -77,3 +77,12 @@ lint.ignore = ["E999", "B008"]
 exclude_dirs = ["tests"]
 skips = ["B101"]
 tests = []
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
+testpaths = ["tests"]
+python_files = ["test_*.py"]
+python_classes = ["Test*"]
+python_functions = ["test_*"]
+addopts = "--color=yes"
+console_output_style = "progress"
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -1,4 +1,5 @@
-pytest==7.4.0
+pytest==9.0.2
+pytest-asyncio>=1.3.0
 poetry
 twine
 pre-commit>=4,<5
diff --git a/requirements.txt b/requirements.txt
@@ -17,7 +17,8 @@ litellm==1.77.7
 loguru==0.7.2
 msrest==0.7.1
 openai>=1.55.3
-pytest==7.4.0
+pytest==9.0.2
+pytest-asyncio>=1.3.0
 PyGithub==1.59.*
 PyYAML==6.0.1
 python-gitlab==3.15.0
@@ -28,7 +29,7 @@ ujson==5.8.0
 uvicorn==0.22.0
 tenacity==8.2.3
 gunicorn==23.0.0
-pytest-cov==5.0.0
+pytest-cov==7.0.0
 pydantic==2.8.2
 html2text==2024.2.26
 giteapy==1.0.8
diff --git a/tests/e2e_tests/langchain_ai_handler.py b/tests/e2e_tests/langchain_ai_handler.py
@@ -1,9 +1,11 @@
 import asyncio
 import os
 import time
+
 from pr_agent.algo.ai_handlers.langchain_ai_handler import LangChainOpenAIHandler
 from pr_agent.config_loader import get_settings
 
+
 def check_settings():
     print('Checking settings...')
     settings = get_settings()
diff --git a/tests/e2e_tests/test_bitbucket_app.py b/tests/e2e_tests/test_bitbucket_app.py
@@ -11,10 +11,14 @@
 
 from pr_agent.config_loader import get_settings
 from pr_agent.log import get_logger, setup_logger
-from tests.e2e_tests.e2e_utils import (FILE_PATH,
-                                       IMPROVE_START_WITH_REGEX_PATTERN,
-                                       NEW_FILE_CONTENT, NUM_MINUTES,
-                                       PR_HEADER_START_WITH, REVIEW_START_WITH)
+from tests.e2e_tests.e2e_utils import (
+    FILE_PATH,
+    IMPROVE_START_WITH_REGEX_PATTERN,
+    NEW_FILE_CONTENT,
+    NUM_MINUTES,
+    PR_HEADER_START_WITH,
+    REVIEW_START_WITH,
+)
 
 log_level = os.environ.get("LOG_LEVEL", "INFO")
 setup_logger(log_level)
@@ -42,7 +46,7 @@ def test_e2e_run_bitbucket_app():
         target_repo = repo.branches.create(new_branch,source_branch.hash)
 
         # Update the file content
-        url = (f"https://api.bitbucket.org/2.0/repositories/{project_key}/{repo_slug}/src")
+        url = f"https://api.bitbucket.org/2.0/repositories/{project_key}/{repo_slug}/src"
         files={FILE_PATH: NEW_FILE_CONTENT}
         data={
             "message": "update cli_pip.py",
diff --git a/tests/e2e_tests/test_gitea_app.py b/tests/e2e_tests/test_gitea_app.py
@@ -1,14 +1,19 @@
 import os
 import time
-import requests
 from datetime import datetime
 
+import requests
+
 from pr_agent.config_loader import get_settings
 from pr_agent.log import get_logger, setup_logger
-from tests.e2e_tests.e2e_utils import (FILE_PATH,
-                                     IMPROVE_START_WITH_REGEX_PATTERN,
-                                     NEW_FILE_CONTENT, NUM_MINUTES,
-                                     PR_HEADER_START_WITH, REVIEW_START_WITH)
+from tests.e2e_tests.e2e_utils import (
+    FILE_PATH,
+    IMPROVE_START_WITH_REGEX_PATTERN,
+    NEW_FILE_CONTENT,
+    NUM_MINUTES,
+    PR_HEADER_START_WITH,
+    REVIEW_START_WITH,
+)
 
 log_level = os.environ.get("LOG_LEVEL", "INFO")
 setup_logger(log_level)
diff --git a/tests/e2e_tests/test_github_app.py b/tests/e2e_tests/test_github_app.py
@@ -6,10 +6,14 @@
 from pr_agent.config_loader import get_settings
 from pr_agent.git_providers import get_git_provider
 from pr_agent.log import get_logger, setup_logger
-from tests.e2e_tests.e2e_utils import (FILE_PATH,
-                                       IMPROVE_START_WITH_REGEX_PATTERN,
-                                       NEW_FILE_CONTENT, NUM_MINUTES,
-                                       PR_HEADER_START_WITH, REVIEW_START_WITH)
+from tests.e2e_tests.e2e_utils import (
+    FILE_PATH,
+    IMPROVE_START_WITH_REGEX_PATTERN,
+    NEW_FILE_CONTENT,
+    NUM_MINUTES,
+    PR_HEADER_START_WITH,
+    REVIEW_START_WITH,
+)
 
 log_level = os.environ.get("LOG_LEVEL", "INFO")
 setup_logger(log_level)
diff --git a/tests/e2e_tests/test_gitlab_webhook.py b/tests/e2e_tests/test_gitlab_webhook.py
@@ -8,10 +8,14 @@
 from pr_agent.config_loader import get_settings
 from pr_agent.git_providers import get_git_provider
 from pr_agent.log import get_logger, setup_logger
-from tests.e2e_tests.e2e_utils import (FILE_PATH,
-                                       IMPROVE_START_WITH_REGEX_PATTERN,
-                                       NEW_FILE_CONTENT, NUM_MINUTES,
-                                       PR_HEADER_START_WITH, REVIEW_START_WITH)
+from tests.e2e_tests.e2e_utils import (
+    FILE_PATH,
+    IMPROVE_START_WITH_REGEX_PATTERN,
+    NEW_FILE_CONTENT,
+    NUM_MINUTES,
+    PR_HEADER_START_WITH,
+    REVIEW_START_WITH,
+)
 
 log_level = os.environ.get("LOG_LEVEL", "INFO")
 setup_logger(log_level)
diff --git a/tests/health_test/main.py b/tests/health_test/main.py
@@ -4,12 +4,11 @@
 import os
 from pathlib import Path
 
-from starlette_context import request_cycle_context, context
+from starlette_context import context, request_cycle_context
 
+from pr_agent.agent.pr_agent import PRAgent, commands
 from pr_agent.cli import run_command
 from pr_agent.config_loader import get_settings, global_settings
-
-from pr_agent.agent.pr_agent import PRAgent, commands
 from pr_agent.log import get_logger, setup_logger
 from tests.e2e_tests import e2e_utils
 
diff --git a/tests/unittest/test_add_docs_trigger.py b/tests/unittest/test_add_docs_trigger.py
@@ -1,10 +1,11 @@
 import pytest
-from pr_agent.servers.github_app import handle_new_pr_opened
-from pr_agent.tools.pr_add_docs import PRAddDocs
+
 from pr_agent.agent.pr_agent import PRAgent
 from pr_agent.config_loader import get_settings
-from pr_agent.identity_providers.identity_provider import Eligibility
 from pr_agent.identity_providers import get_identity_provider
+from pr_agent.identity_providers.identity_provider import Eligibility
+from pr_agent.servers.github_app import handle_new_pr_opened
+from pr_agent.tools.pr_add_docs import PRAddDocs
 
 
 @pytest.mark.asyncio
diff --git a/tests/unittest/test_aws_secrets_manager_provider.py b/tests/unittest/test_aws_secrets_manager_provider.py
@@ -1,6 +1,7 @@
 import json
-import pytest
 from unittest.mock import MagicMock, patch
+
+import pytest
 from botocore.exceptions import ClientError
 
 from pr_agent.secret_providers.aws_secrets_manager_provider import AWSSecretsManagerProvider
diff --git a/tests/unittest/test_azure_devops_comment.py b/tests/unittest/test_azure_devops_comment.py
@@ -1,8 +1,10 @@
 import unittest
+from unittest.mock import MagicMock, patch
+
 from pr_agent.config_loader import get_settings
-from unittest.mock import patch, MagicMock
 from pr_agent.git_providers import AzureDevopsProvider
 
+
 class TestAzureDevopsProviderPublishComment(unittest.TestCase):
     @patch("pr_agent.git_providers.azuredevops_provider.get_settings")
     def test_publish_comment_default_closed(self, mock_get_settings):
diff --git a/tests/unittest/test_azure_devops_parsing.py b/tests/unittest/test_azure_devops_parsing.py
@@ -1,7 +1,7 @@
 from pr_agent.git_providers import AzureDevopsProvider
 
 
-class TestAzureDevOpsParsing():
+class TestAzureDevOpsParsing:
     def test_regular_address(self):
         pr_url = "https://dev.azure.com/organization/project/_git/repo/pullrequest/1"
 
diff --git a/tests/unittest/test_clip_tokens.py b/tests/unittest/test_clip_tokens.py
@@ -1,7 +1,9 @@
+from unittest.mock import MagicMock, patch
+
 import pytest
-from unittest.mock import patch, MagicMock
-from pr_agent.algo.utils import clip_tokens
+
 from pr_agent.algo.token_handler import TokenEncoder
+from pr_agent.algo.utils import clip_tokens
 
 
 class TestClipTokens:
diff --git a/tests/unittest/test_codecommit_provider.py b/tests/unittest/test_codecommit_provider.py
@@ -3,9 +3,7 @@
 import pytest
 
 from pr_agent.algo.types import EDIT_TYPE, FilePatchInfo
-from pr_agent.git_providers.codecommit_provider import (CodeCommitFile,
-                                                        CodeCommitProvider,
-                                                        PullRequestCCMimic)
+from pr_agent.git_providers.codecommit_provider import CodeCommitFile, CodeCommitProvider, PullRequestCCMimic
 
 
 class TestCodeCommitFile:
diff --git a/tests/unittest/test_config_loader_secrets.py b/tests/unittest/test_config_loader_secrets.py
@@ -1,4 +1,3 @@
-import pytest
 from unittest.mock import MagicMock, patch
 
 from pr_agent.config_loader import apply_secrets_manager_config, apply_secrets_to_config
diff --git a/tests/unittest/test_extend_patch.py b/tests/unittest/test_extend_patch.py
@@ -5,6 +5,7 @@
 from pr_agent.algo.token_handler import TokenHandler
 from pr_agent.algo.utils import load_large_diff
 from pr_agent.config_loader import get_settings
+
 get_settings(use_context=False).set("CONFIG.CLI_MODE", True)
 get_settings(use_context=False).config.allow_dynamic_context = False
 
diff --git a/tests/unittest/test_file_filter.py b/tests/unittest/test_file_filter.py
@@ -1,5 +1,3 @@
-import pytest
-
 from pr_agent.algo.file_filter import filter_ignored
 from pr_agent.config_loader import global_settings
 
diff --git a/tests/unittest/test_find_line_number_of_relevant_line_in_file.py b/tests/unittest/test_find_line_number_of_relevant_line_in_file.py
@@ -1,6 +1,4 @@
-
 # Generated by CodiumAI
-import pytest
 
 from pr_agent.algo.types import FilePatchInfo
 from pr_agent.algo.utils import find_line_number_of_relevant_line_in_file
diff --git a/tests/unittest/test_get_max_tokens.py b/tests/unittest/test_get_max_tokens.py
@@ -1,6 +1,8 @@
 import pytest
-from pr_agent.algo.utils import get_max_tokens, MAX_TOKENS
+
 import pr_agent.algo.utils as utils
+from pr_agent.algo.utils import MAX_TOKENS, get_max_tokens
+
 
 class TestGetMaxTokens:
 
diff --git a/tests/unittest/test_gitea_provider.py b/tests/unittest/test_gitea_provider.py
@@ -1,10 +1,6 @@
 from io import BytesIO
 from unittest.mock import MagicMock, patch
 
-import pytest
-
-from pr_agent.git_providers.gitea_provider import GiteaProvider
-
 
 class TestGiteaProvider:
     @patch('pr_agent.git_providers.gitea_provider.get_settings')
@@ -42,8 +38,6 @@ def call_api_side_effect(path, method, **kwargs):
 
         mock_api_client.call_api.side_effect = call_api_side_effect
 
-        import giteapy
-
         from pr_agent.git_providers.gitea_provider import RepoApi
 
         client = mock_api_client
diff --git a/tests/unittest/test_ignore_repositories.py b/tests/unittest/test_ignore_repositories.py
@@ -1,8 +1,10 @@
 import pytest
-from pr_agent.servers.github_app import should_process_pr_logic as github_should_process_pr_logic
+
+from pr_agent.config_loader import get_settings
 from pr_agent.servers.bitbucket_app import should_process_pr_logic as bitbucket_should_process_pr_logic
+from pr_agent.servers.github_app import should_process_pr_logic as github_should_process_pr_logic
 from pr_agent.servers.gitlab_webhook import should_process_pr_logic as gitlab_should_process_pr_logic
-from pr_agent.config_loader import get_settings
+
 
 def make_bitbucket_payload(full_name):
     return {
diff --git a/tests/unittest/test_litellm_reasoning_effort.py b/tests/unittest/test_litellm_reasoning_effort.py
@@ -1,7 +1,9 @@
+from unittest.mock import AsyncMock, MagicMock, call, patch
+
 import pytest
-from unittest.mock import patch, MagicMock, AsyncMock, call
-from pr_agent.algo.ai_handlers.litellm_ai_handler import LiteLLMAIHandler
+
 import pr_agent.algo.ai_handlers.litellm_ai_handler as litellm_handler
+from pr_agent.algo.ai_handlers.litellm_ai_handler import LiteLLMAIHandler
 
 
 def create_mock_settings(reasoning_effort_value):
diff --git a/tests/unittest/test_pr_update_changelog.py b/tests/unittest/test_pr_update_changelog.py
@@ -1,5 +1,7 @@
+from unittest.mock import AsyncMock, MagicMock, patch
+
 import pytest
-from unittest.mock import MagicMock, patch, AsyncMock
+
 from pr_agent.tools.pr_update_changelog import PRUpdateChangelog
 
 
diff --git a/tests/unittest/test_secret_provider_factory.py b/tests/unittest/test_secret_provider_factory.py
@@ -1,6 +1,7 @@
-import pytest
 from unittest.mock import MagicMock, patch
 
+import pytest
+
 from pr_agent.secret_providers import get_secret_provider
 
 
diff --git a/tests/unittest/test_try_fix_yaml.py b/tests/unittest/test_try_fix_yaml.py
@@ -1,6 +1,4 @@
-
 # Generated by CodiumAI
-import pytest
 
 from pr_agent.algo.utils import try_fix_yaml
 
