def test_refresh_legit(self, mock_db):
find_one = mock_db.__getitem__.return_value.find_one
find_one.return_value = {'test': True}
doc = Document(_id=42)
doc.test = Field()
doc.refresh()
find_one.assert_called_once_with({'_id': 42}, session=None)
assert doc.test is True
def test_refresh_without_id(self):
doc = Document(collection='test')
with pytest.raises(ValueError):
doc.refresh()