seleniumを使った自動ログインテスト

seleniumを使ってログインテストができたので簡単に紹介します。
参考にしたサイトは↓です。
qiita.com
#必要なもの

• Firefox
• python3
• pip

#手順
まず、seleniumをpipでインストールします。

$ sudo pip3 install selenium

下記のようなスクリプトを作成します。

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import selenium.webdriver

username = "ログインしたいユーザの名前"
password = "ログインしたいユーザのパスワード"
URL = "ログインするサイトのURL"

webdriver = selenium.webdriver
browser = webdriver.Firefox()
browser.get(URL)
login_username = browser.find_element_by_id("ユーザ名を入力するフィールドのID")
login_username.send_keys(username)
login_password = browser.find_element_by_id("パスワードを入力するフィールドのID")
login_password.send_keys(password)
login_password.send_keys('\n')

後は実行するだけでログインを行います。

$ this_script.py

もし、ログイン後にブラウザを終了させたい場合は下記の処理を適切な箇所に追加します。
(sleepを行っているのはログインする時間を考慮するためです。)

import time
time.sleep(2)
browser.quit()

すると、「あるサイトでログインを行った後ブラウザを閉じる」というテストが自動で行えます。
すばらしいですね。