PythonでSelenium WebDriverをプロキシ経由で実行する方法

PythonのSelenium WebDriverでは、プロキシを設定した状態でブラウザを操作することができます。プロキシは、ローカライズ(地域対応)テストを行ううえで欠かせない要素です。たとえばECサイトのテストでは、ユーザーの所在地に応じて表示される言語や通貨が正しいかどうかを確認できます。

テスト内でプロキシを活用することで、ウェブサイトのUI(ユーザーインターフェース)がアクセス元の地域と一致しているかを検証することも可能です。Seleniumでプロキシを設定するには、以下の手順に従います。

• Seleniumパッケージからwebdriverをインポートする
• プロキシサーバーのアドレスを定義する
• ChromeOptionsクラスのオブジェクトを作成する
• プロキシ情報をChromeOptionsに渡す
• 設定済みのオプションをChrome()オブジェクトに指定して起動する

コード例

実際のコード実装は以下の通りです。なお、Selenium 4以降ではchrome_options=ではなくoptions=を使用する点に注意してください。

from selenium import webdriver

# プロキシサーバーの定義
py = "128.21.0.0:8080"

# ChromeOptionsクラスの設定
chrome_options = webdriver.ChromeOptions()

# オプションにプロキシパラメータを追加
chrome_options.add_argument('--proxy-server=%s' % py)

# オプションをChrome()に渡して起動
driver = webdriver.Chrome(options=chrome_options)
driver.implicitly_wait(0.6)
driver.get("https://www.tutorialspoint.com/index.htm")

続いて、ページ上の要素に現在のユーザーの所在地が正しく表示されているかを確認するには、以下のようなコードを追加します。要素の取得には、新しい書式であるfind_element(By.XPATH, ...)を使用しています。

from selenium.webdriver.common.by import By

def checkL(self):
    self.driver.get(self.url)
    st = self.driver.find_element(By.XPATH, "//div[@id='loc']")
    # アサーションで所在地を検証
    self.assertEqual('India', st.text)

複数の地域を検証したい場合は、プロキシアドレスを引数として受け取るメソッドを作成しておけば、同じコードを再利用しながら効率的にテストを実行できます。