在models.py django中测试一个简单的类

时间:2015-06-16 19:40:49

标签: python django unit-testing

我正在尝试在这里测试PurposeOfVisit类,当我创建对象时保存它并尝试检索它,在终端中它工作正常但在我的测试类中它显示错误。有人可以帮助我,为什么我可能会收到错误。

class PurposeOfVisit(CommonInfo):
    purpose = models.CharField(max_length=50)
    is_visible = models.BooleanField(default=True)

    def __unicode__(self):
        return self.purpose

    class Meta:
        verbose_name = "Purpose of Visit"
        verbose_name_plural = "Purposes of Visit"

class CommonInfo(models.Model):
        create_date = models.DateTimeField(auto_now_add=True)
        update_date = models.DateTimeField(auto_now=True)
        created_by = models.ForeignKey(User, related_name='%(class)s_creator')
        updated_by = models.ForeignKey(User, related_name='%(class)s_modifier')

    class Meta:
        abstract = True

这是我的tests.py

from django.contrib.auth.models import User
from irms.models import CommonInfo
from django.test import TestCase
from django.test.client import Client
from django.core.urlresolvers import reverse
from django.http import HttpRequest
from irms.models import *
from django.utils import unittest
from irms.models import Customer
from irms.models import Email
from django.contrib.auth.models import User
from django.db import IntegrityError
from irms.models import PurposeOfVisit

class PersonTestCase(TestCase):
    def test_PurposeOfVisit(self):
        u= User.objects.get(id=1)
        obj= PurposeOfVisit.objects.create(purpose='Badge',
        is_visible=False, created_by=u,updated_by=u)
        obj.save()
        self.assertTrue(isinstance(obj,PurposeOfVisit))

0 个答案:

没有答案