名單調用蟒蛇測試

Question

我有一個對象

class Obj: 
    def method1(self): 
     print 'method1' 

    def method2(self): 
     print 'method2' 

    def method3(self): 
     print 'method3' 

和功能

def do_something(): 
    obj = Obj() 
    obj.method2() 
    obj.method1() 
    obj.method3() 

我想要寫的測試，測試do_something和的OBJ對象。 如何接收在obj上調用的方法列表而不替換（嘲笑）並改變obj行爲？

喜歡的東西

['method2', 'method1', 'method3'] 

Answer 1

使用trace包。見文檔：http://docs.python.org/2/library/trace.html

從文檔：

import sys 
import trace 

# create a Trace object, telling it what to ignore, and whether to 
# do tracing or line-counting or both. 
tracer = trace.Trace(
    ignoredirs=[sys.prefix, sys.exec_prefix], 
    trace=0, 
    count=1) 

# run the new command using the given tracer 
tracer.run('main()') 

# make a report, placing output in the current directory 
r = tracer.results() 
r.write_results(show_missing=True, coverdir=".") 

Answer 2

您可以創建一個通用的Wrapper類將封裝的對象，並跟蹤更改。

class Obj: 
    def method1(self): 
     print 'method1' 

    def method2(self): 
     print 'method2' 

    def method3(self): 
     print 'method3' 

class Wrapper: 
    def __init__(self, wrapped): 
     self.calls = [] 
     self._wrapped = wrapped 
    def __getattr__(self, n): 
     self.calls.append(n) 
     return getattr(self._wrapped, n) 

通過重新定義__getattr__我們使所有屬性訪問在包裝上，以獲取在被包裝對象的屬性。通過以上的定義，我可以做到以下幾點：

>>> obj = Obj() 
>>> x = Wrapper(obj) 
>>> x.calls 
[] 
>>> x.method2() 
method 2 
>>> x.method1() 
method 1 
>>> x.method3() 
method 3 
>>> x.calls 
['method2', 'method1', 'method3'] 
>>> x.method1() 
method 1 
>>> x.method1() 
method 1 
>>> x.calls 
['method2', 'method1', 'method3', 'method1', 'method1'] 

可以進一步提高Wrapper到__getattr__滿足您的需求。 （記錄方法調用的時間戳，記錄輸出，記錄到數據庫等）