1import pytest
2import os
3 
4os.environ["TELEGRAM_API_ID"] = "12345"
5os.environ["TELEGRAM_API_HASH"] = "dummy_hash"
6from main import validate_id, ValidationError, log_and_format_error
7from functools import wraps
8import asyncio
9from typing import Union, List, Optional
10 
11 
12# A simple async function to be decorated for testing
13@validate_id("user_id", "chat_id", "user_ids")
14async def dummy_function(**kwargs):
15    return "success", kwargs
16 
17 
18@pytest.mark.asyncio
19async def test_valid_integer_id():
20    result, kwargs = await dummy_function(user_id=12345)
21    assert result == "success"
22    assert kwargs["user_id"] == 12345
23 
24 
25@pytest.mark.asyncio
26async def test_valid_negative_integer_id():
27    result, kwargs = await dummy_function(chat_id=-100123456)
28    assert result == "success"
29    assert kwargs["chat_id"] == -100123456
30 
31 
32@pytest.mark.asyncio
33async def test_valid_string_integer_id():
34    result, kwargs = await dummy_function(user_id="12345")
35    assert result == "success"
36    assert kwargs["user_id"] == 12345
37 
38 
39@pytest.mark.asyncio
40async def test_valid_username():
41    result, kwargs = await dummy_function(user_id="@test_user")
42    assert result == "success"
43    assert kwargs["user_id"] == "@test_user"
44 
45 
46@pytest.mark.asyncio
47async def test_valid_username_without_at():
48    result, kwargs = await dummy_function(user_id="test_user_long_enough")
49    assert result == "success"
50    assert kwargs["user_id"] == "test_user_long_enough"
51 
52 
53@pytest.mark.asyncio
54async def test_valid_list_of_ids():
55    result, kwargs = await dummy_function(user_ids=[123, "456", "@test_user"])
56    assert result == "success"
57    assert kwargs["user_ids"] == [123, 456, "@test_user"]
58 
59 
60@pytest.mark.asyncio
61async def test_invalid_float_id():
62    result = await dummy_function(user_id=123.45)
63    assert "Invalid user_id" in result
64    assert "Type must be an integer or a string" in result
65 
66 
67@pytest.mark.asyncio
68async def test_invalid_string_id():
69    result = await dummy_function(user_id="inv")  # too short
70    assert "Invalid user_id" in result
71    assert "Must be a valid integer ID, or a username string" in result
72 
73 
74@pytest.mark.asyncio
75async def test_integer_out_of_range():
76    result = await dummy_function(user_id=2**64)
77    assert "Invalid user_id" in result
78    assert "out of the valid integer range" in result
79 
80 
81@pytest.mark.asyncio
82async def test_invalid_item_in_list():
83    result = await dummy_function(user_ids=[123, "456", 123.45])
84    assert "Invalid user_ids" in result
85    assert "Type must be an integer or a string" in result
86 
87 
88@pytest.mark.asyncio
89async def test_no_id_provided():
90    result, kwargs = await dummy_function()
91    assert result == "success"
92 
93 
94@pytest.mark.asyncio
95async def test_none_id_provided():
96    result, kwargs = await dummy_function(user_id=None)
97    assert result == "success"
98