SeleniumBase/examples/proxy_test.py at fa5364805e53670d2696b3c897ebc1c18af1da7d · seleniumbase/SeleniumBase · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
from seleniumbase.config import settings
from seleniumbase import BaseCase
BaseCase.main(__name__, __file__)


class ProxyTests(BaseCase):
    def test_proxy(self):
        if self.headless or self.recorder_mode or self.browser == "safari":
            self.open_if_not_url("about:blank")
            print("\n  Unsupported mode for this test.")
            self.skip("Unsupported mode for this test.")
        settings.SKIP_JS_WAITS = True
        if not self.page_load_strategy == "none" and not self.undetectable:
            # This page takes too long to load otherwise
            self.get_new_driver(page_load_strategy="none")
        self.open("https://api.ipify.org/")
        ip_address = self.get_text("body")
        if "ERR" in ip_address:
            raise Exception("Failed to determine IP Address!")
        print("\n\nMy IP Address = %s\n" % ip_address)
        self.open("https://ipinfo.io/%s" % ip_address)
        self.sleep(2)
        self.wait_for_text(ip_address, "h1", timeout=20)
        self.wait_for_element_present('[href="/signup"]')
        self.wait_for_text("Hosted domains", timeout=20)
        self.highlight("h1")
        pop_up = '[role="dialog"] span.cursor-pointer'
        self.click_if_visible(pop_up)
        self.highlight("#block-summary")
        self.click_if_visible(pop_up)
        self.highlight("#block-geolocation")
        self.click_if_visible(pop_up)
        self.sleep(2)
        print("Displaying Host Info:")
        text = self.get_text("#block-summary").split("Hosted domains")[0]
        rows = text.split("\n")
        data = []
        for row in rows:
            if row.strip() != "":
                data.append(row.strip())
        print("\n".join(data).replace('\n"', ' "'))
        print("\nDisplaying GeoLocation Info:")
        text = self.get_text("#block-geolocation")
        text = text.split("IP Geolocation data")[0]
        rows = text.split("\n")
        data = []
        for row in rows:
            if row.strip() != "":
                data.append(row.strip())
        print("\n".join(data).replace('\n"', ' "'))
        self.sleep(3)