From bd653f2c492cd20a77ad381e136f285c26b42eaf Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Tue, 7 Jan 2020 21:16:01 +1100
Subject: [PATCH] Some more unit tests

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

diff --git a/InvenTree/part/test_views.py b/InvenTree/part/test_views.py
index ecf4c936d1..840f0a94ca 100644
--- a/InvenTree/part/test_views.py
+++ b/InvenTree/part/test_views.py
@@ -198,6 +198,23 @@ class CategoryTest(PartViewTestCase):
         response = self.client.get(reverse('category-edit', args=(1,)), HTTP_X_REQUESTED_WITH='XMLHttpRequest')
         self.assertEqual(response.status_code, 200)
 
+    def test_set_category(self):
+        """ Test that the "SetCategory" view works """
+
+        url = reverse('part-set-category')
+
+        response = self.client.get(url, {'parts[]': 1}, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        self.assertEqual(response.status_code, 200)
+
+        data = {
+            'part_id_10': True,
+            'part_id_1': True,
+            'part_category': 5
+        }
+
+        response = self.client.post(url, data, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        self.assertEqual(response.status_code, 200)
+
 
 class BomItemTests(PartViewTestCase):
     """ Tests for BomItem related views """