lubna-fatimaandClaude Sonnet 4.6
c2dec96b1f
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
"""Tests for multi-attachment URL collection."""
|
|
from __future__ import annotations
|
|
|
|
from onedrive_planner import (
|
|
_attachment_tab_count,
|
|
_merge_attachment_hrefs,
|
|
_should_mark_task_in_progress,
|
|
)
|
|
|
|
SAMPLE_A = (
|
|
"https://netorg393890.sharepoint.com/sites/ArborHills/Shared%20Documents/"
|
|
"RICHARDSON%20ARBRHLS%205.26.26.pdf"
|
|
)
|
|
SAMPLE_B = (
|
|
"https://netorg393890.sharepoint.com/sites/ArborHills/Shared%20Documents/"
|
|
"RICHARDSON%20ARBRHLS%206.4.26.pdf"
|
|
)
|
|
|
|
|
|
class TestAttachmentMulti:
|
|
def test_merge_keeps_distinct_pdf_urls(self):
|
|
merged = _merge_attachment_hrefs(
|
|
[SAMPLE_A],
|
|
[SAMPLE_B],
|
|
[SAMPLE_A],
|
|
)
|
|
assert len(merged) == 2
|
|
assert SAMPLE_A in merged
|
|
assert SAMPLE_B in merged
|
|
|
|
def test_merge_dedupes_same_filename(self):
|
|
merged = _merge_attachment_hrefs([SAMPLE_A], [SAMPLE_A + "?download=1"])
|
|
assert len(merged) == 1
|
|
|
|
def test_mark_status_only_after_all_files(self, tmp_path):
|
|
one = tmp_path / "a.pdf"
|
|
two = tmp_path / "b.pdf"
|
|
one.write_bytes(b"x")
|
|
two.write_bytes(b"y")
|
|
assert not _should_mark_task_in_progress([one], 2)
|
|
assert _should_mark_task_in_progress([one, two], 2)
|
|
assert not _should_mark_task_in_progress([], 2)
|
|
assert _should_mark_task_in_progress([one], 0)
|
|
|
|
def test_attachment_tab_count_parses_number(self):
|
|
from unittest.mock import MagicMock
|
|
|
|
tab = MagicMock()
|
|
tab.is_displayed.return_value = True
|
|
tab.text = "Attachments (2)"
|
|
tab.get_attribute.return_value = ""
|
|
driver = MagicMock()
|
|
driver.find_elements.return_value = [tab]
|
|
assert _attachment_tab_count(driver) == 2
|