Mobilizon-Reshare-condividi-sui-social/tests/publishers/test_telegram.py

import pytest
import requests

from mobilizon_reshare.publishers.exceptions import (
    InvalidEvent,
    InvalidResponse,
    InvalidMessage,
)
from mobilizon_reshare.publishers.platforms.telegram import (
    TelegramFormatter,
    TelegramPublisher,
)


def test_message_length_success(event):
    message = "a" * 500
    event.description = message
    assert (
        TelegramFormatter().validate_message(
            TelegramFormatter().get_message_from_event(event)
        )
        is None
    )


def test_message_length_failure(event):
    message = "a" * 10000
    event.description = message

    with pytest.raises(InvalidMessage):
        TelegramFormatter().validate_message(
            TelegramFormatter().get_message_from_event(event)
        )


@pytest.mark.parametrize(
    "message, result",
    [
        ["", ""],
        ["a#b", "ab"],
        ["-", "\\-"],
        ["(", "\\("],
        ["!", "\\!"],
        [")", "\\)"],
        [")!", "\\)\\!"],
        ["[link](https://link.com)", "[link](https://link.com)"],
        [
            "[link](https://link.com) [link2](https://link.com)",
            "[link](https://link.com) [link2](https://link.com)",
        ],
    ],
)
def test_escape_message(message, result):
    assert TelegramFormatter().escape_message(message) == result


def test_event_validation(event):
    event.description = None
    with pytest.raises(InvalidEvent):
        TelegramFormatter().validate_event(event)


def test_validate_response():
    response = requests.Response()
    response.status_code = 200
    response._content = b"""{"ok":true}"""
    TelegramPublisher()._validate_response(response)


def test_validate_response_invalid_json():
    response = requests.Response()
    response.status_code = 200
    response._content = b"""{"osxsa"""
    with pytest.raises(InvalidResponse) as e:
        TelegramPublisher()._validate_response(response)

    e.match("json")


def test_validate_response_invalid_request():
    response = requests.Response()
    response.status_code = 400
    response._content = b"""{"error":true}"""
    with pytest.raises(InvalidResponse) as e:

        TelegramPublisher()._validate_response(response)

    e.match("400 Client Error")


def test_validate_response_invalid_response():
    response = requests.Response()
    response.status_code = 200
    response._content = b"""{"error":true}"""
    with pytest.raises(InvalidResponse) as e:

        TelegramPublisher()._validate_response(response)

    e.match("Invalid request")