bellingcat/cisticola
1
0
Fork 0
mirror of https://github.com/bellingcat/cisticola.git synced 2026-06-12 21:38:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

fixed some issues with Telegram tests

Browse Source
This commit is contained in:
Tristan Lee
2023-07-27 13:07:44 -05:00
parent 99cc4d80b2
commit 249f411a1d
4 changed files with 12 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
tests/scraper/telegram_snscrape.py
View File

@@ -1,33 +0,0 @@
import pytest
from cisticola.base import Channel
from cisticola.scraper import TelegramSnscrapeScraper
@pytest.mark.unarchived
def test_scrape_telegram_snscrape_channel_no_media(controller, channel_kwargs):
channels = [Channel(**channel_kwargs['telegram'])]
controller.register_scraper(scraper = TelegramSnscrapeScraper())
controller.scrape_channels(channels = channels, archive_media = False)
@pytest.mark.media
@pytest.mark.unarchived
def test_scrape_telegram_snscrape_channel_unarchived_media(controller):
controller.archive_unarchived_media()
@pytest.mark.media
def test_scrape_telegram_snscrape_channel(controller, channel_kwargs):
controller.reset_db()
channels = [Channel(**channel_kwargs['telegram'])]
controller.register_scraper(scraper = TelegramSnscrapeScraper())
controller.scrape_channels(channels = channels, archive_media = True)
@pytest.mark.profile
def test_scrape_telegram_snscrape_profile(channel_kwargs):
scraper = TelegramSnscrapeScraper()
channel = Channel(**channel_kwargs['telegram'])
scraper.get_profile(channel=channel)