From 38a8ac1eb477b69c5a3fdef8541cc158b2c71aea Mon Sep 17 00:00:00 2001
From: David Taylor <david@taylorhq.com>
Date: Wed, 10 May 2017 20:25:14 +0100
Subject: [PATCH] Add failing tests for event paperwork printing

---
 RIGS/test_unit.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/RIGS/test_unit.py b/RIGS/test_unit.py
index 0b22d84e..5b531636 100644
--- a/RIGS/test_unit.py
+++ b/RIGS/test_unit.py
@@ -216,6 +216,39 @@ class TestInvoiceDelete(TestCase):
         self.assertTrue(models.Invoice.objects.get(pk=self.invoices[1].pk))
 
 
+class TestPrintPaperwork(TestCase):
+    @classmethod
+    def setUpTestData(cls):
+        cls.profile = models.Profile.objects.create(username="testuser1", email="1@test.com", is_superuser=True, is_active=True, is_staff=True)
+
+        cls.vatrate = models.VatRate.objects.create(start_at='2014-03-05', rate=0.20, comment='test1')
+
+        cls.events = {
+            1: models.Event.objects.create(name="TE E1", start_date=date.today()),
+        }
+
+        cls.invoices = {
+            1: models.Invoice.objects.create(event=cls.events[1]),
+        }
+
+    def setUp(self):
+        self.profile.set_password('testuser')
+        self.profile.save()
+        self.assertTrue(self.client.login(username=self.profile.username, password='testuser'))
+
+    def test_print_paperwork_success(self):
+        request_url = reverse('event_print', kwargs={'pk': self.events[1].pk})
+
+        response = self.client.get(request_url, follow=True)
+        self.assertEqual(response.status_code, 200)
+
+    def test_print_invoice_success(self):
+        request_url = reverse('invoice_print', kwargs={'pk': self.invoices[1].pk})
+
+        response = self.client.get(request_url, follow=True)
+        self.assertEqual(response.status_code, 200)
+
+
 class TestEmbeddedViews(TestCase):
     @classmethod
     def setUpTestData(cls):