kopia lustrzana https://github.com/bellingcat/auto-archiver
39 wiersze
1.3 KiB
Python
39 wiersze
1.3 KiB
Python
|
import os
|
||
|
|
from datetime import date
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from .test_extractor_base import TestExtractorBase
|
||
|
|
from auto_archiver.modules.telethon_extractor import TelethonExtractor
|
||
|
|
|
||
|
|
|
||
|
|
class TestTelethonExtractor(TestExtractorBase):
|
||
|
|
extractor_module = "telethon_extractor"
|
||
|
|
extractor: TelethonExtractor
|
||
|
|
config = {
|
||
|
|
"api_id": 123,
|
||
|
|
"api_hash": "ABC",
|
||
|
|
}
|
||
|
|
|
||
|
|
@pytest.fixture(autouse=True)
|
||
|
|
def mock_client_setup(self, mocker):
|
||
|
|
mocker.patch("telethon.client.auth.AuthMethods.start")
|
||
|
|
|
||
|
|
def test_setup_fails_clear_session_file(self, get_lazy_module, tmp_path, mocker):
|
||
|
|
start = mocker.patch("telethon.client.auth.AuthMethods.start")
|
||
|
|
start.side_effect = Exception("Test exception")
|
||
|
|
|
||
|
|
# make sure the default setup file is created
|
||
|
|
session_file = tmp_path / "test.session"
|
||
|
|
|
||
|
|
lazy_module = get_lazy_module("telethon_extractor")
|
||
|
|
|
||
|
|
with pytest.raises(Exception):
|
||
|
|
lazy_module.load(
|
||
|
|
{"telethon_extractor": {"session_file": str(session_file), "api_id": 123, "api_hash": "ABC"}}
|
||
|
|
)
|
||
|
|
|
||
|
|
assert session_file.exists()
|
||
|
|
assert f"telethon-{date.today().strftime('%Y-%m-%d')}" in lazy_module._instance.session_file
|
||
|
|
assert os.path.exists(lazy_module._instance.session_file + ".session")
|