3

我刚刚开始在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

在我使用数据库的旧测试中，我有这个来测试我的模型方法：

    ...

    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.

4

3 回答 3

4

您可以模拟该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 查询集功能而无需访问数据库的django_mock_queries项目。它使我们的许多测试变得更快。

于 2016-12-10T02:02:58.933 回答
3
  1. 你的测试看起来很奇怪

  2. 您可以测试模型方法而无需在数据库中创建记录，只是不要使用 .create() 或 .save()：

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

  3. 有时模拟 .save() 方法以禁用在数据库中创建记录和加速测试是一个不错的决定。查看几个如何在测试中使用模拟的示例：http: //nanvel.name/2014/04/testing-django-application#patchmock

于 2014-05-13T14:37:17.567 回答
1

尝试使用Mixer：

from mixer.backend.django import mixer

skill = mixer.blend(Skill, category__title='foo')
userprofile = mixer.blend(UserProfile, user__username='john')
于 2014-01-07T19:30:28.073 回答