def test_find_by_jti(self, mock_query: Mock) -> None: expected = Token(id=123) filter_by = mock_query.return_value.filter_by first = filter_by.return_value.first first.return_value = expected result = Token.find_by_jti('name-xyz') filter_by.assert_called_once_with(jti='name-xyz') first.assert_called_once_with() assert result == expected