From 7faaf824e211a5a4b611a62d9eea456b6306434b Mon Sep 17 00:00:00 2001
From: jordan <j.schatzman@outlook.com>
Date: Thu, 16 Feb 2017 21:35:16 -0800
Subject: [PATCH] added a few more tests for checking add owner functionality

---
 pychart/pychart_datarender/tests.py | 52 +++++++++++++++++++++++++----
 1 file changed, 45 insertions(+), 7 deletions(-)

diff --git a/pychart/pychart_datarender/tests.py b/pychart/pychart_datarender/tests.py
index 8c17151..e0f8d23 100644
--- a/pychart/pychart_datarender/tests.py
+++ b/pychart/pychart_datarender/tests.py
@@ -177,7 +177,7 @@ def test_render_is_bar_graph(self):
         this_render.save()
         self.assertTrue(self.renders[0].render_type == "Bar")
 
-    def test_render_is_a_historgram(self):
+    def test_render_is_a_histogram(self):
         """Test photo has their published setting as Public."""
         this_render = self.renders[0]
         this_render.render_type = "Histogram"
@@ -191,6 +191,16 @@ def render_has_a_description(self):
         this_render.save()
         self.assertTrue(self.renders[0].description == "My render description")
 
+    def test_data_has_multiple_owners(self):
+        """Test that adding mulitple renders to a data object works."""
+        data_set = Data.objects.first()
+        profile = User.objects.all()
+        user1 = profile[0].profile
+        user2 = profile[1].profile
+        data_set.owner.add(user1)
+        data_set.owner.add(user2)
+        self.assertTrue(len(data_set.owner.all()) == 2)
+
 
 class FrontEndTests(TestCase):
     """Adding functional tests."""
@@ -287,12 +297,40 @@ def test_data_edit_return_200(self):
                                                 kwargs={'pk': data.id}))
         self.assertTrue(response.status_code == 200)
 
-    # def test_render_detail_return_200(self):
-    #     """Test that data detail page is accessible."""
-    #     render = Render.objects.first()
-    #     response = self.client.get(reverse_lazy('render_detail',
-    #                                             kwargs={'pk': render.id}))
-    #     self.assertTrue(response.status_code == 200)
+    def test_render_detail_return_200(self):
+        """Test that data detail page is accessible."""
+        render = Render.objects.first()
+        response = self.client.get(reverse_lazy('render_detail',
+                                                kwargs={'pk': render.id}))
+        self.assertTrue(response.status_code == 200)
+
+    def test_add_owner_return_302(self):
+        """Test that add owner page redirects."""
+        user = self.user_login()
+        self.client.force_login(user)
+        data = Data.objects.first()
+        response = self.client.get(reverse_lazy('add_owner',
+                                                kwargs={'pk': data.id}))
+        self.assertTrue(response.status_code == 302)
+
+    def test_add_owner_adds_data_to_user(self):
+        """Test that add owner page redirects."""
+        user = self.user_login()
+        self.client.force_login(user)
+        data = Data.objects.first()
+        self.client.get(reverse_lazy('add_owner',
+                                     kwargs={'pk': data.id}))
+        self.assertTrue(len(user.profile.data_sets.all()) == 1)
+
+    def test_add_owner_adds_user_to_data(self):
+        """Test that add owner page redirects."""
+        user = self.user_login()
+        self.client.force_login(user)
+        data = Data.objects.first()
+        owner_count = len(data.owner.all())
+        self.client.get(reverse_lazy('add_owner',
+                                     kwargs={'pk': data.id}))
+        self.assertTrue(len(data.owner.all()) == owner_count + 1)
 
 
 class RenderTests(TestCase):