pr-agent/tests/unittest/test_azure_devops_comment.py

import unittest
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):
        # Simulate config with no default_comment_status
        mock_settings = MagicMock()
        mock_settings.azure_devops.get.return_value = "closed"
        mock_settings.config.publish_output_progress = True
        mock_get_settings.return_value = mock_settings

        with patch.object(AzureDevopsProvider, "_get_azure_devops_client", return_value=(MagicMock(), MagicMock())):
            provider = AzureDevopsProvider()
            provider.workspace_slug = "ws"
            provider.repo_slug = "repo"
            provider.pr_num = 1

            # Patch CommentThread and create_thread
            with patch("pr_agent.git_providers.azuredevops_provider.CommentThread") as MockThread:
                provider.azure_devops_client.create_thread.return_value.comments = [MagicMock()]
                provider.azure_devops_client.create_thread.return_value.comments[0].thread_id = 123
                provider.azure_devops_client.create_thread.return_value.id = 123

                provider.publish_comment("test comment")
                args, kwargs = MockThread.call_args
                assert kwargs.get("status") == "closed"

    @patch("pr_agent.git_providers.azuredevops_provider.get_settings")
    def test_publish_comment_active(self, mock_get_settings):
        # Simulate config with default_comment_status = "active"
        mock_settings = MagicMock()
        mock_settings.azure_devops.get.return_value = "active"
        mock_settings.config.publish_output_progress = True
        mock_get_settings.return_value = mock_settings

        with patch.object(AzureDevopsProvider, "_get_azure_devops_client", return_value=(MagicMock(), MagicMock())):
            provider = AzureDevopsProvider()
            provider.workspace_slug = "ws"
            provider.repo_slug = "repo"
            provider.pr_num = 1

            # Patch CommentThread and create_thread
            with patch("pr_agent.git_providers.azuredevops_provider.CommentThread") as MockThread:
                provider.azure_devops_client.create_thread.return_value.comments = [MagicMock()]
                provider.azure_devops_client.create_thread.return_value.comments[0].thread_id = 123
                provider.azure_devops_client.create_thread.return_value.id = 123

                provider.publish_comment("test comment")
                args, kwargs = MockThread.call_args
                assert kwargs.get("status") == "active"

    def test_default_comment_status_from_config_file(self):
        # Import get_settings directly to read from configuration.toml
        status = get_settings().azure_devops.default_comment_status
        # The expected value should match what's in your configuration.toml
        self.assertEqual(status, "closed")