From 7db97f22262258799606421fd7bea42a19750f5e Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Thu, 8 Aug 2019 22:45:10 +1000
Subject: [PATCH] Test Part QR Code view

---
 InvenTree/part/test_views.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/InvenTree/part/test_views.py b/InvenTree/part/test_views.py
index ae268e873a..acf24eb515 100644
--- a/InvenTree/part/test_views.py
+++ b/InvenTree/part/test_views.py
@@ -66,3 +66,27 @@ class PartDetailTest(PartViewTestCase):
         self.assertEqual(response.status_code, 200)
         self.assertTrue(response.context['editing_enabled'])
 
+
+class PartQRTest(PartViewTestCase):
+    """ Tests for the Part QR Code AJAX view """
+
+    def test_html_redirect(self):
+        # A HTML request for a QR code should be redirected (use an AJAX request instead)
+        response = self.client.get(reverse('part-qr', args=(1,)))
+        self.assertEqual(response.status_code, 302)
+
+    def test_valid_part(self):
+        response = self.client.get(reverse('part-qr', args=(1,)), HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        self.assertEqual(response.status_code, 200)
+        
+        data = str(response.content)
+
+        self.assertIn('Part QR Code', data)
+        self.assertIn('<img src=', data)
+
+    def test_invalid_part(self):
+        response = self.client.get(reverse('part-qr', args=(9999,)), HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+
+        data = str(response.content)
+        
+        self.assertIn('Error:', data)