python+selenium实例： 登录 与文本框输入内容，保存并返回断言标题的正确性

#coding: utf-8
import unittest
from selenium import webdriver
import time

class LoginCase(unittest.TestCase):

def setUp(self): #每个用例执行之前执行
print 'before test'
self.dr = webdriver.Chrome()
self.dr.get('http://localhost/wordpress/wp-login.php')

# def test_login(self):
# user_name = password = 'admin'
# self.login(user_name, password)
# self.assertTrue('wp-admin' in self.dr.current_url)
# greeting_link = self.by_css('#wp-admin-bar-my-account .ab-item')
# self.assertTrue(user_name in greeting_link.text)

def test_post(self):
user_name = password = 'admin'
self.login(user_name, password)
title = 'this is title %s' %(time.time())
#加上时间戳，是为了区分跟上次的内容不一样

self.dr.get('http://localhost/wordpress/wp-admin/post-new.php')
self.by_name('post_title').send_keys(title)
self.set_content('content content')
self.by_name('publish').click()
self.dr.get('http://localhost/wordpress/wp-admin/edit.php')
self.assertEqual(self.by_css('.row-title').text, title)
#断言，返回到列表，判断第一列的标题内容是否与原输入的一致。

def set_content(self, text):
js = "document.getElementById('content_ifr').contentWindow.document.body.innerText = '%s'" %(text)
#在文本框里输入内容
print js
self.dr.execute_script(js)

def login(self, user_name, password):
self.by_id('user_login').send_keys(user_name)
self.by_id('user_pass').send_keys(password)
self.by_id('wp-submit').click()

def by_id(self, the_id):
return self.dr.find_element_by_id(the_id)

def by_css(self, css):
return self.dr.find_element_by_css_selector(css)

def by_name(self, name):
return self.dr.find_element_by_name(name)

def tearDown(self): #每个用例执行之后
print 'after every test'
self.dr.quit()

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