web自动化

import unittest
from selenium import webdriver
from time import sleep
import time


class IWebShopLogin(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
        url = 'http://localhost/iwebshop/'
        self.driver.get(url)
        self.driver.maximize_window()
        self.driver.implicitly_wait(30)

    def tearDown(self):
        sleep(2)
        self.driver.find_element_by_link_text("安全退出").click()
        sleep(2)
        self.driver.quit()

    def test_login(self):
        driver = self.driver
        driver.find_element_by_link_text('登录').click()
        driver.find_element_by_name('login_info').send_keys('admin')
        driver.find_element_by_name('password').send_keys('123456')
        driver.find_element_by_class_name('submit_login').click()
        login_info = driver.find_element_by_class_name('loginfo').text
        print(login_info)
        try:
            self.assertIn('adminA', login_info)
        except AssertionError:
            now_time = time.strftime("%Y-%m-%d %H%M%S")
            driver.get_screenshot_as_file("./image/%s.png" % (now_time))
            raise

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