import yaml
import pytest

from twhatter.parser import TweetBase, User

@pytest.fixture(scope="session")
def fixtures_factory():
    """Factory for any kind of data that can be stored in YAML format"""
    def _fixtures_factory(yaml_file):
        with open(yaml_file, 'r') as f:
            fixtures = yaml.load(f)
        return fixtures

    return _fixtures_factory


@pytest.fixture(scope="session")
def tweets_factory(fixtures_factory):
    """Factory for tweets from YAML file"""
    def _tweets_factory(yaml_file):
        all_fixtures = fixtures_factory(yaml_file)
        return [t for t in all_fixtures if isinstance(t, TweetBase)]

    return _tweets_factory


@pytest.fixture(scope="session")
def users_factory(fixtures_factory):
    """Factory for tweets from YAML file"""
    def _users_factory(yaml_file):
        all_fixtures = fixtures_factory(yaml_file)
        return [u for u in all_fixtures if isinstance(u, User)]

    return _users_factory