diff --git a/apps/files/tests.py b/apps/files/tests.py index 2be7872..22812e8 100644 --- a/apps/files/tests.py +++ b/apps/files/tests.py @@ -53,13 +53,15 @@ class FileTests(TestCase): }) data = response.json() - + self.assertIsInstance(data['id'], int) self.assertEqual(data['is_valid'], True) response = self.client.post(reverse('files:delete', kwargs={ 'pk': data['id'] })) + self.assertEqual(response.status_code, 200) + data = response.json() self.assertEqual(data['success'], True) diff --git a/apps/files/views.py b/apps/files/views.py index cbb36ce..83f0e43 100644 --- a/apps/files/views.py +++ b/apps/files/views.py @@ -18,7 +18,8 @@ class FileUploadView(LoginRequiredMixin, FormView): self.object.save() return JsonResponse({ - 'is_valid': True, 'url': self.object.file.url, + 'is_valid': True, + 'url': self.object.file.url, 'name': self.object.file.name, 'id': self.object.pk }) @@ -45,7 +46,7 @@ class FileDeleteView(LoginRequiredMixin, DetailView): raise PermissionDenied self.object.delete() - + return JsonResponse({ 'success': True })