python 实现对象模型

 # -*- coding:utf-8 -*-

 """
 python 实现对象模型
 创建 bmicalcpage 类
 """

 class bmicalcpage(object):
     def __init__(self, driver):
         self._driver = driver
         self._url = r'F:\电子书\selenium\Selenium Testing Tools Cookbook(中文)\demo\BMICalculator.html'
         self._title = 'BMI Calculator'

     @property
     def is_loaded(self):
         return self._driver.title == self._title

     @property
     def bmi(self):
         bmi_field = self._driver.find_element_by_id('bmi')
         return bmi_field.get_attribute('value')

     @property
     def bmi_category(self):
         bmi_category_field = self._driver.find_element_by_id('bmi_category')
         return bmi_category_field.get_attribute('value')

     def open(self):
         self._driver.get(self._url)

     def calculate(self, height, weight):
         height_field = self._driver.find_element_by_id('heightCMS')
         weight_field = self._driver.find_element_by_id('weightKg')
         calc_button = self._driver.find_element_by_id('Calculate')

         height_field.send_keys(height)
         weight_field.send_keys(weight)
         calc_button.click()

     def close(self):
         self._driver.close()

 # -*-coding:utf-8 -*-
 """
 创建 BMI 计算的测试
 """
 from selenium import webdriver
 from BmiCalcPage import bmicalcpage
 import unittest

 class BmiCalcTest(unittest.TestCase):
     def testCalc(self):
         driver = webdriver.Firefox()
         bmi_calc = bmicalcpage(driver)
         bmi_calc.open()

         self.assertTrue(bmi_calc.is_loaded)

         bmi_calc.calculate('', '')

         self.assertEqual('24.4', bmi_calc.bmi)
         self.assertEqual('Normal', bmi_calc.bmi_category)
         bmi_calc.close()