Add accounts, tickets and checkout test files

Author: tdunn891

accounts/test_accounts.py

@@ -0,0 +1,18 @@
+
+from django.test import TestCase, Client
+from django.shortcuts import get_object_or_404
+from accounts.models import User, Profile
+
+
+class TestAccountsModels(TestCase):
+    def setUp(self):
+        # Every test refers to this logged in user
+        username = 'testuser'
+        password = '12345'
+        self.user = User.objects.create_user(
+            username=username, password=password)
+        login = self.client.login(username=username, password=password)
+
+    def test_profile_class_returns_username(self):
+        user = get_object_or_404(Profile, pk=self.user.id)
+        self.assertEqual(str(user), user.user.username)

accounts/test_views.py

@@ -0,0 +1,46 @@
+from django.test import TestCase, Client
+from django.shortcuts import get_object_or_404
+from accounts.models import User, Profile
+
+
+class TestAccountsViewsLoggedIn(TestCase):
+    def setUp(self):
+        # Every test refers to this logged in user
+        username = 'testuser'
+        password = '12345'
+        self.user = User.objects.create_user(
+            username=username, password=password)
+        login = self.client.login(username=username, password=password)
+
+    def test_get_user_profile_page(self):
+        page = self.client.get("/accounts/profile/")
+        self.assertEqual(page.status_code, 200)
+        self.assertTemplateUsed(page, "profile.html")
+
+    def test_redirect_to_home_page_after_logout(self):
+        page = self.client.get("/accounts/logout/")
+        self.assertRedirects(page, "/")
+
+    def test_redirect_to_tickets_after_successful_login(self):
+        page = self.client.get("/accounts/logout/")
+        self.assertRedirects(page, "/")
+
+    # POSTS
+    def test_staff_access_is_granted(self):
+        response = self.client.post(
+            "/accounts/grant_staff_access/{}/".format(self.user.id))
+        user = get_object_or_404(User, pk=self.user.id)
+        self.assertEqual(user.is_staff, True)
+        self.assertRedirects(response, "/accounts/profile/")
+
+
+class TestAccountsViewsNotLoggedIn(TestCase):
+    def test_get_registration_page(self):
+        page = self.client.get("/accounts/register/")
+        self.assertEqual(page.status_code, 200)
+        self.assertTemplateUsed(page, "registration.html")
+
+    def test_get_login_page(self):
+        page = self.client.get("/accounts/login/")
+        self.assertEqual(page.status_code, 200)
+        self.assertTemplateUsed(page, "login.html")

checkout/test_forms.py

@@ -2,12 +2,6 @@
 from .forms import MakePaymentForm, OrderForm
 
 
-# class TestMakePaymentForm(TestCase):
-#     def test_correct_message_for_missing_credit_card_number(self):
-#         form = MakePaymentForm({'credit_card_number': ''})
-#         self.assertFalse(form.is_valid)
-
-
 class TestOrderForm(TestCase):
     def test_correct_message_for_missing_full_name(self):
         form = OrderForm({'full_name': ''})
@@ -20,8 +14,3 @@ def test_correct_message_for_missing_street_address1(self):
         self.assertEqual(form.errors['street_address1'], [
                          u'This field is required.'])
         self.assertFalse(form.is_valid())
-
-    # def test_no_field_error_message_if_missing_street_address2(self):
-    #     form = OrderForm({'street_address2': ''})
-    #     self.assertFalse(form.errors['street_address2'])
-        # self.assertFalse(form.is_valid())

tickets/test_forms.py

@@ -0,0 +1,44 @@
+from django.test import TestCase
+from .forms import AddTicketForm, EditTicketForm, AddCommentForm
+
+
+class TestAddTicketForm(TestCase):
+    def test_cannot_create_a_ticket_with_just_ticket_type_equals_bug(self):
+        form = AddTicketForm({'ticket_type': 'Bug'})
+        self.assertFalse(form.is_valid())
+
+    def test_cannot_create_a_ticket_with_other_ticket_type(self):
+        form = AddTicketForm({'ticket_type': 'Other'})
+        self.assertFalse(form.is_valid())
+        self.assertEqual(form.errors['ticket_type'],
+                         [u'Select a valid choice. Other is not one of the available choices.'])
+
+    def test_correct_message_for_missing_ticket_type(self):
+        form = AddTicketForm({'ticket_type': ''})
+        self.assertFalse(form.is_valid())
+        self.assertEqual(form.errors['ticket_type'], [
+                         u'This field is required.'])
+
+    def test_correct_message_for_missing_priority(self):
+        form = AddTicketForm({'priority': ''})
+        self.assertFalse(form.is_valid())
+        self.assertEqual(form.errors['priority'], [
+                         u'This field is required.'])
+
+    def test_correct_message_for_missing_status(self):
+        form = AddTicketForm({'status': ''})
+        self.assertFalse(form.is_valid())
+        self.assertEqual(form.errors['status'], [
+                         u'This field is required.'])
+
+    def test_correct_error_message_for_missing_summary(self):
+        form = AddTicketForm({'summary': ''})
+        self.assertFalse(form.is_valid())
+        self.assertEqual(form.errors['summary'], [
+                         u'This field is required.'])
+
+    def test_correct_error_message_for_missing_description(self):
+        form = AddTicketForm({'description': ''})
+        self.assertFalse(form.is_valid())
+        self.assertEqual(form.errors['description'], [
+                         u'This field is required.'])

tickets/test_models.py

@@ -0,0 +1,42 @@
+from django.test import TestCase, Client
+from accounts.models import User
+from .models import Ticket, Comment
+
+
+class TestViews(TestCase):
+    def setUp(self):
+        # Every test needs access to this logged in user
+        self.user = User.objects.create_user(
+            username='testuser', password='12345')
+        login = self.client.login(username='testuser', password='12345')
+
+    def test_create_a_ticket_with_summary_and_description(self):
+        # Create Ticket (TicketID = 1)
+        ticket = Ticket(ticket_type="Bug",
+                        summary='Test Summary',
+                        description='Test Description',
+                        status='New',
+                        priority='Low',
+                        submitted_by=self.user,
+                        assigned_to=self.user
+                        )
+        ticket.save()
+        self.assertEqual(ticket.summary, 'Test Summary')
+        self.assertEqual(ticket.description, 'Test Description')
+
+    def test_add_a_comment(self):
+        # Create Ticket (TicketID = 1)
+        ticket = Ticket(ticket_type="Bug",
+                        summary='Test Summary',
+                        description='Test Description',
+                        status='New',
+                        priority='Low',
+                        submitted_by=self.user,
+                        assigned_to=self.user
+                        )
+        ticket.save()
+        comment = Comment(ticket=ticket, user=self.user,
+                          comment_body='Test Comment')
+        comment.save()
+        comment = Comment.objects.get(pk=1)
+        self.assertEqual(str(comment), comment.comment_body)

tickets/test_views.py

@@ -0,0 +1,110 @@
+from django.shortcuts import get_object_or_404
+from django.test import TestCase
+from accounts.models import User, Profile
+from .models import Ticket
+
+
+class TestViews(TestCase):
+    def setUp(self):
+        # Every test needs access to this logged in user
+        self.user = User.objects.create_user(
+            username='testuser', password='12345')
+        login = self.client.login(username='testuser', password='12345')
+
+        # Create Ticket (TicketID = 1)
+        Ticket.objects.create(ticket_type="Bug",
+                              summary='Test Summary',
+                              description='Test Description',
+                              status='New',
+                              priority='Low',
+                              submitted_by=self.user,
+                              assigned_to=self.user
+                              )
+
+    def test_get_index_page(self):
+        page = self.client.get("/")
+        self.assertEqual(page.status_code, 200)
+        self.assertTemplateUsed(page, "index.html")
+
+    def test_get_view_tickets_page(self):
+        page = self.client.get("/tickets/")
+        self.assertEqual(page.status_code, 200)
+        self.assertTemplateUsed(page, "tickets.html")
+
+    def test_add_ticket_page(self):
+        page = self.client.get("/tickets/add_ticket/")
+        self.assertEqual(page.status_code, 200)
+        self.assertTemplateUsed(page, "add_ticket.html")
+
+    def test_dashboard_page(self):
+        page = self.client.get("/tickets/dashboard/")
+        self.assertEqual(page.status_code, 200)
+        self.assertTemplateUsed(page, "dashboard.html")
+
+    def test_kanban_page_as_pro_user(self):
+        # Set user to Pro User
+        self.user.profile.is_pro_user = True
+        self.user.profile.save()
+        page = self.client.get("/tickets/kanban/")
+        self.assertEqual(page.status_code, 200)
+        self.assertTemplateUsed(page, "kanban.html")
+
+    def test_kanban_page_as_basic_user(self):
+        # Set user to Basic
+        self.user.profile.is_pro_user = False
+        self.user.profile.save()
+        page = self.client.get("/tickets/kanban/")
+        # Redirects to Checkout page
+        self.assertRedirects(page, "/checkout/")
+
+    def test_edit_ticket_page(self):
+        page = self.client.get("/tickets/edit_ticket/1/")
+        self.assertEqual(page.status_code, 200)
+        self.assertTemplateUsed(page, "edit_ticket.html")
+
+    def test_view_ticket_page(self):
+        page = self.client.get("/tickets/1/")
+        self.assertEqual(page.status_code, 200)
+        self.assertTemplateUsed(page, "view_ticket.html")
+
+    def test_get_view_page_for_ticket_that_does_not_exist(self):
+        page = self.client.get("tickets/1000/")
+        self.assertEqual(page.status_code, 404)
+
+    def test_get_edit_page_for_ticket_that_does_not_exist(self):
+        page = self.client.get("tickets/edit/1000/")
+        self.assertEqual(page.status_code, 404)
+
+    def test_upvote_for_ticket_that_does_not_exist(self):
+        page = self.client.get("tickets/upvote/1000/")
+        self.assertEqual(page.status_code, 404)
+
+    def test_change_status_for_ticket_that_does_not_exist(self):
+        page = self.client.get("tickets/change_status/1000/Cancelled")
+        self.assertEqual(page.status_code, 404)
+
+    # POSTs
+    def test_post_create_a_ticket(self):
+        new_ticket = {"ticket_type": "Bug",
+                      "summary": 'Test Summary 2',
+                      "description": 'Test Description',
+                      "status": 'New',
+                      "priority": 'Low',
+                      "submitted_by": self.user,
+                      "assigned_to": self.user
+                      }
+        response = self.client.post("/tickets/add_ticket/", new_ticket)
+        ticket = get_object_or_404(Ticket, pk=2)
+        self.assertEqual(ticket.summary, "Test Summary 2")
+
+    def test_post_upvote_a_ticket(self):
+        response = self.client.post(
+            "/tickets/upvote/1/")
+        ticket = get_object_or_404(Ticket, pk=1)
+        self.assertEqual(ticket.upvotes, 1)
+
+    def test_change_ticket_status(self):
+        response = self.client.post(
+            "/tickets/change_status/1/Resolved/")
+        ticket = get_object_or_404(Ticket, pk=1)
+        self.assertEqual(ticket.status, "Resolved")