So my concrete problem is that I am trying to create a unit test for something like this:

from mongo_models import Record

async def do_something(record_id: str):

    record = await Record.find_one(Record.id == record_id)

    if record is None:
        record = Record(id='randomly_generated_string', content='')

    # Other operations with `record` but we can ignore them for this

    return record

Where the mongo_models.py contains:

from beanie import Document

class Record(Document):
    id: str
    content: str

So I tried doing something like this:

import pytest

from core_code import do_something

@pytest.mark.asyncio
async def test_do_something():
    """ Test do_something method."""

    # Create a mock for the object that will be returned by find_one
    record_mock = AsyncMock(spec=Record)
    record_mock.id = "test-id"
    record_mock.content = "Test content"

    # Test with the find_one method patched
    with patch('mongo_models.Record.find_one', return_value=record_mock) as mock_find_one:

        result = await do_something(record_id="input_id")

        # Assert that find_one was called
        mock_find_one.assert_awaited_once()

        # Assert the right object is being used
        assert result == record_mock

But I am getting an AttributeError: id error when is executed the instruction:
record = await Record.find_one(Record.id == record_id) line.