def test_add_del(self, setup):
proxypool = ProxyPool()
p1 = Proxy(ip="127.0.0.1", port=80)
p2 = Proxy(ip="127.0.0.1", port=8080)
p3 = Proxy(ip="127.0.0.2", port=8080)
proxypool.add_proxy(p1)
assert len(ProxyModel.load_all()) == 1
from sqlalchemy.exc import IntegrityError
with pytest.raises(IntegrityError):
proxypool.add_proxy(p1)
proxypool.add_proxy(p2)
proxypool.add_proxy(p3)
assert proxypool.count() == 3
assert proxypool.has_proxy(p1.proxy_url())
proxypool.del_proxy(p1.proxy_url())
assert len(ProxyModel.load_all()) == 2
assert proxypool.count() == 2
proxypool.del_proxy(p2.proxy_url())
proxypool.del_proxy(p3.proxy_url())
assert proxypool.count() == 0
proxypool.del_proxy(p1.proxy_url())
