mirror of
https://github.com/bellingcat/cisticola.git
synced 2026-06-08 03:18:34 +03:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from sqlalchemy.orm import sessionmaker
|
|
import json
|
|
|
|
import pytest
|
|
|
|
from cisticola.base import Channel
|
|
from cisticola.scraper import VkontakteScraper
|
|
from cisticola.transformer import VkontakteTransformer
|
|
from cisticola.base import Post, Media
|
|
|
|
@pytest.mark.media
|
|
def test_scrape_etl_vkontakte(engine, controller, etl_controller, channel_kwargs):
|
|
controller.reset_db()
|
|
|
|
channels = [Channel(**channel_kwargs['vkontakte'])]
|
|
controller.register_scraper(scraper = VkontakteScraper())
|
|
controller.scrape_channels(channels = channels, archive_media = True)
|
|
controller.scrape_all_channel_info()
|
|
|
|
etl_controller.register_transformer(VkontakteTransformer())
|
|
etl_controller.transform_all_untransformed()
|
|
etl_controller.transform_all_untransformed_info()
|
|
|
|
sessionfactory = sessionmaker()
|
|
sessionfactory.configure(bind=engine)
|
|
session = sessionfactory()
|
|
|
|
posts = session.query(Post).all()
|
|
media = session.query(Media).all()
|
|
|
|
assert len(posts) == 23
|
|
# assert len(media) == 0
|
|
|
|
assert 'Nigerian gender studies' in posts[-1].content
|
|
# assert json.loads(media[0].exif)['Composite:ImageSize'] == "826 728" |