django.db.utils.IntegrityError,同时为我的视图编写APITestCase吗?

时间:2019-11-27 15:06:27

标签: django django-rest-framework

我在这里为我的视图编写测试用例,但是测试失败并出现以下错误。

  

django.db.utils.IntegrityError:“ content_type_id”列中的值为空

     

违反了非null约束的详细信息:失败行包含(31,permission_name_1,null,perms1)。

我正在编写自己的角色模型,因为我需要更改ManyToManyGroup模型的默认User关系。

class CreateRoleTest(APITestCase):

    def setUp(self):
        self.user = User.objects.create_user(username='username',password='passwprd@321',email='email@gmail.com')
        self.permission = Permission.objects.create(name='permission_name',codename='perms')
        self.client = APIClient()

    def test_create(self):
        url = reverse('roles:create_role')
        data = {'user':self.user,'permissions':[self.permission.pk],'name':'role_name'}
        response = self.client.post(url,data,format='json')
        print(response.json())
        return self.assertEqual(response.status_code,status.HTTP_201_CREATED)

class UpdateDeleteRoleTest(APITestCase):
    def setUp(self) -> None:
        self.client = APIClient()
        self.user = User.objects.create_user(username='username1', password='passwprd@3211',
                                                         email='email@gmail.com1')
        self.permission = Permission.objects.create(name='permission_name_1', codename='perms1')
        self.role = Role.objects.create(name='role1',user=self.user,permissions=[self.permission.pk])

    def test_update(self):
        data = {'user':self.user,'permissions':[self.permission.pk],'name':'role_name2'}
        url = reverse('roles:update_role',kwargs={'pk':self.role.pk})
        response = self.client.put(url,data,format='json')
        print(response.json())
        return self.assertEqual(response.status_code,status.HTTP_200_OK)

models.py

class Role(models.Model):
    user = models.OneToOneField(User,on_delete=models.CASCADE)
    name = models.CharField(max_length=255)
    permissions = models.ManyToManyField(Permission)
    created = models.DateTimeField(auto_now_add=True)

views.py

class CreateRole(CreateAPIView):
    serializer_class = RoleSerializer
    queryset = Role.objects.all()

class UpdateRole(UpdateAPIView):
    serializer_class = RoleSerializer
    queryset = Role.objects.all()

    def put(self, request, *args, **kwargs):
        return self.partial_update(request, *args, **kwargs)

2 个答案:

答案 0 :(得分:1)

好吧,它说它缺少content_type_id列的值。 这是Django自动生成的一列,用于跟踪您的模型。这条线很可能是错误的。

self.permission = Permission.objects.create(name='permission_name_1', codename='perms1')

根据Django Docs,如果您需要以编程方式创建权限(可以通过管理员创建权限,但这确实不利于测试),则应首先获取准确生成的content_type_id,然后在创建权限时添加它。

content_type = ContentType.objects.get_for_model(Role)
self.permission = Permission.objects.create(name='permission_name_1', codename='perms1', content_type=content_type)

测试功能也不需要return self.assertEqual(**)。只需self.assertEqual(**)就足够了。

答案 1 :(得分:0)

将其导入到您的tests.py 文件中

from django.contrib.contenttypes.models import ContentType 

在创建用户之后添加此行并在向用户添加权限之前解决了问题

ContentType.objects.clear_cache()