webdriver高级应用- 在HTML5的画布元素上进行绘画操作

#encoding=utf-8
import unittest
from selenium import webdriver
import time

class TestDemo(unittest.TestCase):
    def setUp(self):
        # 获取浏览器驱动实例
        self.driver = webdriver.Ie(executable_path = "e:\\IEDriverServer")

    def test_HTML5Canvas(self):
        url = "http://www.w3school.com.cn/tiy/loadtext.asp?f=html5_canvas_line"
        # 访问指定的网址
        self.driver.get(url)
        # 调用JavaScript语句，在页面画布上画一个红色的图案
        # getElementById('myCanvas'); 语句获取页面上的画布元素
        # var cxt = c.getContext('2d'); 设定画布为2d
        # cxt.fillStyle = '#FF0000'; 设定填充色为  # FF0000 红色
        # cxt.fillRect(0, 0, 150, 150); 在画布上绘制矩形
        self.driver.execute_script("var c = document.getElementById('myCanvas');"
               + "var cxt=c.getContext('2d');"
               + "cxt.fillStyle='#FF0000';"
               +"cxt.fillRect(0,0,150,150);")
        time.sleep(3)
        # 将绘制的红色矩形页面，进行截屏，并保存为E盘的HTML5Canvas.jpg
        self.driver.save_screenshot("E:\\HTML5Canvas.jpg")

    def tearDown(self):
        # 退出浏览器
        self.driver.quit()

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