import time
from selenium import webdriver

'''
测试用例：打开百度首页，搜索“胡歌”，然后检索列表，有无“胡歌的新浪微博”这个链接
场景拆分：
 1）启动Chrome浏览器
 2) 打开百度首页，https://www.baidu.com
 3）定位搜索输入框，输入框元素XPath表达式：//*[@id="kw"]
 4）定位搜索提交按钮（百度一下）：//*[@id="su"]
 5）在搜索框输入“胡歌”，点击百度一下按钮
 6）在搜索结果列表判断是否存在“胡歌的新浪微博”这个链接
 7）退出浏览器，结束测试
'''
driver = webdriver.Chrome()
driver.maximize_window()
driver.implicitly_wait(8) # 设置隐式等待时间

driver.get("https://www.baidu.com") # 地址栏里输入网址
driver.find_element_by_xpath('//*[@id="kw"]').send_keys("胡歌") # 搜索框输入胡歌
driver.find_element_by_xpath('//*[@id="su"]').click() # 点击百度一下按钮

time.sleep(2) # 等待2秒
# 通过元素XPath来确定该元素是否显示在结果列表，从而判断“壁纸”这个链接是否显示在结果列表
# find_element_by_link_text当找不到此链接时报错，程序停止
driver.find_element_by_link_text('胡歌的新浪微博').is_displayed()
driver.quit()

# 获取当前页面标题显示的字段
print(driver.title)

# 1) 包含判断
# assert:断言，声称
try:
 assert "百度一下" in driver.title
 print("断言测试成功.")
except Exception as e:
 print("断言失败.",format(e))

# 2) 完全相等判断
if "百度一下，你就知道" == driver.title:
 print("成功.")
else:
 print("失败.")

print(driver.title)

try:
 # 新标签页，此处用js实现，在有些博客上显示使用
 # driver.find_element_by_tag_name('body').send_keys(Keys.CONTROL, 't')
 # 我这测试无效，原因不知，于是采用如下方法
 js = "window.open('http://www.acfun.cn/')"
 driver.execute_script(js)

 # 切换到新的窗口
 handles = driver.window_handles # 获取窗口句柄
 driver.switch_to.window(handles[-1]) # 切换到最后一个既最新打开的窗口

 # 先切换窗口再打开新网址，才是在新窗口打开网址，不然还是在原来的百度页面打开此网址
 driver.get('http://map.baidu.com/')

except Exception as e:
 print("发现异常，",format(e))

driver.get("https://www.baidu.com")
time.sleep(2)
'''前进，后退'''
elem_news = driver.find_element_by_link_text("新闻").click() # 点击进入新闻
time.sleep(2)
driver.back() # 后退到百度首页
time.sleep(2)
driver.forward() # 从百度前进到新闻页
time.sleep(2)

# 获取浏览器版本号
"""
 Creates a new session with the desired capabilities.

 :Args:
 - browser_name - The name of the browser to request.
 - version - Which browser version to request.
 - platform - Which platform to request the browser on.
 - javascript_enabled - Whether the new session should support JavaScript.
 - browser_profile - A selenium.webdriver.firefox.firefox_profile.FirefoxProfile object. Only used if Firefox is requested.
"""
print(driver.capabilities['version'])