一、單元測試並統計測試的覆蓋率

1.1 Flask中的單元測試

        在之前的文章中介紹過在flask中進行單元測試的方法，目前我們的程式碼中存在下面的單元測試：

classBasicTestCase(unittest.TestCase):
    def Setup(self):
        self.app=create_app('testing')
        self.app_context=self.app.app_context()
        self.app_context.push()
        db.create_all()
 
    def teardown(self):
        db.session.remove()
        db.drop_all()
        self.app_context.pop()
 
    def test_app_exits(self):
        self.assertFalse(current_app is None)
 
    def test_app_is_testing(self):
       self.assertFalse(current_app.config['TESTING'])
 

        並且，我們在manager的命令列中增加了執行單元測試的命令：

@manager.command 
def test(): 
    """Run the unittests""" 
    import unittest 
   tests=unittest.TestLoader().discover('test') 
   unittest.TextTestRunner(verbosity=2).run(tests)

        可以通過命令列，按照下面的方式進行單元測試：

(flasky)[email protected]:~/Github/flaskTs$ python manager.py test
test_app_exits(test_basic.BasicTestCase) ... ok
test_app_is_testing(test_basic.BasicTestCase) ... ok
 
----------------------------------------------------------------------
Ran 2 tests in0.005s
 
OK
 
 

1.2 使用coverage進行單元測試覆蓋率的統計

        Flask中可以使用coverage模組來統計進行單元測試的程式碼覆蓋率，其安裝方式如下:

pip installcoverage

        安裝完成後，我們將coverage作為一個自定義命令的選項附加到test命令中：

COV=None
ifos.environ.get('COVERAGE'):
    import coverage
    COV=coverage.coverage(branch=True,include='*')
    COV.start()
 
@manager.command 
deftest(coverage=False): 
    """Run the unittests"""
    if coverage and notos.environ.get('COVERAGE'):
        import sys
        os.environ['COVERAGE']='1'
       os.execvp(sys.executable,[sys.executable]+sys.argv)
    import unittest 
   tests=unittest.TestLoader().discover('tests') 
   unittest.TextTestRunner(verbosity=2).run(tests) 
    if COV:
        COV.stop()
        COV.save()
        print('Coverage:')
        COV.report()
        COV.erase()
 

        首先在全域性的範圍內，我們先檢查是否具有'COVERAGE'環境變數，如果定義了該變數，就利用coverage的coverage函式構造一個coverage物件COV。coverage函式有兩個引數，第一個引數branch設定為True表示檢查的模組包含指定目錄下的子目錄，而第二個引數include是在指定目錄，設定成’*’的意思就是設定當前目錄和當前目錄下的模組位檢查模組。

        而在test()函式中，附加的命令列引數coverage會作為test()函式的一個bool型別的引數，如果命令列附加了該引數，coverage將會等於True。在函式內部，先判斷是否coverage已經是否定義了'COVERAGE'環境變數。因為我們定義以及執行COV的程式碼為全域性程式碼，在執行test命令列之前已經執行過，所以在判斷完之後，呼叫了os.execvp重新執行了程式。接下來COV變數不再是None，就可進行if COV後面的操作。

        執行的程式結果如下，由於將當前目錄下以及子目錄下所有的模組都作為了統計物件，所以結果中包含了flask本身自帶的一些模組。

二、使用Flask Web測試客戶端進行測試   

        Flask中的成員函式test_client()可以返回一個客戶端物件，可以模擬Web客戶端，用來同Web服務端進行互動。它可以測試Web程式，也可以用來測試Rest API，下面就該客戶端的使用進行討論。

2.1 測試Web程式

        新建一個FlaskClientTest單元測試類如下：

classFlaskClientTest(unittest.TestCase):
    def setUp(self):
        self.app=create_app('testing')
        self.app_context=self.app.app_context()
        self.app_context.push()
        db.create_all()
        self.client=self.app.test_client()
   
    def tearDown(self):
        db.session.remove()
        db.drop_all()
        self.app_context.pop()
   
    def test_home_page(self):
        response=self.client.get(url_for('main.index'))
        self.assertTrue('Home' inresponse.get_data(as_text=True))
   
    def test_register(self):
       response=self.client.post(url_for('main.register'),data={
            'email':'[email protected]',
            'name':'Hyman',
            'password1':'123',
            'password2':'123'})
       self.assertTrue(response.status_code==302)

        test_home_page是用來進行測試主頁的測試用例，這裡使用self.client.get(url_for('main.index'))的方法，使用GET方法訪問相應的url，並獲取響應結果。然後呼叫response的get_data()獲取響應結果，這裡面設定as_text=True，會把結果格式化成一個字串，進而判斷‘Home’是否在這個字串中。

        test_register用來模擬註冊使用者過程，對註冊的路由進行測試。這裡採用了POST的方法傳送了一個表單，而data中是對錶單各個欄位的賦值，這裡面的key值一定要和我們定義的登錄檔單的欄位名稱一致。由於註冊路由最終向主頁進行了重定向，屆時服務端回返回給瀏覽器302的狀態碼，我們通過判斷響應的status_code是否是302而判斷測試是否通過。

2.2 測試Rest API

        接下來，測試下上篇文章編寫的一個用來增加新部落格的Rest API：

deftest_posts(self):
        response=self.client.post(
            url_for('main.new_post'),
            data=json.dumps({'body':'I am a newpost'}),
            content_type='application/json')
       self.assertTrue(response.status_code==200)

        這裡面需要注意的就是，測試Rest API時需要在client的post方法中顯示的指定傳送的資料型別content_type為'application/json'，而且傳送的data需要用json.dumps()方法將資料格式化成json的格式。assertTrue中還是檢驗響應的status_code，如果文章建立成功，服務端會返回200的狀態碼，所以在這裡同200進行了比較。

三、使用selenium進行自動化測試

        Selenium可以自動啟動瀏覽器，模擬使用者點選瀏覽器的連線，selenium的安裝方法如下：

pip install selenium

        我們進行測試的思路時，利用後臺執行緒啟動服務端，然後在主執行緒中利用selenium自動操作firefox進行自動化的測試。在使用selenium之前，我們先要定義一個路由可以關閉服務端的執行，以使在測試結束後服務端自動結束執行。

@main.route('/shutdown')
defserv_shutdown():
   shutdown=request.environ.get('werkzeug.server.shutdown')
    shutdown()
    return 'shuting down'

         werkzeug伺服器提供了一個shutdown函式供外部使用，在上述的檢視函式中，我們獲取到了這個函式，並進行呼叫，這樣通過訪問http://localhost:5000/shutdown這個url我們就可以實現關閉服務端。

        接下來是使用selenium進行自動化測試的程式碼：

#coding:utf-8
import unittest
from flaskimport current_app,url_for,json
from app importcreate_app,db
from seleniumimport webdriver
 
classSeleniumTestCase(unittest.TestCase):
    client=None
    @classmethod
    def setUpClass(cls):
        #啟動firefox
        try:
            cls.client=webdriver.Firefox()
        except:
            pass
        if cls.client:
            cls.app=create_app('testing')
           cls.app_context=cls.app.app_context()
            cls.app_context.push()
            db.create_all()
           threading.Thread(target=cls.app.run).start()
    @classmethod
    def tearDownClass(cls):
        if cls.client:
           cls.client.get('http://localhost:5000/shutdown')
            cls.client.close()
            db.drop_all()
            db.session.remove()
            cls.app_context.pop()
 
    def setUp(self):
        if not self.client:
            self.skipTest('Firefox isinvailable')
   
    def tearDown(self):
        pass
   
    def test_home_page(self):
       self.client.get('http://localhost:5000/')
        self.assertTrue(re.search('Home',self.client.page_source))

        通過修飾器classmethod定義了setUpClass()和tearDownClass()兩個成員函式，分別在該類中的所有的測試用例執行之前和之後執行，而setUp()和tearDown()是在每個測試用例執行前後都執行。在setUpClass()中，通過webdriver.Firefox()獲取到了Firefox()瀏覽器物件，並在threading.Thread(target=cls.app.run).start()中在後臺執行緒執行服務端，然後在測試用例test_home_page中我們通過這個瀏覽器物件訪問'http://localhost:5000/'，並利用正則表示式判斷返回的結果中是否包含'Home'字串。在所有的測試完成之後，在tearDownClass()中訪問'http://localhost:5000/shutdown'從而完成了服務端的關閉。至此，一個完整的selenium測試流程完成。

        Selenium的強大在於，它可以操作瀏覽器實現自動化的測試，彷彿就像有人在瀏覽器上點選一樣，它建立的Firefox客戶端物件提供了很多方法可以在Web介面找到相關的元素並進行後續的操作。比如find_element_by_link_text()方法，可以根據超連結的名稱找到該連結，並在呼叫其Click()方法模擬點選超連結；find_element_by_name()方法根據根據名字找到對應的欄位，然後可以使用期send_keys()的方法在各個欄位中填入值。

Github位置：
https://github.com/HymanLiuTS/flaskTs
克隆本專案：
Git clone Git@github.com:HymanLiuTS/flaskTs.Git
獲取本文原始碼：
Git checkout FL40