2016-02-15 51 views
0

我有以下類:如何在類實例化過程中模擬正在使用setattr調用的方法調用?

class ClassA(Object): 
    VERSIONS = { 
     '1': { 
      'x': a.b.x.X # where x is the module and X is the class 
     } 
    }  
    ATTRS = ['y', 'z'] 
    def __init__(self, **kwargs): 
     ... 
     do_something... 
     ... 
     for attr in ATTRS: 
      setattr(self, attr, VERSIONS[ver][attr]()) 

我使用上面的代碼如下:

class_a = ClassA() 
class_a.x.y 

其中x是一個模塊,y是該模塊

我需要在該方法中現在嘲笑這個電話。我如何嘲笑它,因爲方法y不是ClassA的一部分?

回答

0

這是可以做到如下:

mock_class_a.return_value.x.y = value_to_return 
相關問題