def test_initialization_correct_order(self):
config = {
"providers": [
{
"name": "vova",
"type": "vpn",
"priority": 5,
"countries": [],
"params": {},
},
{
"name": "pita",
"type": "vpn",
"priority": 1,
"countries": [],
"params": {},
},
{
"name": "misha",
"type": "vpn",
"priority": 1,
"countries": [],
"params": {},
},
{
"name": "baba",
"type": "luminati",
"priority": 10,
"countries": [],
"username": "******",
"password": "******",
}
]
}
teleport = Teleport(config)
providers = teleport.get_sorted_providers()
# pita and misha are highest priority so they
# go first, the order between them is random
# then vova, and finally baba
names = [p.name for p in providers]
expected_order_a = ["pita", "misha", "vova", "baba"]
expected_order_b = ["misha", "pita", "vova", "baba"]
if names != expected_order_a and names != expected_order_b:
self.fail("providers in incorrect order")
def test_initialization_correct_order(self):
config = {
"providers": [{
"name": "vova",
"type": "vpn",
"priority": 5,
"countries": [],
"params": {},
}, {
"name": "pita",
"type": "vpn",
"priority": 1,
"countries": [],
"params": {},
}, {
"name": "misha",
"type": "vpn",
"priority": 1,
"countries": [],
"params": {},
}, {
"name": "baba",
"type": "luminati",
"priority": 10,
"countries": [],
"username": "******",
"password": "******",
}]
}
teleport = Teleport(config)
providers = teleport.get_sorted_providers()
# pita and misha are highest priority so they
# go first, the order between them is random
# then vova, and finally baba
names = [p.name for p in providers]
expected_order_a = ["pita", "misha", "vova", "baba"]
expected_order_b = ["misha", "pita", "vova", "baba"]
if names != expected_order_a and names != expected_order_b:
self.fail("providers in incorrect order")
