Question

我刚刚开始使用Mock来测试Django应用程序，而无需触及数据库。我可以在测试中成功模拟我的模型对象，但是我对如何处理涉及对象的模型方法感到困惑。

作为一个例子，我在models.py中有这个：

class Skill(models.Model):
   title = models.CharField(max_length=255)
   category = models.ForeignKey(
       SkillCategory, default=None, null=True, blank=True
   )

def __unicode__(self):
    return self.title

class UserProfile(models.Model):
    user = models.OneToOneField(User)
    skill = models.ManyToManyField(Skill, default=None)
    avatar = models.URLField(
        max_length=400, default=None, null=True, blank=True
    )

    def __unicode__(self):
        return self.user.username

    def update_skills(self, cleaned_skills):
        for s in cleaned_skills:
            skill, created = Skill.objects.get_or_create(title=s)
            self.skill.add(skill)

到目前为止，在我的重新考试中，我做到了这一点：

class ProfileTestCase(unittest.TestCase):
    def setUp(self):
        # mock user
        self.user = mock.Mock(spec=User)
        self.user.username = "john"
        self.user.email = "johndoe@hotmail.com"
        self.user.password = "pass"
        # mock profile
        self.userprofile = mock.Mock(spec=UserProfile)
        self.userprofile.user = self.user
        # mock skill category
        self.skill_category = mock.Mock(spec=SkillCategory)
        self.skill_category.title = "foo"
        # mock skill
        self.skill = mock.Mock(spec=Skill)
        self.skill.title = "ninja"
        self.skill.category = self.skill_category

    def test_skillcategory_created(self):
        self.assertEqual(self.skill.category.title, 'foo')

    def test_skill_created(self):
        self.assertEqual(self.skill.title, 'ninja')

    ...
    # and so on

在我使用db的旧测试中，我用它来测试我的模型方法：

    ...

    def test_update_skills(self):
        cleaned_skills = ['mysql', 'unix']
        self.userprofile.update_skills(cleaned_skills)
        skills = self.userprofile.skill.values_list('title', flat=True)
        self.assertTrue('mysql' in skills)

我正在摸索如何重构这一行：

self.userprofile.update_skills(cleaned_skills)

利用Mock。

Answer 1

你的测试看起来很奇怪

您可以在不创建数据库记录的情况下测试模型方法，只是不要使用.create（）或.save（）：

def test_my_method(self):
    my_model = MyModel(someattr='someval', ...)
    self.assertEqual(my_model.my_method(), ...)

有时候很好的决定模拟.save（）方法来禁止在数据库和加速测试中创建记录。请参阅几个示例，了解如何在测试中使用模拟：http://nanvel.name/2014/04/testing-django-application#patchmock

Answer 2

尝试使用Mixer：

from mixer.backend.django import mixer

skill = mixer.blend(Skill, category__title='foo')
userprofile = mixer.blend(UserProfile, user__username='john')

Answer 3

您可以模拟get_or_create()方法以及skill关系，以便在不访问数据库的情况下运行测试。

def test_update_skills(self):
    def mock_get_or_create(title):
        return title, True
    mock_skill = set()
    with mock.patch('analytics.models.Skill.objects.get_or_create',
                    mock_get_or_create), \
            mock.patch.object(UserProfile, 'skill', mock_skill):
        userprofile = UserProfile()

        cleaned_skills = ['mysql', 'unix']
        userprofile.update_skills(cleaned_skills)

    self.assertEqual({'mysql', 'unix'}, mock_skill)

我帮助django_mock_queries project帮助提供一堆Django查询集功能，而无需访问数据库。它以更快的速度完成了一系列测试。

模拟Django中的模型方法

3 个答案: