added test for comments

cookbook/tests/edits/test_edits_comment.py

@@ -0,0 +1,49 @@
+from django.contrib import auth
+from django.urls import reverse
+
+from cookbook.models import Comment, Recipe
+from cookbook.tests.views.test_views import TestViews
+
+
+class TestEditsComment(TestViews):
+
+    def setUp(self):
+        super(TestEditsComment, self).setUp()
+
+        self.recipe = Recipe.objects.create(
+            internal=True,
+            instructions='Do something',
+            working_time=1,
+            waiting_time=1,
+            created_by=auth.get_user(self.user_client_1)
+        )
+
+        self.comment = Comment.objects.create(
+            text='TestStorage',
+            created_by=auth.get_user(self.guest_client_1),
+            recipe=self.recipe
+        )
+        self.url = reverse('edit_comment', args=[self.comment.pk])
+
+    def test_new_comment(self):
+        r = self.user_client_1.post(reverse('view_recipe', args=[self.recipe.pk]), {'comment-text': 'Test Comment Text', 'comment-recipe': self.recipe.pk})
+        self.assertEqual(r.status_code, 200)
+
+    def test_edit_comment_permissions(self):
+        r = self.anonymous_client.get(self.url)
+        self.assertEqual(r.status_code, 302)
+
+        r = self.guest_client_1.get(self.url)
+        self.assertEqual(r.status_code, 200)
+
+        r = self.guest_client_2.get(self.url)
+        self.assertEqual(r.status_code, 302)
+
+        r = self.user_client_1.get(self.url)
+        self.assertEqual(r.status_code, 302)
+
+        r = self.admin_client_1.get(self.url)
+        self.assertEqual(r.status_code, 302)
+
+        r = self.superuser_client.get(self.url)
+        self.assertEqual(r.status_code, 200)

cookbook/views/edit.py

@@ -217,7 +217,7 @@ def edit_storage(request, pk):
         pseudo_instance.token = '__NO__CHANGE__'
         form = StorageForm(instance=pseudo_instance)
 
-    return render(request, 'generic/edit_template.html', {'form': form})
+    return render(request, 'generic/edit_template.html', {'form': form, 'title': _('Storage')})
 
 
 class CommentUpdate(OwnerRequiredMixin, UpdateView):