【发布时间】:2020-04-20 17:29:01
【问题描述】:
现在我正在学习使用 Django 创建 REST API,但我正在努力实现 Update 方法。我的测试总是失败,我不确定是因为序列化程序还是其他原因。希望对我的代码提供一些快速反馈!
class Company(models.Model):
owner = models.ForeignKey(User, on_delete=models.CASCADE)
name = models.CharField(max_length=50)
class JobPost(models.Model):
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
company = models.ForeignKey(Company, on_delete=models.CASCADE, related_name='guest_posts')
heading = models.CharField(max_length=100, default="")
class CompanySerializer(serializers.ModelSerializer):
class Meta:
model = Company
fields = ('name')
class JobPostSerializer(serializers.ModelSerializer):
company = CompanySerializer(read_only=True, required=False)
class Meta:
model = GuestPost
fields = ('id', 'heading', 'company')
def create(self, validated_data):
request = self.context.get('request')
company = Company.objects.get(title=request.data.get('company_name'))
return GuestPost.objects.create(owner=request.user,
heading=validated_data.get('heading'),
company=company)
def update(self, instance, validated_data):
instance.heading = validated_data.pop('heading')
instance.save()
return instance
class JobPostViewSet(viewsets.ViewSet,
viewsets.generics.ListCreateAPIView,
viewsets.mixins.UpdateModelMixin,
viewsets.mixins.RetrieveModelMixin):
permission_classes = (JobPostPermissions,)
serializer_class = JobPostSerializer
queryset = JobPost.objects.all()
然后我测试此代码,我对其他操作(创建、获取、列表)的所有测试都可以正常工作,但是在使用以下方法测试更新时出现错误 415:
class Test(TestCase):
def setUp(self):
self.user = User.objects.create(username="elon")
self.factory = RequestFactory()
self.request_url = '/api/job-post/'
self.boring_company = Company.objects.create(owner=self.user, title="Boring Company")
self.job_post = JobPost.objects.create(company=self.boring_company, heading="Looking for boring person")
self.sample_uuid = self.job_post.id
def test_update_post(self):
data = {
'id': self.sample_uuid,
'heading': 'New Heading',
'description': 'new description',
}
request = self.factory.put(self.request_url, data=data)
view = GuestPostViewSet.as_view({'put': 'update'})
force_authenticate(request, user=self.user)
response = view(request, pk=self.sample_uuid)
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.data['heading'], 'New Heading')
提前致谢!
【问题讨论】:
-
GuestPost型号是什么? -
愚蠢的问题。你能用 REST 客户端重现同样的问题吗?只是想知道是代码错误还是测试错误。
-
你是对的,这是测试。
标签: python django rest unit-testing django-rest-framework