from textwrap import dedent

import pytest
from bs4 import BeautifulSoup

from ows_text_campaigns.campaigns.enums import ShorteningMethod
from ows_text_campaigns.campaigns.exceptions import InvalidShortLinkTagError
from ows_text_campaigns.campaigns.types import ShortenedUrl
from ows_text_campaigns.campaigns.utils import (
    extract_shortened_urls_from_markup,
    replace_short_link_tags,
    replace_standard_links_tags,
)


def test_extract_shortened_urls_from_content() -> None:
    input_html = """
    <a
        data-short-link-id="url1"
        data-short-link-url="https://sme.com/test1"
        data-short-link-method="STANDARD"
        data-short-link-path="path1"
        data-short-link-domain="sme.com"
    >link1</a>
    <a
        data-short-link-id="url2"
        data-short-link-url="https://sme.com/test2"
        data-short-link-method="STANDARD"
        data-short-link-path="path2"
        data-short-link-domain="sme.com"
    >link2</a>
    """

    result = extract_shortened_urls_from_markup(input_html)

    assert result == [
        ShortenedUrl(
            id="url1",
            url="https://sme.com/test1",
            method=ShorteningMethod.STANDARD,
            domain="sme.com",
            path="path1",
        ),
        ShortenedUrl(
            id="url2",
            url="https://sme.com/test2",
            method=ShorteningMethod.STANDARD,
            domain="sme.com",
            path="path2",
        ),
    ]


def test_extract_shortened_url_from_invalid_content() -> None:
    input_html = """
    <a
        data-short-link-id="url1"
        data-short-link-url="https://sme.com/test1"
        data-short-link-method="UNSUPPORTED"
        data-short-link-path="path1"
    >link</a>
    """

    with pytest.raises(InvalidShortLinkTagError) as exc_info:
        extract_shortened_urls_from_markup(input_html, raise_on_error=True)

    assert exc_info.value.message == (
        "Missing or invalid `data-short-link-*` attribute(s) in short link tag: "
        "method, domain"
    )


def test_replace_short_link_tags() -> None:
    input_html = dedent("""
    <p>
    <a data-short-link-id="url1" href="https://sme.com/test1">link1</a>
    <a data-short-link-id="url2" href="https://sme.com/test2">link2</a>
    </p>
    """)

    result = replace_short_link_tags(
        input_html,
        shortened_urls={
            "url1": "https://sme.com/link-abc1",
            "url2": "https://sme.com/link-abc2",
        },
    )

    expected = dedent("""
    <p>
    <a href="https://sme.com/link-abc1">https://sme.com/link-abc1</a>
    <a href="https://sme.com/link-abc2">https://sme.com/link-abc2</a>
    </p>
    """)

    assert result == expected


def test_replace_standard_links_tags() -> None:
    input_html = """
    <p>
    <a
      data-short-link-id="url1"
      data-short-link-method="STANDARD"
      data-short-link-url="https://artist.com/test1"
      href="https://artist.com/test1">
      https://sme.com/test1
    </a>
    <a
      data-short-link-id="url2"
      data-short-link-method="PERSONALIZED"
      data-short-link-url="https://artist.com/test2"
      href="https://artist.com/test2">
      https://sme.com/test2
    </a>
    </p>
"""

    expected_html = """
    <p>
    <a href="https://artist.com/test1">https://artist.com/test1</a>
    <a data-short-link-id="url2" data-short-link-method="PERSONALIZED" data-short-link-url="https://artist.com/test2" href="https://artist.com/test2">
      https://sme.com/test2
    </a>
    </p>
    """
    expected_soup = BeautifulSoup(expected_html, "html.parser")

    result = replace_standard_links_tags(input_html)
    result_soup = BeautifulSoup(result, "html.parser")

    assert result_soup.prettify() == expected_soup.prettify()