proxy-pool/tests/integration/test_proxy_models.py

25 lines
650 B
Python

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