From a9fe82106729210e88e5b9d4f1cd69b285bd0a2f Mon Sep 17 00:00:00 2001
From: vabene1111 <vabene1234@googlemail.com>
Date: Mon, 27 Apr 2020 17:48:11 +0200
Subject: [PATCH] added test for comments

---
 cookbook/tests/edits/test_edits_comment.py | 49 ++++++++++++++++++++++
 cookbook/views/edit.py                     |  2 +-
 2 files changed, 50 insertions(+), 1 deletion(-)
 create mode 100644 cookbook/tests/edits/test_edits_comment.py

diff --git a/cookbook/tests/edits/test_edits_comment.py b/cookbook/tests/edits/test_edits_comment.py
new file mode 100644
index 00000000..11c53316
--- /dev/null
+++ b/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)
diff --git a/cookbook/views/edit.py b/cookbook/views/edit.py
index 2bbf9dfe..5c845b43 100644
--- a/cookbook/views/edit.py
+++ b/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):