refactor: IP代理池重构

This commit is contained in:
Relakkes
2024-04-05 10:44:05 +08:00
parent d0c578c2bf
commit dde3c0429e
11 changed files with 178 additions and 244 deletions

View File

@@ -5,13 +5,14 @@
from unittest import IsolatedAsyncioTestCase
from proxy.proxy_ip_pool import create_ip_pool
from proxy.proxy_ip_provider import IpInfoModel
from proxy.types import IpInfoModel
class TestIpPool(IsolatedAsyncioTestCase):
async def test_ip_pool(self):
pool = await create_ip_pool(ip_pool_count=3, enable_validate_ip=True)
for i in range(1):
pool = await create_ip_pool(ip_pool_count=1, enable_validate_ip=True)
print("\n")
for i in range(3):
ip_proxy_info: IpInfoModel = await pool.get_proxy()
print(ip_proxy_info)
self.assertIsNotNone(ip_proxy_info.ip, msg="验证 ip 是否获取成功")