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()
