测试使用继承的参数化构造函数

Question

我想测试如下所示的函数

async def post_lookup(self, request: web.Request):
    response = await ZipcodeLookupModel(
        request.app[constants.CONNECTOR], session_store=request._state[constants.SESSION_STORAGE]
    ).lookup(request=request)
    return web.json_response(response)

类 ZipcodeLookupModel 及其层次结构如下所示

class ZipcodeLookupModel(BaseModel):
    def __init__(self, *args, **kwargs):
        super(ZipcodeLookupModel, self).__init__(*args, **kwargs)
        
class BaseModel:
    def __init__(self, connector=None, session_store=None):
        self.session_store = session_store
        self.connector = connector
        self.loop = get_running_loop()
        self.service_args = dict(connector=self.connector, loop=self.loop)

我想修补该对象创建

    with patch(
        "src.api.v1.zipcode.lookup.model.ZipcodeLookupModel",
        new=ZipcodeLookupModelStub()
    ):
        response = await client.session.post(url=f"http://{client.host}:{client.port}{path}", json=self.message)
        assert response.status == 200
        data = await response.json()
        assert data

我已经为测试创建了一个存根对象

class ZipcodeLookupModelStub(BaseModel):
    def __init__(self, *args, **kwargs):
        super(ZipcodeLookupModelStub, self).__init__(*args, **kwargs)

    async def lookup(self, request):

但是当我运行测试时，出现此错误（在 post_lookup 中创建对象的行上），

super() argument 1 must be type, not ZipcodeLookupModelStub

问题是什么？

Answer 1

您要将名称 ZipCodeLookupModel 替换为您要使用的类的实例，而不是类本身。

with patch(
    "src.api.v1.zipcode.lookup.model.ZipcodeLookupModel",
    new=ZipcodeLookupModelStub  # No parentheses
):
    response = await client.session.post(url=f"http://{client.host}:{client.port}{path}", json=self.message)
    assert response.status == 200
    data = await response.json()
    assert data