3

下面的 FastApi 测试应该使用我的get_mock_db函数而不是get_db函数，但事实并非如此。目前测试失败，因为它使用真实的数据库。

def get_mock_db():
    example_todo = Todo(title="test title", done=True, id=1)

    class MockDb:
        def query(self, _model):
            mock = Mock()
            mock.get = lambda _param: example_todo

        def all(self):
            return [example_todo]

        def add(self):
            pass

        def commit(self):
            pass

        def refresh(self, todo: CreateTodo):
            return Todo(title=todo.title, done=todo.done, id=1)

    return MockDb()


client = TestClient(app)


app.dependency_overrides[get_db] = get_mock_db


def test_get_all():
    response = client.get("/api/v1/todo")
    assert response.status_code == 200
    assert response.json() == [
        {
            "title": "test title",
            "done": True,
            "id": 1,
        }
    ]
4

1 回答 1

1

关键是要了解dependency_overrides 只是一个字典。为了覆盖某些东西，您需要指定一个与原始依赖项匹配的键。

def get_db():
   return {'db': RealDb()}

def home(commons: dict= Depends(get_db))
   commons['db'].doStuff()
 
app.dependency_overrides[get_db] = lambda: {'db': MockDb()}

在这里，您在Depends函数内部调用了对get_db函数的引用。然后，您指的是与 完全相同的功能dependency_overrides[get_db]。因此它被覆盖。首先验证这两个中的“xxx”是否完全匹配：Depends(xxx) 和 dependency_overrides[xxx]。

我花了一些时间来理解这样一个事实，即Depends调用中的任何内容实际上都是依赖项的标识符。所以在这个例子中，标识符是函数 get_db 并且相同的函数被用作字典中的键。

因此，这意味着以下示例不起作用，因为您覆盖的不是为 Depends 指定的内容。

def get_db(connection_string):
   return {'db': RealDb(connection_string)}

def home(commons: dict= Depends(get_db(os.environ['connectionString']))
   commons['db'].doStuff()

# Does not work 
app.dependency_overrides[get_db] = lambda: {'db': MockDb()}
于 2021-04-03T05:57:29.597 回答