google_workspace_mcp/tests/core/test_comments.py at bdc815a7238b2aa4bf3927dfc105f96e99b59171 · drewgillson/google_workspace_mcp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
"""Tests for core comments module."""

import sys
import os
import pytest
from unittest.mock import AsyncMock, Mock, patch

sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../..")))

from core.comments import _read_comments_impl


@pytest.mark.asyncio
async def test_read_comments_includes_quoted_text():
    """Verify that quotedFileContent.value is surfaced in the output."""
    mock_service = Mock()
    mock_service.comments.return_value.list.return_value.execute = Mock(
        return_value={
            "comments": [
                {
                    "id": "c1",
                    "content": "Needs a citation here.",
                    "author": {"displayName": "Alice"},
                    "createdTime": "2025-01-15T10:00:00Z",
                    "modifiedTime": "2025-01-15T10:00:00Z",
                    "resolved": False,
                    "quotedFileContent": {
                        "mimeType": "text/html",
                        "value": "the specific text that was highlighted",
                    },
                    "replies": [],
                },
                {
                    "id": "c2",
                    "content": "General comment without anchor.",
                    "author": {"displayName": "Bob"},
                    "createdTime": "2025-01-16T09:00:00Z",
                    "modifiedTime": "2025-01-16T09:00:00Z",
                    "resolved": False,
                    "replies": [],
                },
            ]
        }
    )

    result = await _read_comments_impl(mock_service, "document", "doc123")

    # Comment with anchor text should show the quoted text
    assert "Quoted text: the specific text that was highlighted" in result
    assert "Needs a citation here." in result

    # Comment without anchor text should not have a "Quoted text" line between Bob's author and content
    # The output uses literal \n joins, so split on that
    parts = result.split("\\n")
    bob_section_started = False
    for part in parts:
        if "Author: Bob" in part:
            bob_section_started = True
        if bob_section_started and "Quoted text:" in part:
            pytest.fail(
                "Comment without quotedFileContent should not show 'Quoted text'"
            )
        if bob_section_started and "Content: General comment" in part:
            break


@pytest.mark.asyncio
async def test_read_comments_empty():
    """Verify empty comments returns appropriate message."""
    mock_service = Mock()
    mock_service.comments.return_value.list.return_value.execute = Mock(
        return_value={"comments": []}
    )

    result = await _read_comments_impl(mock_service, "document", "doc123")
    assert "No comments found" in result


@pytest.mark.asyncio
async def test_read_comments_with_replies():
    """Verify replies are included in output."""
    mock_service = Mock()
    mock_service.comments.return_value.list.return_value.execute = Mock(
        return_value={
            "comments": [
                {
                    "id": "c1",
                    "content": "Question?",
                    "author": {"displayName": "Alice"},
                    "createdTime": "2025-01-15T10:00:00Z",
                    "modifiedTime": "2025-01-15T10:00:00Z",
                    "resolved": False,
                    "quotedFileContent": {"value": "some text"},
                    "replies": [
                        {
                            "id": "r1",
                            "content": "Answer.",
                            "author": {"displayName": "Bob"},
                            "createdTime": "2025-01-15T11:00:00Z",
                            "modifiedTime": "2025-01-15T11:00:00Z",
                        }
                    ],
                }
            ]
        }
    )

    result = await _read_comments_impl(mock_service, "document", "doc123")
    assert "Question?" in result
    assert "Answer." in result
    assert "Bob" in result
    assert "Quoted text: some text" in result