python 实现对象模型

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

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

class bmicalcpage(object):
    def __init__(self, driver):
        self._driver = driver
        self._url = r'F:电子书seleniumSelenium Testing Tools Cookbook(中文)demoBMICalculator.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('181', '80')
        
        self.assertEqual('24.4', bmi_calc.bmi)
        self.assertEqual('Normal', bmi_calc.bmi_category)
        bmi_calc.close()