PrivacyScanData/archive/new2/bulk-downloader-for-reddit-development/tests/conftest.py

#!/usr/bin/env python3
# coding=utf-8

import configparser
import socket
from pathlib import Path

import praw
import pytest

from bdfr.oauth2 import OAuth2TokenManager


@pytest.fixture(scope='session')
def reddit_instance():
    rd = praw.Reddit(
        client_id='U-6gk4ZCh3IeNQ',
        client_secret='7CZHY6AmKweZME5s50SfDGylaPg',
        user_agent='test',
    )
    return rd


@pytest.fixture(scope='session')
def authenticated_reddit_instance():
    test_config_path = Path('./tests/test_config.cfg')
    if not test_config_path.exists():
        pytest.skip('Refresh token must be provided to authenticate with OAuth2')
    cfg_parser = configparser.ConfigParser()
    cfg_parser.read(test_config_path)
    if not cfg_parser.has_option('DEFAULT', 'user_token'):
        pytest.skip('Refresh token must be provided to authenticate with OAuth2')
    token_manager = OAuth2TokenManager(cfg_parser, test_config_path)
    reddit_instance = praw.Reddit(
        client_id=cfg_parser.get('DEFAULT', 'client_id'),
        client_secret=cfg_parser.get('DEFAULT', 'client_secret'),
        user_agent=socket.gethostname(),
        token_manager=token_manager,
    )
    return reddit_instance