# Mocks
from requests.models import Response
import requests
import time
import csv
# Test Env
import pytest
from unittest.mock import patch
# Test functions
from baangt.base.ProxyRotate import ProxyRotate, proxy_data
import baangt.base.GlobalConstants as GC
from dataclasses import dataclass, asdict
from http import HTTPStatus
import copy
MIN_PROXIES_FOR_FIRST_RUN = 3
def create_init_proxy_list(length):
proxyList = []
for index in range(length):
proxy = asdict(proxy_data(ip = (str)(index), port = (str)(0)))
for key in proxy:
proxy[key] = str(proxy[key])
proxyList.append(proxy)
return proxyList
def init_ProxyRotate(mock_csv_DictReader, proxyListLength, firstRun = True):
proxyRotate = ProxyRotate()
proxyRotate.proxy_gather_link = "https://www.sslproxies.org/"
proxyRotate.proxies = {}
proxyRotate.all_proxies = {}
proxyRotate._ProxyRotate__temp_proxies = []
proxyRotate.firstRun = firstRun
proxyRotate.MIN_PROXIES_FOR_FIRST_RUN = MIN_PROXIES_FOR_FIRST_RUN
mock_csv_DictReader.return_value = create_init_proxy_list(proxyListLength)
proxyRotate._ProxyRotate__read_proxies()
return proxyRotate
def create_proxy_verification_response(status_code, length, deleteString=None):
ret = []
for index in range(length):
res_youtube = Response()
res_youtube._content = ""
res_youtube.status_code = status_code
res_google = copy.copy(res_youtube)
res_google._content = "