test_site.py

# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re, os, sys, codecs, util


class TestSite(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome("./chromedriver.exe")
        self.driver.implicitly_wait(30)
        self.base_url = "http://www.xxx.com/"
        self.verificationErrors = []
        self.accept_next_alert = True
        self.times = 2
        self.userId = ''
        self.password = ''
        self.userName = ''
        self.windowsize_x = ''
        self.windowsize_y = ''
        self.form_address_prefecture = ''
        self.form_city1 = ''
        self.form_city2 = ''
        self.form_email = ''
        self.form_email_check = ''
        it = (line.split("\t") for line in codecs.open('data.tsv', 'r', 'utf-8'))
        for x in range(self.times):
            xs = next(it)
            self.userId = xs[0]
            self.password = xs[1]
            self.windowsize_x = xs[2]
            self.windowsize_y = xs[3]
            self.form_userName = xs[6]
            self.form_address_prefecture = xs[10]
            self.form_email = xs[16]
            self.form_email_check = xs[17]
        it.close()


    def test_site(self):
        driver = self.driver
        driver.get(self.base_url + "/login")
        driver.find_element_by_id("userId").clear()
        driver.find_element_by_id("userId").send_keys(self.userId)
        driver.find_element_by_id("password").clear()
        driver.find_element_by_id("password").send_keys(self.form_password)
        driver.set_window_size(self.windowsize_x, self.windowsize_y)
        time.sleep(1)
        self.save_screenshot_chrome(os.path.join(os.path.dirname(os.path.abspath(__file__)), "screenshot01.png"))
        ##
        driver.maximize_window()
        driver.find_element_by_id("submit").click()
        driver.set_window_size(self.windowsize_x, self.windowsize_y)
        time.sleep(1)
        self.save_screenshot_chrome(os.path.join(os.path.dirname(os.path.abspath(__file__)), "screenshot02.png"))
        driver.maximize_window()
        driver.find_element_by_xpath("(//a[contains(text(),'Detail')])[1]").click()
        driver.find_element_by_xpath("(//a[contains(text(),'Detail')])[2]").click()

        Select(driver.find_element_by_id("form_className1")).select_by_visible_text(self.form_className1) if self.form_classCtName1 else None
        time.sleep(0.5)
        Select(driver.find_element_by_id("form_className2")).select_by_visible_text(self.form_className2) if self.form_className2 else None
#        driver.set_window_size(self.windowsize_x, self.windowsize_y)
        time.sleep(1)
        self.save_screenshot_chrome(os.path.join(os.path.dirname(os.path.abspath(__file__)), "screenshot03.png"))
#        driver.maximize_window()
        driver.find_element_by_id("form_submit").click()

        ##
        driver.find_element_by_id("form_userName").clear()
        driver.find_element_by_id("form_userName").send_keys(self.form_userName)
        driver.find_element_by_id("form_furigana").clear()
        driver.find_element_by_id("form_postCode1").clear()
        Select(driver.find_element_by_id("form_address_prefecture")).select_by_visible_text(self.form_address_prefecture)
        driver.find_element_by_id("form_city1").clear()
        driver.find_element_by_id("form_city1").send_keys(self.form_city1)
        driver.find_element_by_id("form_city2").clear()
        driver.find_element_by_id("form_city2").send_keys(self.form_city2)
        driver.find_element_by_id("form_telNum1").clear()
        driver.find_element_by_id("form_email").send_keys(self.form_email)
        driver.find_element_by_id("form_email_check").clear()
        driver.find_element_by_id("form_email_check").send_keys(self.form_email_check)
        driver.set_window_size(self.windowsize_x, self.windowsize_y)
        time.sleep(1)
        self.save_screenshot_chrome(os.path.join(os.path.dirname(os.path.abspath(__file__)), "screehshot04.png"))
        driver.maximize_window()
        driver.find_element_by_id("form_submit").click()

        ##
        driver.set_window_size(self.windowsize_x, self.windowsize_y)
        time.sleep(1)
        self.save_screenshot_chrome(os.path.join(os.path.dirname(os.path.abspath(__file__)), "screenshot05.png"))
        driver.maximize_window()
        driver.find_element_by_link_text("戻る").click()
        driver.find_element_by_link_text("Confirm").click()
        ##
        driver.set_window_size(self.windowsize_x, self.windowsize_y)
        time.sleep(1)
        self.save_screenshot_chrome(os.path.join(os.path.dirname(os.path.abspath(__file__)), "screenshot06.png"))
        driver.close()

    def is_element_present(self, how, what):
        try: self.driver.find_element(by=how, value=what)
        except NoSuchElementException as e: return False
        return True
    
    def is_alert_present(self):
        try: self.driver.switch_to_alert()
        except NoAlertPresentException as e: return False
        return True
    
    def close_alert_and_get_its_text(self):
        try:
            alert = self.driver.switch_to_alert()
            alert_text = alert.text
            if self.accept_next_alert:
                alert.accept()
            else:
                alert.dismiss()
            return alert_text
        finally: self.accept_next_alert = True
        
    def save_screenshot_chrome(self, file):
        ''' Generate document-height screenshot '''
        util.fullpage_screenshot(self.driver, file)

    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)

if __name__ == "__main__":
    unittest.main()