怎样从0开始搭建一个测试框架_1——配置

乐哈哈yoyo

我们先创建一个简单的脚本吧，在test文件夹创建test_baidu.py：

1. <font face="宋体" size="3">import os
   
2. import time
   
3. from selenium import webdriver
   
4. from selenium.webdriver.common.by import By
   
5. 
   
6. URL = "http://www.baidu.com"
   
7. base_path = os.path.dirname(os.path.abspath(__file__)) + '\..'
   
8. driver_path = os.path.abspath(base_path+'\drivers\chromedriver.exe')
   
9. 
   
10. locator_kw = (By.ID, 'kw')
    
11. locator_su = (By.ID, 'su')
    
12. locator_result = (By.XPATH, '//div[contains(@class, "result")]/h3/a')
    
13. 
    
14. driver = webdriver.Chrome(executable_path=driver_path)
    
15. driver.get(URL)
    
16. driver.find_element(*locator_kw).send_keys('selenium 灰蓝')
    
17. driver.find_element(*locator_su).click()
    
18. time.sleep(2)
    
19. links = driver.find_elements(*locator_result)
    
20. for link in links:
    
21.     print(link.text)
    
22. driver.quit()</font>

复制代码

如果想要搜索“Python selenium”，是不是要再创建一个脚本？还是把原来的脚本修改一下？

或者我们可以用unittest来改一下，把两次搜索分别写一个测试方法：

1. <font face="宋体" size="3">import os
   
2. import time
   
3. import unittest
   
4. from selenium import webdriver
   
5. from selenium.webdriver.common.by import By
   
6. 
   
7. 
   
8. class TestBaiDu(unittest.TestCase):
   
9.     URL = "http://www.baidu.com"
   
10.     base_path = os.path.dirname(os.path.abspath(__file__)) + '\..'
    
11.     driver_path = os.path.abspath(base_path+'\drivers\chromedriver.exe')
    
12. 
    
13.     locator_kw = (By.ID, 'kw')
    
14.     locator_su = (By.ID, 'su')
    
15.     locator_result = (By.XPATH, '//div[contains(@class, "result")]/h3/a')
    
16. 
    
17.     def setUp(self):
    
18.         self.driver = webdriver.Chrome(executable_path=self.driver_path)
    
19.         self.driver.get(self.URL)
    
20. 
    
21.     def tearDown(self):
    
22.         self.driver.quit()
    
23. 
    
24.     def test_search_0(self):
    
25.         self.driver.find_element(*self.locator_kw).send_keys('selenium 灰蓝')
    
26.         self.driver.find_element(*self.locator_su).click()
    
27.         time.sleep(2)
    
28.         links = self.driver.find_elements(*self.locator_result)
    
29.         for link in links:
    
30.             print(link.text)
    
31. 
    
32.     def test_search_1(self):
    
33.         self.driver.find_element(*self.locator_kw).send_keys('Python selenium')
    
34.         self.driver.find_element(*self.locator_su).click()
    
35.         time.sleep(2)
    
36.         links = self.driver.find_elements(*self.locator_result)
    
37.         for link in links:
    
38.             print(link.text)
    
39. 
    
40. 
    
41. if __name__ == '__main__':
    
42.     unittest.main()</font>

复制代码

现在，我们把配置抽出来放到config.yml中：

1. <font face="宋体" size="3">URL: http://www.baidu.com</font>

复制代码

为了读取yaml文件，我们需要一个封装YamlReader类，在utils中创建file_reader.py文件：

1. <font face="宋体" size="3">import yaml
   
2. import os
   
3. 
   
4. 
   
5. class YamlReader:
   
6.     def __init__(self, yamlf):
   
7.         if os.path.exists(yamlf):
   
8.             self.yamlf = yamlf
   
9.         else:
   
10.             raise FileNotFoundError('文件不存在！')
    
11.         self._data = None
    
12. 
    
13.     @property
    
14.     def data(self):
    
15.         # 如果是第一次调用data，读取yaml文档，否则直接返回之前保存的数据
    
16.         if not self._data:
    
17.             with open(self.yamlf, 'rb') as f:
    
18.                 self._data = list(yaml.safe_load_all(f))  # load后是个generator，用list组织成列表
    
19.         return self._data</font>

复制代码

而且我们需要一个Config类来读取配置，config.py：

1. <font face="宋体" size="3">"""
   
2. 读取配置。这里配置文件用的yaml，也可用其他如XML,INI等，需在file_reader中添加相应的Reader进行处理。
   
3. """
   
4. import os
   
5. from utils.file_reader import YamlReader
   
6. 
   
7. # 通过当前文件的绝对路径，其父级目录一定是框架的base目录，然后确定各层的绝对路径。如果你的结构不同，可自行修改。
   
8. # 之前直接拼接的路径，修改了一下，用现在下面这种方法，可以支持linux和windows等不同的平台，也建议大家多用os.path.split()和os.path.join()，不要直接+'\\xxx\\ss'这样
   
9. BASE_PATH = os.path.split(os.path.dirname(os.path.abspath(__file__)))[0]
   
10. CONFIG_FILE = os.path.join(BASE_PATH, 'config', 'config.yml')
    
11. DATA_PATH = os.path.join(BASE_PATH, 'data')
    
12. DRIVER_PATH = os.path.join(BASE_PATH, 'drivers')
    
13. LOG_PATH = os.path.join(BASE_PATH, 'log')
    
14. REPORT_PATH = os.path.join(BASE_PATH, 'report')
    
15. 
    
16. 
    
17. class Config:
    
18.     def __init__(self, config=CONFIG_FILE):
    
19.         self.config = YamlReader(config).data
    
20. 
    
21.     def get(self, element, index=0):
    
22.         """
    
23.         yaml是可以通过'---'分节的。用YamlReader读取返回的是一个list，第一项是默认的节，如果有多个节，可以传入index来获取。
    
24.         这样我们其实可以把框架相关的配置放在默认节，其他的关于项目的配置放在其他节中。可以在框架中实现多个项目的测试。
    
25.         """
    
26.         return self.config[index].get(element)</font>

复制代码

修改test_baidu.py:

1. <font face="宋体" size="3">import time
   
2. import unittest
   
3. from selenium import webdriver
   
4. from selenium.webdriver.common.by import By
   
5. from utils.config import Config, DRIVER_PATH
   
6. 
   
7. 
   
8. class TestBaiDu(unittest.TestCase):
   
9.     URL = Config().get('URL')
   
10. 
    
11.     locator_kw = (By.ID, 'kw')
    
12.     locator_su = (By.ID, 'su')
    
13.     locator_result = (By.XPATH, '//div[contains(@class, "result")]/h3/a')
    
14. 
    
15.     def setUp(self):
    
16.         self.driver = webdriver.Chrome(executable_path=DRIVER_PATH + '\chromedriver.exe')
    
17.         self.driver.get(self.URL)
    
18. 
    
19.     def tearDown(self):
    
20.         self.driver.quit()
    
21. 
    
22.     def test_search_0(self):
    
23.         self.driver.find_element(*self.locator_kw).send_keys('selenium 灰蓝')
    
24.         self.driver.find_element(*self.locator_su).click()
    
25.         time.sleep(2)
    
26.         links = self.driver.find_elements(*self.locator_result)
    
27.         for link in links:
    
28.             print(link.text)
    
29. 
    
30.     def test_search_1(self):
    
31.         self.driver.find_element(*self.locator_kw).send_keys('Python selenium')
    
32.         self.driver.find_element(*self.locator_su).click()
    
33.         time.sleep(2)
    
34.         links = self.driver.find_elements(*self.locator_result)
    
35.         for link in links:
    
36.             print(link.text)
    
37. 
    
38. 
    
39. if __name__ == '__main__':
    
40.     unittest.main()</font>

复制代码

我们已经把配置分离出来了，虽然现在看起来似乎很麻烦，但是想想如果你有50个用例文件甚至更多，一旦项目URL变了，你还要一个个去修改吗？

Miss_love

支持分享