多选框处理

# -*- coding:utf-8 -*-
"""
多选框处理
"""
from selenium import webdriver
from compiler.ast import Assert
import unittest

class Checkbox_Demo(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.get(r'F:电子书seleniumSelenium Testing Tools Cookbook(中文)democheckbox.html')

    def test_checkbox(self):
        driver = self.driver
        apple = driver.find_element_by_css_selector("input[value='Apple']")        
        pear = driver.find_element_by_css_selector("input[value='Pear']")        
        orange = driver.find_element_by_css_selector("input[value='Orange']")

        #检查是否已选择，如果没有则点击选择        
        if not apple.is_selected():
            apple.click()
        
        if not pear.is_selected():
            pear.click()
            
        if not orange.is_selected():
            orange.click()
        
        self.assertTrue(apple.is_selected())
        self.assertTrue(pear.is_selected())
        self.assertTrue(orange.is_selected())
        
        #检查是否已选择，如果已选择则点击取消选择
        if apple.is_selected():
            apple.click()
        
        self.assertTrue(apple.is_selected())

    def tearDown(self):
        self.driver.quit()