from proxy_pool.proxy.models import Proxy, ProxyProtocol, ProxySource, ProxyStatus


async def test_create_source_and_proxy(db_session):
    source = ProxySource(
        url="https://example.com/proxies.txt",
        parser_name="plaintext",
    )
    db_session.add(source)
    await db_session.flush()

    proxy = Proxy(
        ip="203.0.113.42",
        port=8080,
        protocol=ProxyProtocol.HTTP,
        source_id=source.id,
    )
    db_session.add(proxy)
    await db_session.flush()

    assert proxy.id is not None
    assert proxy.status == ProxyStatus.UNCHECKED
    assert proxy.score == 0.0
    assert proxy.source_id == source.id