# coding:utf-8

from selenium import webdriver

import unittest

import time

class QZone(unittest.TestCase):

    """登录QQ空间"""

    def setUp(self):

        self.driver = webdriver.Firefox()

        base_url = "https://i.qq.com"

        self.driver.get(base_url)

        self.driver.implicitly_wait(30)

    def login_qzone(self,user,password):

        """登录方法，账号密码参数化"""

        # self.driver.get(self.url + "/".format(self))

        self.driver.switch_to_frame('login_frame')

        self.driver.find_element_by_id('switcher_plogin').click()

        self.driver.find_element_by_id('u').clear()

        self.driver.find_element_by_id('u').send_keys(user)

        self.driver.find_element_by_id('p').clear()

        self.driver.find_element_by_id("p").send_keys(password)

        self.driver.find_element_by_id("login_button").click()

        time.sleep(3)

    def is_login_sucess(self):

        """判断是否登录成功"""

        try:

            result_text = self.driver.find_element_by_xpath(".//*[@id='headContainer']/div[2]/div[1]/span[1]").text

            print result_text

            return True

        except:

            return False

    def test_01(self):

        """登录case01"""

        self.login_qzone("1222222", "*********") #运行时输入对应的账号

        result = self.is_login_sucess()

        self.assertTrue(result)

    def test_02(self):

        """登录case02"""

        self.login_qzone("00000000", "*********")  # 运行时输入对应的账号

        result = self.is_login_sucess()

        self.assertTrue(result)

    def tearDown(self):

         self.driver.quit()

if __name__ == "__main__":

    unittest.main()