From db82b64ce940b12dc9e76e1bbd3d170e6c0e5cd3 Mon Sep 17 00:00:00 2001
From: Lucas Hiroshi Horinouchi <lucas.hh@hotmail.com>
Date: Mon, 4 Dec 2017 23:39:08 -0200
Subject: [PATCH] #112 - Fixing tests.

---
 .../recommendation/test/__init__.py           |  2 +-
 .../test_view_reccomendation_custom_delete.py | 39 +++++++++++++++++++
 .../test/test_view_recommendation_delete.py   | 34 ----------------
 .../views/deletecustomrecommendation.py       |  1 -
 4 files changed, 40 insertions(+), 36 deletions(-)
 create mode 100644 medical_prescription/recommendation/test/test_view_reccomendation_custom_delete.py
 delete mode 100644 medical_prescription/recommendation/test/test_view_recommendation_delete.py

diff --git a/medical_prescription/recommendation/test/__init__.py b/medical_prescription/recommendation/test/__init__.py
index b6e690fd..b9742821 100644
--- a/medical_prescription/recommendation/test/__init__.py
+++ b/medical_prescription/recommendation/test/__init__.py
@@ -1 +1 @@
-from . import *
+from . import *
\ No newline at end of file
diff --git a/medical_prescription/recommendation/test/test_view_reccomendation_custom_delete.py b/medical_prescription/recommendation/test/test_view_reccomendation_custom_delete.py
new file mode 100644
index 00000000..54b890a9
--- /dev/null
+++ b/medical_prescription/recommendation/test/test_view_reccomendation_custom_delete.py
@@ -0,0 +1,39 @@
+from django.test import TestCase, RequestFactory, Client
+from recommendation.views import CustomRecommendationDeleteView
+from recommendation.models import CustomRecommendation
+
+from user.models import HealthProfessional
+
+
+class TestArchiveMessageOutboxView(TestCase):
+
+    def setUp(self):
+
+        self.user = HealthProfessional.objects.create(name='User Test',
+                                                      email='test@teste.com',
+                                                      sex='M',
+                                                      phone='1111111111',
+                                                      is_active=True)
+        self.view = CustomRecommendationDeleteView()
+        self.view_class = CustomRecommendationDeleteView
+        self.factory = RequestFactory()
+        self.client = Client()
+        # Create Message.
+
+        self.recommendation = CustomRecommendation()
+        self.recommendation.name = "meu texto"
+        self.recommendation.recommendation = "Assunto"
+        self.recommendation.health_professional = self.user
+        self.recommendation.is_active = True
+        self.recommendation.pk = '1'
+        self.recommendation.save()
+
+    def test_post_true(self):
+        request = self.factory.post('/')
+
+        request.user = self.user
+        self.view.request = request
+        self.view.object = self.recommendation
+
+        recommendation = self.view_class.post(request, pk=1)
+        self.assertEqual(recommendation.status_code, 302)
diff --git a/medical_prescription/recommendation/test/test_view_recommendation_delete.py b/medical_prescription/recommendation/test/test_view_recommendation_delete.py
deleted file mode 100644
index 5eb35336..00000000
--- a/medical_prescription/recommendation/test/test_view_recommendation_delete.py
+++ /dev/null
@@ -1,34 +0,0 @@
-# Django
-from django.test import TestCase
-from django.test.client import RequestFactory
-
-# Django Local
-from recommendation.views import CustomRecommendationDeleteView
-from user.models import HealthProfessional
-from recommendation.models import CustomRecommendation
-
-
-class DeleteRecomendationCustomViewTeste(TestCase):
-    def setUp(self):
-        self.health_professional_1 = HealthProfessional.objects.create_user(email='doctor1@doctor.com',
-                                                                            password='senha12')
-        self.health_professional_2 = HealthProfessional.objects.create_user(email='doctor2@doctor.com',
-                                                                            password='senha12')
-        self.custom_recommendation = CustomRecommendation()
-        self.custom_recommendation.pk = 1
-        self.custom_recommendation.name = "Testando"
-        self.custom_recommendation.description = "Testa isso direito meu amigo"
-        self.custom_recommendation.health_professional = self.health_professional
-        self.custom_recommendation.save()
-        self.view = CustomRecommendationDeleteView()
-        self.factory = RequestFactory()
-
-    def post_custom_recommendation_true(self):
-        request = self.factory.post('/')
-
-        request.user = self.user
-        self.view.request = request
-        self.view.object = self.custom_recommendation
-
-        custom_delete = self.view_class.post(request, pk=1)
-        self.assertEqual(custom_delete.status_code, 302)
diff --git a/medical_prescription/recommendation/views/deletecustomrecommendation.py b/medical_prescription/recommendation/views/deletecustomrecommendation.py
index 63ab15c5..e393de1d 100644
--- a/medical_prescription/recommendation/views/deletecustomrecommendation.py
+++ b/medical_prescription/recommendation/views/deletecustomrecommendation.py
@@ -19,5 +19,4 @@ def post(self, pk):
         custom_recommendation = CustomRecommendation.objects.get(pk=pk)
         custom_recommendation.is_active = False
         custom_recommendation.save()
-        print("FOI A POHA DO POST")
         return HttpResponseRedirect(reverse_lazy('list_custom_recommendations'))