嘲笑python类字典

时间:2017-01-06 05:27:46

标签: python mocking magicmock

我对python很新。我正在尝试使用mock来编写单元测试。这是代码的模式。

# mod3.py
import mod1.class1
import mod2.class2

d = {
"c1": class1
"c2": class2
} 

def func1(c, v):
   cl = d[c]
   o = cl().meth1(v)
   return o

我想为func1写一个单元测试。

def test_func1(c, v):
   c, v = mock.Mock(), mock.Mock()
   r = mod3.func1(c,v)
   e = {"key1": "value1"}
   #want to check if the ret val is as expected

我如何使用mock基本上模拟cl().meth1(v)

1 个答案:

答案 0 :(得分:0)

# cat class1.py
def func(v):
    return v
# cat class2.py
def func(v):
    return v
#cat mock.py
import class1, class2
d = { "c1": class1, "c2": class2 }
def func1(c, v):
    cl = d[c]
    o = cl.func(v)
    return o

class1和class2是python中的模块。你想要这个吗?