在我的应用程序中,每个模型都有一个类,保存常用查询(我猜这在DDD语言中有点像“仓储”)。这些类中的每一个在构造时都会传递SQLAlchemy会话对象以创建查询。我在确定最佳方式来断言某些查询是否在我的单元测试中运行时遇到了一些困难。使用无处不在的博客示例,假设我有一个“Post”模型,它具有列和属性“date”和“content”。我还有一个“PostRepository”,其中包含方法“find_latest”,该方法应按“date”的降序查询所有帖子。它看起来像这样:
from myapp.models import Post
class PostRepository(object):
def __init__(self, session):
self._s = session
def find_latest(self):
return self._s.query(Post).order_by(Post.date.desc())
我在模拟Post.date.desc()调用时遇到了困难。目前我在单元测试中使用monkey patching来替换Post.date.desc的mock,但是我觉得可能有更好的方法。
编辑:我正在使用mox进行mock对象,我的当前单元测试大致如下:
import unittest
import mox
class TestPostRepository(unittest.TestCase):
def setUp(self):
self._mox = mox.Mox()
def _create_session_mock(self):
from sqlalchemy.orm.session import Session
return self._mox.CreateMock(Session)
def _create_query_mock(self):
from sqlalchemy.orm.query import Query
return self._mox.CreateMock(Query)
def _create_desc_mock(self):
from myapp.models import Post
return self._mox.CreateMock(Post.date.desc)
def test_find_latest(self):
from myapp.models.repositories import PostRepository
from myapp.models import Post
expected_result = 'test'
session_mock = self._create_session_mock()
query_mock = self._create_query_mock()
desc_mock = self._create_desc_mock()
# Monkey patch
tmp = Post.date.desc
Post.date.desc = desc_mock
session_mock.query(Post).AndReturn(query_mock)
query_mock.order_by(Post.date.desc().AndReturn('test')).AndReturn(query_mock)
query_mock.offset(0).AndReturn(query_mock)
query_mock.limit(10).AndReturn(expected_result)
self._mox.ReplayAll()
r = PostRepository(session_mock)
result = r.find_latest()
self._mox.VerifyAll()
self.assertEquals(expected_result, result)
Post.date.desc = tmp
虽然感觉很丑陋,但这确实有效。我不确定为什么没有 "AndReturn('test')" 这部分代码就会失败。