mirror of
https://github.com/soxoj/maigret.git
synced 2026-05-06 22:19:01 +00:00
Soxoj
121 lines
3.8 KiB
Python
121 lines
3.8 KiB
Python
"""Maigret main module test functions"""
|
|
|
|
import asyncio
|
|
import copy
|
|
|
|
import pytest
|
|
from mock import Mock
|
|
|
|
from maigret.maigret import self_check, maigret
|
|
from maigret.maigret import (
|
|
extract_ids_from_page,
|
|
extract_ids_from_results,
|
|
)
|
|
from maigret.sites import MaigretSite
|
|
from maigret.result import MaigretCheckResult, MaigretCheckStatus
|
|
from tests.conftest import RESULTS_EXAMPLE
|
|
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.asyncio
|
|
async def test_self_check_db(test_db):
|
|
# initalize logger to debug
|
|
logger = Mock()
|
|
|
|
assert test_db.sites_dict['InvalidActive'].disabled is False
|
|
assert test_db.sites_dict['ValidInactive'].disabled is True
|
|
assert test_db.sites_dict['ValidActive'].disabled is False
|
|
assert test_db.sites_dict['InvalidInactive'].disabled is True
|
|
|
|
await self_check(test_db, test_db.sites_dict, logger, silent=False)
|
|
|
|
assert test_db.sites_dict['InvalidActive'].disabled is True
|
|
assert test_db.sites_dict['ValidInactive'].disabled is False
|
|
assert test_db.sites_dict['ValidActive'].disabled is False
|
|
assert test_db.sites_dict['InvalidInactive'].disabled is True
|
|
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.skip(reason="broken, fixme")
|
|
def test_maigret_results(test_db):
|
|
logger = Mock()
|
|
|
|
username = 'Skyeng'
|
|
loop = asyncio.get_event_loop()
|
|
results = loop.run_until_complete(
|
|
maigret(username, site_dict=test_db.sites_dict, logger=logger, timeout=30)
|
|
)
|
|
|
|
assert isinstance(results, dict)
|
|
|
|
reddit_site = results['Reddit']['site']
|
|
assert isinstance(reddit_site, MaigretSite)
|
|
|
|
assert reddit_site.json == {
|
|
'tags': ['news', 'social', 'us'],
|
|
'checkType': 'status_code',
|
|
'presenseStrs': ['totalKarma'],
|
|
'disabled': True,
|
|
'alexaRank': 17,
|
|
'url': 'https://www.reddit.com/user/{username}',
|
|
'urlMain': 'https://www.reddit.com/',
|
|
'usernameClaimed': 'blue',
|
|
'usernameUnclaimed': 'noonewouldeverusethis7',
|
|
}
|
|
|
|
del results['Reddit']['site']
|
|
del results['GooglePlayStore']['site']
|
|
|
|
reddit_status = results['Reddit']['status']
|
|
assert isinstance(reddit_status, MaigretCheckResult)
|
|
assert reddit_status.status == MaigretCheckStatus.ILLEGAL
|
|
|
|
playstore_status = results['GooglePlayStore']['status']
|
|
assert isinstance(playstore_status, MaigretCheckResult)
|
|
assert playstore_status.status == MaigretCheckStatus.CLAIMED
|
|
|
|
del results['Reddit']['status']
|
|
del results['GooglePlayStore']['status']
|
|
|
|
assert results['Reddit'].get('future') is None
|
|
del results['GooglePlayStore']['future']
|
|
del results['GooglePlayStore']['checker']
|
|
|
|
assert results == RESULTS_EXAMPLE
|
|
|
|
|
|
@pytest.mark.slow
|
|
def test_extract_ids_from_url(default_db):
|
|
assert default_db.extract_ids_from_url('https://www.reddit.com/user/test') == {
|
|
'test': 'username'
|
|
}
|
|
assert default_db.extract_ids_from_url('https://vk.com/id123') == {'123': 'vk_id'}
|
|
assert default_db.extract_ids_from_url('https://vk.com/ida123') == {
|
|
'ida123': 'username'
|
|
}
|
|
assert default_db.extract_ids_from_url(
|
|
'https://my.mail.ru/yandex.ru/dipres8904/'
|
|
) == {'dipres8904': 'username'}
|
|
assert default_db.extract_ids_from_url(
|
|
'https://reviews.yandex.ru/user/adbced123'
|
|
) == {'adbced123': 'yandex_public_id'}
|
|
|
|
|
|
@pytest.mark.slow
|
|
def test_extract_ids_from_page(test_db):
|
|
logger = Mock()
|
|
extract_ids_from_page('https://www.reddit.com/user/test', logger) == {
|
|
'test': 'username'
|
|
}
|
|
|
|
|
|
def test_extract_ids_from_results(test_db):
|
|
TEST_EXAMPLE = copy.deepcopy(RESULTS_EXAMPLE)
|
|
TEST_EXAMPLE['Reddit']['ids_usernames'] = {'test1': 'yandex_public_id'}
|
|
TEST_EXAMPLE['Reddit']['ids_links'] = ['https://www.reddit.com/user/test2']
|
|
|
|
extract_ids_from_results(TEST_EXAMPLE, test_db) == {
|
|
'test1': 'yandex_public_id',
|
|
'test2': 'username',
|
|
}
|