webstat-collector/tests/conftest.py

import asyncio
import pytest
from chweb.config import create_config
from chweb.models import Check


@pytest.fixture()
def config():
    config_dict = {
        'kafka': {
            'servers': ["localhost:9992"],
            'topic': "sample",
            'cafile': "/dev/null",
            'cert': "/dev/null",
            'key': "/dev/null",
            'passwd': "",
        },
        'postgres': {
            'dbhost': "localhost",
            'dbport': 5432,
            'dbname': "chweb",
            'dbuser': "vladan",
            'dbpass': "",
        },
        'sites': [{
            'url': "https://example.com",
            'regex': "aaaaaaaaaaaaa",
            'check_interval': 8,
        },
        ]
    }
    return create_config(config_dict)


@pytest.fixture
def config_invalid():
    config_dict = {
        'kafka': {
            'servers': ["localhost:9992"],
            'topic': "sample",
            'cafile': "",
            'cert': "",
            'key': "",
            'passwd': "",
        },
        'postgres': {
            'dbhost': "localhost",
            'dbport': 5432,
            'dbname': "chweb",
            'dbuser': "vladan",
            'dbpass': "",
            'dbcert': "",
        },
        'sites': [{
            'url': "https://dsadakjhkjsahkjh.com",
            'regex': "domain",
            'check_interval': 5,
        },
        ]
    }
    return create_config(config_dict)


@pytest.fixture
def check():
    return Check(
        domain="example.com",
        response_time=3265,
        status=200,
        url="https://example.com",
    )