def test_filter_servers_good(): servers_stub = get_servers_stub() # test default features argument is None try: actual_servers = features.filter_servers(servers_stub) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113", "us2853"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test default features argument is empyt list try: actual_servers = features.filter_servers(servers_stub) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113", "us2853"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test multiple features are good try: actual_servers = features.filter_servers( servers_stub, ["openvpn_udp", "openvpn_tcp"]) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113", "us2853"]) assert actual_servers == expected_servers except features.FeatureError: assert False
def test_filter_servers_with_every_type(): servers_stub = get_servers_stub() # test with double try: actual_servers = categories.filter_servers(servers_stub, ["double"]) expected_servers = get_expected_servers_by_domain(["us-ca5"]) assert actual_servers == expected_servers except categories.CategoriesError: assert False # test with dedicated try: actual_servers = categories.filter_servers(servers_stub, ["dedicated"]) expected_servers = get_expected_servers_by_domain(["us2853"]) assert actual_servers == expected_servers except categories.CategoriesError: assert False # test with standard try: actual_servers = categories.filter_servers(servers_stub, ["standard"]) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113"]) assert actual_servers == expected_servers except categories.CategoriesError: assert False # test with p2p try: actual_servers = categories.filter_servers(servers_stub, ["p2p"]) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113"]) assert actual_servers == expected_servers except categories.CategoriesError: assert False # test with obfuscated try: actual_servers = categories.filter_servers(servers_stub, ["obfuscated"]) expected_servers = get_expected_servers_by_domain( ["nl80", "nl81", "nl82"]) assert actual_servers == expected_servers except categories.CategoriesError: assert False
def test_filter_servers_when_match_is_default(): servers_ = get_servers_stub() actual_servers = filter_servers(servers_, 10) domains = list() domains.append("nl80") expected_servers = get_expected_servers_by_domain(domains) assert actual_servers == expected_servers
def test_max_load_filter_when_load_is_valid(): servers_ = get_servers_stub() actual_servers = MaxLoadFilter(servers_).apply(10) domains = list() domains.append("nl80") expected_servers = get_expected_servers_by_domain(domains) assert actual_servers == expected_servers
def test_filter_servers_when_match_is_exact(): servers_ = get_servers_stub() actual_servers = filter_servers(servers_, 20, "exact") domains = list() domains.append("de112") domains.append("de113") expected_servers = get_expected_servers_by_domain(domains) assert actual_servers == expected_servers
def test_load_filter_when_load_is_valid(): servers_ = get_servers_stub() domains = list() domains.append("de112") domains.append("de113") expected_servers = get_expected_servers_by_domain(domains) actual_servers = LoadFilter(servers_).apply(20) assert actual_servers == expected_servers
def test_filter_servers_good(): servers_stub = get_servers_stub() # test default categories with no argument try: actual_servers = categories.filter_servers(servers_stub) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113"]) assert actual_servers == expected_servers except categories.CategoriesError: assert False # test default type argument with empty list try: actual_servers = categories.filter_servers(servers_stub, list()) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113"]) assert actual_servers == expected_servers except categories.CategoriesError: assert False # test multiple categories try: actual_servers = categories.filter_servers(servers_stub, ["standard", "p2p"]) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113"]) assert actual_servers == expected_servers except categories.CategoriesError: assert False # test all categories at once should return empty list try: actual_servers = categories.filter_servers( servers_stub, ["double", "dedicated", "standard", "p2p", "obfuscated"]) assert actual_servers == list() except categories.CategoriesError: assert False
def test_filter_servers_when_match_is_min(): servers_ = get_servers_stub() actual_servers = filter_servers(servers_, 20, "min") domains = list() domains.append("de111") domains.append("de112") domains.append("de113") domains.append("nl81") domains.append("nl82") domains.append("us-ca5") domains.append("us2853") expected_servers = get_expected_servers_by_domain(domains) assert actual_servers == expected_servers
def test_min_load_filter_when_load_is_valid(): servers_ = get_servers_stub() actual_servers = MinLoadFilter(servers_).apply(20) domains = list() domains.append("de111") domains.append("de112") domains.append("de113") domains.append("nl81") domains.append("nl82") domains.append("us-ca5") domains.append("us2853") expected_servers = get_expected_servers_by_domain(domains) assert actual_servers == expected_servers
def test_filter_servers_good(): servers_stub = get_servers_stub() # test param when param is not given try: actual_servers = countries.filter_servers(servers_stub) expected_servers = servers_stub assert actual_servers == expected_servers except countries.CountryError: assert False # test param when param is an empty list try: actual_servers = countries.filter_servers(servers_stub) expected_servers = servers_stub assert actual_servers == expected_servers except countries.CountryError: assert False # test param with multiple valid countries try: actual_servers = countries.filter_servers(servers_stub, ["de", "nl"]) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113", "nl80", "nl81", "nl82"] ) assert actual_servers == expected_servers except countries.CountryError: assert False # test with uppercase country code try: actual_servers = countries.filter_servers(servers_stub, ["DE"]) expected_servers = get_expected_servers_by_domain(["de111", "de112", "de113"]) assert actual_servers == expected_servers except countries.CountryError: assert False # test with mixedcase country code try: actual_servers = countries.filter_servers(servers_stub, ["De"]) expected_servers = get_expected_servers_by_domain(["de111", "de112", "de113"]) assert actual_servers == expected_servers except countries.CountryError: assert False # test with multiple mixedcase country codes try: actual_servers = countries.filter_servers(servers_stub, ["De", "nL"]) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113", "nl80", "nl81", "nl82"] ) assert actual_servers == expected_servers except countries.CountryError: assert False # test param: server is empty try: actual_servers = countries.filter_servers(list(), ["de"]) assert actual_servers == list() except BaseException: assert False
def test_filter_servers_with_every_feature(): servers_stub = get_servers_stub() # test with ikev2 try: actual_servers = features.filter_servers(servers_stub, ["ikev2"]) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with openvpn_udp try: actual_servers = features.filter_servers(servers_stub, ["openvpn_udp"]) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113", "us2853"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with openvpn_tcp try: actual_servers = features.filter_servers(servers_stub, ["openvpn_tcp"]) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113", "us-ca5", "us2853"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with socks try: actual_servers = features.filter_servers(servers_stub, ["socks"]) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with proxy try: actual_servers = features.filter_servers(servers_stub, ["proxy"]) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with pptp try: actual_servers = features.filter_servers(servers_stub, ["pptp"]) assert actual_servers == list() except features.FeatureError: assert False # test with l2tp try: actual_servers = features.filter_servers(servers_stub, ["l2tp"]) assert actual_servers == list() except features.FeatureError: assert False # test with openvpn_xor_udp try: actual_servers = features.filter_servers(servers_stub, ["openvpn_xor_udp"]) expected_servers = get_expected_servers_by_domain( ["nl80", "nl81", "nl82"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with openvpn_xor_tcp try: actual_servers = features.filter_servers(servers_stub, ["openvpn_xor_tcp"]) expected_servers = get_expected_servers_by_domain( ["nl80", "nl81", "nl82"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with proxy_cybersec try: actual_servers = features.filter_servers(servers_stub, ["proxy_cybersec"]) expected_servers = get_expected_servers_by_domain( ["de111", "de112", "de113"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with proxy_ssl try: actual_servers = features.filter_servers(servers_stub, ["proxy_ssl"]) expected_servers = get_expected_servers_by_domain(["de112", "de113"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with proxy_ssl_cybersec try: actual_servers = features.filter_servers(servers_stub, ["proxy_ssl_cybersec"]) expected_servers = get_expected_servers_by_domain(["de111", "de112"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with ikev2_v6 try: actual_servers = features.filter_servers(servers_stub, ["ikev2_v6"]) expected_servers = get_expected_servers_by_domain(["nl82"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with openvpn_udp_v6 try: actual_servers = features.filter_servers(servers_stub, ["openvpn_udp_v6"]) expected_servers = get_expected_servers_by_domain(["nl82"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with openvpn_tcp_v6 try: actual_servers = features.filter_servers(servers_stub, ["openvpn_tcp_v6"]) expected_servers = get_expected_servers_by_domain(["nl82"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with wireguard_udp try: actual_servers = features.filter_servers(servers_stub, ["wireguard_udp"]) expected_servers = get_expected_servers_by_domain(["nl82"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with openvpn_udp_tls_crypt try: actual_servers = features.filter_servers(servers_stub, ["openvpn_udp_tls_crypt"]) expected_servers = get_expected_servers_by_domain(["nl82"]) assert actual_servers == expected_servers except features.FeatureError: assert False # test with openvpn_tcp_tls_crypt try: actual_servers = features.filter_servers(servers_stub, ["openvpn_tcp_tls_crypt"]) expected_servers = get_expected_servers_by_domain(["nl82"]) assert actual_servers == expected_servers except features.FeatureError: assert False