mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-11 09:55:21 +00:00
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from aiogram.utils.token import TokenValidationError, validate_token, extract_bot_id
|
||
|
|
|
||
|
|
BOT_ID = 123456789
|
||
|
|
VALID_TOKEN = '123456789:AABBCCDDEEFFaabbccddeeff-1234567890'
|
||
|
|
INVALID_TOKENS = [
|
||
|
|
'123456789:AABBCCDDEEFFaabbccddeeff 123456789', # space is exists
|
||
|
|
'ABC:AABBCCDDEEFFaabbccddeeff123456789', # left part is not digit
|
||
|
|
':AABBCCDDEEFFaabbccddeeff123456789', # there is no left part
|
||
|
|
'123456789:', # there is no right part
|
||
|
|
'ABC AABBCCDDEEFFaabbccddeeff123456789', # there is no ':' separator
|
||
|
|
None, # is None
|
||
|
|
12345678, # is digit
|
||
|
|
(42, 'TEST'), # is tuple
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(params=INVALID_TOKENS, name='invalid_token')
|
||
|
|
def invalid_token_fixture(request):
|
||
|
|
return request.param
|
||
|
|
|
||
|
|
|
||
|
|
class TestCheckToken:
|
||
|
|
def test_valid(self):
|
||
|
|
assert validate_token(VALID_TOKEN) is True
|
||
|
|
|
||
|
|
def test_invalid_token(self, invalid_token):
|
||
|
|
with pytest.raises(TokenValidationError):
|
||
|
|
validate_token(invalid_token)
|
||
|
|
|
||
|
|
|
||
|
|
class TestExtractBotId:
|
||
|
|
def test_extract_bot_id(self):
|
||
|
|
with patch("aiogram.utils.token.validate_token") as mocked_validate_token:
|
||
|
|
result = extract_bot_id(VALID_TOKEN)
|
||
|
|
|
||
|
|
mocked_validate_token.assert_called_once_with(VALID_TOKEN)
|
||
|
|
assert result == BOT_ID
|