1 # -*- coding:utf-8 -*-
 2 """
 3 数据驱动测试，从 csv 文件中读取数据
 4 """
 5 from selenium import webdriver
 6 import csv, sys
 7
 8
 9 test_executed = 0
10 test_passed = 0
11 test_failed = 0
12 test_status = True
13
14
15 try:
16     driver = webdriver.Firefox()
17     driver.get('C:\BMICalculator.html')
18
19     datafile = open(r'c:\data.csv', 'rb')
20     reader = csv.reader(datafile)
21
22     test_executed = 0
23
24     for row in reader:
25         test_executed += 1
26         print 'Test' + str(test_executed)
27
28         heightField = driver.find_element_by_name('heightCMS')
29         heightField.clear()
30         heightField.send_keys(row[0])
31
32         weightField = driver.find_element_by_name('weightKg')
33         weightField.clear()
34         weightField.send_keys(row[1])
35
36         calculateButton = driver.find_element_by_id('Calculate')
37         calculateButton.click()
38
39         bmiLabel = driver.find_element_by_name('bmi')
40         bmiCategoryLabel = driver.find_element_by_name('bmi_category')
41
42         if bmiLabel.get_attribute('value') == row[2]:
43             print "PASS, expected value for BMI <" + row[2] + "> actual <" + bmiLabel.get_attribute('value') + ">"
44         else:
45             print "FAIL, expected value for BMI <" + row[2] + "> actual <" + bmiLabel.get_attribute('value') + ">"
46             test_status = False
47
48         if bmiCategoryLabel.get_attribute('value') == row[3]:
49             print "PASS, expected value for BMI Category <" + row[3] + "> actual <" + bmiCategoryLabel.get_attribute('value') + ">"
50         else:
51             print "Fail, expected value for BMI Category <" + row[3] + "> actual <" + bmiCategoryLabel.get_attribute('value') + ">"
52             test_status = False
53
54         if test_status == True:
55             test_passed = test_passed + 1
56         else:
57             test_failed = test_failed + 1
58
59 except :
60     print "Unexpected error: ", sys.exc_info()[0]
61     raise
62
63 finally:
64     print "---------------------------------------------------------------"
65     print "Total (" + str(test_executed) + ") Tests Executed"
66     print "Total (" + str(test_passed) + ") Tests Passed"
67     print "Total (" + str(test_failed ) + ") Tests Failed"
68     driver.close()
69     datafile.close()