django-helpdesk/helpdesk/tests/test_ticket_lookup.py

# -*- coding: utf-8 -*-
from django.contrib.auth import get_user_model
from django.urls import reverse
from django.test import TestCase
from helpdesk.models import Ticket, Queue
from django.test.utils import override_settings


User = get_user_model()


@override_settings(
    HELPDESK_VIEW_A_TICKET_PUBLIC=True
)
class TestTicketLookupPublicEnabled(TestCase):
    def setUp(self):
        q = Queue(title='Q1', slug='q1')
        q.save()
        t = Ticket(title='Test Ticket', submitter_email='test@domain.com')
        t.queue = q
        t.save()
        self.ticket = t

    def test_ticket_by_id(self):
        """Can a ticket be looked up by its ID"""
        # get the ticket from models
        t = Ticket.objects.get(id=self.ticket.id)
        self.assertEqual(t.title, self.ticket.title)

    def test_ticket_by_link(self):
        """Can a ticket be looked up by its link from (eg) an email"""
        # Instead of using the ticket_for_url link,
        # we will exercise 'reverse' to lookup/build the URL
        # from the ticket info we have
        # http://example.com/helpdesk/view/?ticket=q1-1&email=None
        response = self.client.get(reverse('helpdesk:public_view'),
                                   {'ticket': self.ticket.ticket_for_url,
                                    'email': self.ticket.submitter_email})
        self.assertEqual(response.status_code, 200)

    def test_ticket_with_changed_queue(self):
        # Make a ticket (already done in setup() )
        # Now make another queue
        q2 = Queue(title='Q2', slug='q2')
        q2.save()
        # grab the URL / params which would have been emailed out to submitter.
        url = reverse('helpdesk:public_view')
        params = {'ticket': self.ticket.ticket_for_url,
                  'email': self.ticket.submitter_email}
        # Pickup the ticket created in setup() and change its queue
        self.ticket.queue = q2
        self.ticket.save()

        # confirm that we can still get to a url which was emailed earlier
        response = self.client.get(url, params)
        self.assertNotContains(response, "Invalid ticket ID")

    def test_add_email_to_ticketcc_if_not_in(self):
        staff_email = 'staff@mail.com'
        staff_user = User.objects.create(
            username='staff', email=staff_email, is_staff=True)
        self.ticket.assigned_to = staff_user
        self.ticket.save()
        email_1 = 'user1@mail.com'
        ticketcc_1 = self.ticket.ticketcc_set.create(email=email_1)

        # Add new email to CC
        email_2 = 'user2@mail.com'
        ticketcc_2 = self.ticket.add_email_to_ticketcc_if_not_in(email=email_2)
        self.assertEqual(list(self.ticket.ticketcc_set.all()),
                         [ticketcc_1, ticketcc_2])

        # Add existing email, doesn't change anything
        self.ticket.add_email_to_ticketcc_if_not_in(email=email_1)
        self.assertEqual(list(self.ticket.ticketcc_set.all()),
                         [ticketcc_1, ticketcc_2])

        # Add mail from assigned user, doesn't change anything
        self.ticket.add_email_to_ticketcc_if_not_in(email=staff_email)
        self.assertEqual(list(self.ticket.ticketcc_set.all()),
                         [ticketcc_1, ticketcc_2])
        self.ticket.add_email_to_ticketcc_if_not_in(user=staff_user)
        self.assertEqual(list(self.ticket.ticketcc_set.all()),
                         [ticketcc_1, ticketcc_2])

        # Move a ticketCC from ticket 1 to ticket 2
        ticket_2 = Ticket.objects.create(
            queue=self.ticket.queue, title='Ticket 2', submitter_email=email_2)
        self.assertEqual(ticket_2.ticketcc_set.count(), 0)
        ticket_2.add_email_to_ticketcc_if_not_in(ticketcc=ticketcc_1)
        self.assertEqual(ticketcc_1.ticket, ticket_2)
        self.assertEqual(ticket_2.ticketcc_set.count(), 1)

        # Adding email_2 doesn't change since it is already submitter email
        ticket_2.add_email_to_ticketcc_if_not_in(email=email_2)
        self.assertEqual(ticket_2.ticketcc_set.get(), ticketcc_1)
        ticket_2.add_email_to_ticketcc_if_not_in(ticketcc=ticketcc_2)
        self.assertEqual(ticket_2.ticketcc_set.get(), ticketcc_1)

        # Finally test function raises a Value error when no parameter is given
        self.assertRaises(ValueError, ticket_2.add_email_to_ticketcc_if_not_in)