用unittest測試web2py應用中的非頁面部分

雖然web2py在文檔中建議採用doctest對controller進行測試的方式，但是一則我對doctest不熟，二則只對controller層進行測試覆蓋雖然很全面，但是測試粒度太大，對我來說測得不夠細。

我
的設計習慣是controller層只進行簡單的調用轉換，實際的商務邏輯放在module層（注意：不是model層）處理，雖然在model裡進行設
置後藉助於Generic
View也能幹很多事情，但是一方面我不習慣這種方式，另一方面這種方式還是有一定的局限性——最主要是我不想把太多介面上的東西放到model裡。

既
然商務邏輯在module中，那麼我就需要在不藉助頁面的情況下對這些module進行單獨的測試，這是文檔建議的測試方式所做不到的——我又不想為了測
試而特地弄幾個頁面。於是研究了一下如何用unittest對module進行測試——問題在關鍵在於如何在unittest中建立web2py的運行環
境。而這一環境的核心就是db——如何初始化一個db供被測試的module使用。

其實要做到這一點並不難。gluon.shell包中提供了一個函數exec_environment來實現這一功能。假設有一個db.py是這樣的：

db.define_table('tabletest',<br /> SQLField('code'),<br /> SQLField('name'))

那麼測試案例可以寫成這樣：

import sys<br />import os<br />db_path = path.realpath('../models/db.py')<br />web2py_path = '../../..'<br />sys.path.append(os.path.realpath(web2py_path))<br />sys.path.append(os.path.realpath('..'))<br />os.chdir(web2py_path)<br />import unittest<br />from gluon.shell import exec_environment<br />from modules import datamodule<br />class TestDefaultController(unittest.TestCase):<br /> def setUp(self):<br /> self.dbm=exec_environment(db_path)<br /> def tearDown(self):<br /> db=self.dbm.db<br /> query=(db.tabletest.code=="000111")<br /> db(query).delete()<br /> def testDB(self):<br /> db=self.dbm.db<br /> x=db.tabletest.insert(code="000111", name="test name")<br /> query=(db.tabletest.id==x)<br /> y=db(query).select()[0].id<br /> self.assertEqual(x,y)<br /> def testDataModule(self):<br /> db=self.dbm.db<br /> x=datamodule.foo(db, ...)<br /> self.assert...<br />if __name__ == '__main__':<br /> unittest.main()

代碼很簡單，最開始一段是準備程式的運行路徑，因為這個測試案例的檔案是放在： web2py/applications/yourapp/tests下的，所以用"../../.."來指向web2py的根目錄，並定位當前位置於此。

重
點在於setUp函數中調用了exec_environment函數來執行db.py，這樣就能夠為測試準備好web2py的運行環境，並啟用db.py
中的model定義。之後在testDB和testDataModule中就可以使用db進行資料庫訪問並檢測執行結果。這與一般的unittest應用
沒什麼不同。

最後需要注意的一點是：不能用windows exe版的web2py，要用源碼版的web2py才能正常使用unittest，否則會出錯。