摘要:一些常用的示例先簡單定義個類,方便舉例這個類里有兩個成員方法,一個有參數,一個無參數。有個模塊方法,返回數據庫中所有指定的人員,并按排序掉整個數據庫訪問先得到一個的對象,再在中設置一個對象,此時不需要自己再創建
一些常用的mock示例
先簡單定義個類,方便舉例:
class Person: def __init__(self): self.__age = 10 def get_fullname(self, first_name, last_name): return first_name + " " + last_name def get_age(self): return self.__age @staticmethod def get_class_name(): return Person.__name__
這個類里有兩個成員方法,一個有參數,一個無參數。還有一個靜態方法
class PersonTest(TestCase): def test_should_get_age(self): p = Person() # 不mock時,get_age應該返回10 self.assertEqual(p.get_age(), 10) # mock掉get_age方法,讓它返回20 p.get_age = Mock(return_value=20) self.assertEqual(p.get_age(), 20) def test_should_get_fullname(self): p = Person() # mock掉get_fullname,讓它返回"James Harden" p.get_fullname = Mock(return_value="James Harden") self.assertEqual(p.get_fullname(), "James Harden")
上面的例子你也許已經注意到了,調用p.get_fullname時沒有給任何的參數,但是依然可以工作。
如果想校驗參數需要用create_autospec模塊方法替代Mock類。
class PersonTest(TestCase): def test_should_get_fullname(self): p = Person() p.get_fullname = create_autospec(p.get_fullname, return_value="James Harden") # 隨便給兩個參數,依然會返回mock的值 self.assertEqual(p.get_fullname("1", "2"), "James Harden") # 如果參數個數不對,會報錯TypeError: missing a required argument: "last_name" p.get_fullname("1")
class PersonTest(TestCase): def test_should_get_age(self): p = Person() p.get_age = Mock(side_effect=[10, 11, 12]) self.assertEqual(p.get_age(), 10) self.assertEqual(p.get_age(), 11) self.assertEqual(p.get_age(), 12)
class PersonTest(TestCase): def test_should_get_fullname(self): p = Person() values = {("James", "Harden"): "James Harden", ("Tracy", "Grady"): "Tracy Grady"} p.get_fullname = Mock(side_effect=lambda x, y: values[(x, y)]) self.assertEqual(p.get_fullname("James", "Harden"), "James Harden") self.assertEqual(p.get_fullname("Tracy", "Grady"), "Tracy Grady")
class PersonTest(TestCase): def test_should_raise_exception(self): p = Person() p.get_age = Mock(side_effect=TypeError("integer type")) # 只要調就會拋出異常 self.assertRaises(TypeError, p.get_age)
class PersonTest(TestCase): def test_should_validate_method_calling(self): p.get_fullname = Mock(return_value="James Harden") # 沒調用過 p.get_fullname.assert_not_called() # Python 3.5 p.get_fullname("1", "2") # 調用過任意次數 p.get_fullname.assert_called() # Python 3.6 # 只調用過一次, 不管參數 p.get_fullname.assert_called_once() # Python 3.6 # 只調用過一次,并且符合指定的參數 p.get_fullname.assert_called_once_with("1", "2") p.get_fullname("3", "4") # 只要調用過即可,必須指定參數 p.get_fullname.assert_any_call("1", "2") # 重置mock,重置之后相當于沒有調用過 p.get_fullname.reset_mock() p.get_fullname.assert_not_called() # Mock對象里除了return_value, side_effect屬性外, # called表示是否調用過,call_count可以返回調用的次數 self.assertEqual(p.get_fullname.called, False) self.assertEqual(p.get_fullname.call_count, 0) p.get_fullname("1", "2") p.get_fullname("3", "4") self.assertEqual(p.get_fullname.called, True) self.assertEqual(p.get_fullname.call_count, 2)mock靜態方法
靜態方法和模塊方法需要使用patch來mock。
class PersonTest(TestCase): # 以字符串的形式列出靜態方法的路徑,在測試的參數里會自動得到一個Mock對象 @patch("your.package.module.Person.get_class_name") def test_should_get_class_name(self, mock_get_class_name): mock_get_class_name.return_value = "Guy" self.assertEqual(Person.get_class_name(), "Guy")
class PersonTest(TestCase): mock_get_class_name = Mock(return_value="Guy") # 在patch中給出定義好的Mock的對象,好處是定義好的對象可以復用 @patch("your.package.module.Person.get_class_name", mock_get_class_name) def test_should_get_class_name(self): self.assertEqual(Person.get_class_name(), "Guy")
class PersonTest(TestCase): mock_get_class_name = Mock(return_value="Guy") # 使用patch.object來mock,好處是Person類不是以字符串形式給出的 @patch.object(Person, "get_class_name", mock_get_class_name) def test_should_get_class_name(self, ): self.assertEqual(Person.get_class_name(), "Guy")
class PersonTest(TestCase): # 作用域之外,依然返回真實值 def test_should_get_class_name(self, ): mock_get_class_name = Mock(return_value="Guy") with patch("your.package.module.Person.get_class_name", mock_get_class_name): self.assertEqual(Person.get_class_name(), "Guy") self.assertEqual(Person.get_class_name(), "Person")mock鏈式調用
在django里,我們經常需要mock數據庫,而訪問數據庫時經常是鏈式調用,看個例子。
def get_person(name): return Person.objects.filter(name=name).order_by("age")
有個模塊方法,返回數據庫中所有指定name的人員,并按age排序
mock掉整個數據庫訪問
@patch("your.package.module.Person.objects.filter") def test_should_get_person(self, mock_filter): # 先得到一個filter的Mock對象,再在return_value中設置一個Mock對象,此時不需要自己再創建 mock_filter.return_value.order_by.return_value = None self.assertIsNone(get_person())
文章版權歸作者所有,未經允許請勿轉載,若此文章存在違規行為,您可以聯系管理員刪除。
轉載請注明本文地址:http://specialneedsforspecialkids.com/yun/45535.html
摘要:每個測試方法的名稱以單詞開頭,單元測試是如何識別它們是測試的。它還意味著知道測試文件中有多少單元測試,而不是簡單地知道有多少表達式您可能已經注意到將每個行作為單獨的測試計數。 showImg(https://segmentfault.com/img/remote/1460000019140153); 來源 | 愿碼(ChainDesk.CN)內容編輯 愿碼Slogan | 連接每個程...
摘要:準確的說,是中一個用于支持單元測試的庫,它的主要功能是使用對象替代掉指定的對象,以達到模擬對象的行為。下面我們使用對象在單元測試中分別測試訪問正常和訪問不正常的情況。 Mock是什么 Mock這個詞在英語中有模擬的這個意思,因此我們可以猜測出這個庫的主要功能是模擬一些東西。準確的說,Mock是Python中一個用于支持單元測試的庫,它的主要功能是使用mock對象替代掉指定的Python...
摘要:接下來我們將介紹如何對對象的方法進行模擬測試。選項創建模擬測試接口我們可以在的構造函數中提供一個模擬測試實例,而不是模擬創建具體的模擬測試方法。 如何不靠耐心測試 通常,我們編寫的軟件會直接與那些我們稱之為骯臟的服務交互。通俗地說,服務對我們的應用來說是至關重要的,它們之間的交互是我們設計好的,但這會帶來我們不希望的副作用——就是那些在我們自己測試的時候不希望的功能。 比如,可能我們...
摘要:也就是說,如果不需要,兩者使用起來并沒有什么分別。來看個例子,先定義個類,里面只有一個成員方法,返回倍的數值使用類來掉這個成員方法使用類來兩者沒有任何區別,都成功了了成員方法。再看下兩者的區別因為使用類時,默認不會創建這個的,所以報錯。 Python的unittest.mock模塊中提供了兩個主要的mock類,分別是Mock和MagicMock. 先看一下官方文檔的定義: MagicM...
閱讀 2312·2021-11-17 09:33
閱讀 843·2021-10-13 09:40
閱讀 574·2019-08-30 15:54
閱讀 778·2019-08-29 15:38
閱讀 2417·2019-08-28 18:15
閱讀 2475·2019-08-26 13:38
閱讀 1842·2019-08-26 13:36
閱讀 2129·2019-08-26 11:36