mirror of
https://github.com/inventree/InvenTree.git
synced 2025-04-28 19:46:46 +00:00
149 lines
4.4 KiB
Python
149 lines
4.4 KiB
Python
""" Unit tests for Part Views (see views.py) """
|
|
|
|
from django.test import TestCase
|
|
from django.urls import reverse
|
|
from django.contrib.auth import get_user_model
|
|
|
|
from .models import Part
|
|
|
|
class PartViewTestCase(TestCase):
|
|
|
|
fixtures = [
|
|
'category',
|
|
'part',
|
|
'bom',
|
|
'location',
|
|
'company',
|
|
'supplier_part',
|
|
]
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
|
|
# Create a user
|
|
User = get_user_model()
|
|
User.objects.create_user('username', 'user@email.com', 'password')
|
|
|
|
self.client.login(username='username', password='password')
|
|
|
|
|
|
class PartListTest(PartViewTestCase):
|
|
|
|
def test_part_index(self):
|
|
response = self.client.get(reverse('part-index'))
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
keys = response.context.keys()
|
|
self.assertIn('csrf_token', keys)
|
|
self.assertIn('parts', keys)
|
|
self.assertIn('user', keys)
|
|
|
|
def test_export(self):
|
|
""" Export part data to CSV """
|
|
|
|
response = self.client.get(reverse('part-export'), {'parts': '1,2,3,4,5,6,7,8,9,10'}, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertIn('streaming_content', dir(response))
|
|
|
|
|
|
class PartDetailTest(PartViewTestCase):
|
|
|
|
def test_part_detail(self):
|
|
""" Test that we can retrieve a part detail page """
|
|
|
|
pk = 1
|
|
|
|
response = self.client.get(reverse('part-detail', args=(pk,)))
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
part = Part.objects.get(pk=pk)
|
|
|
|
keys = response.context.keys()
|
|
|
|
self.assertIn('part', keys)
|
|
self.assertIn('category', keys)
|
|
|
|
self.assertEqual(response.context['part'].pk, pk)
|
|
self.assertEqual(response.context['category'], part.category)
|
|
|
|
self.assertFalse(response.context['editing_enabled'])
|
|
|
|
def test_editable(self):
|
|
|
|
pk = 1
|
|
response = self.client.get(reverse('part-detail', args=(pk,)), {'edit': True})
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTrue(response.context['editing_enabled'])
|
|
|
|
def test_bom_download(self):
|
|
""" Test downloading a BOM for a valid part """
|
|
|
|
response = self.client.get(reverse('bom-export', args=(1,)), HTTP_X_REQUESTED_WITH='XMLHttpRequest')
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertIn('streaming_content', dir(response))
|
|
|
|
|
|
class PartEditTest(PartViewTestCase):
|
|
""" Tests for Part editing form """
|
|
|
|
def test_get_edit(self):
|
|
response = self.client.get(reverse('part-edit', args=(1,)), HTTP_X_REQUESTED_WITH='XMLHttpRequest')
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
keys = response.context.keys()
|
|
data = str(response.content)
|
|
|
|
self.assertIn('part', keys)
|
|
self.assertIn('csrf_token', keys)
|
|
|
|
self.assertIn('html_form', data)
|
|
self.assertIn('"title":', data)
|
|
|
|
|
|
class PartAttachmentTests(PartViewTestCase):
|
|
|
|
def test_valid_create(self):
|
|
""" test creation of an attachment for a valid part """
|
|
|
|
response = self.client.get(reverse('part-attachment-create'), {'part': 1}, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
def test_invalid_create(self):
|
|
""" test creation of an attachment for an invalid part """
|
|
|
|
with self.assertRaises(Part.DoesNotExist):
|
|
response = self.client.get(reverse('part-attachment-create'), {'part': 999}, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
|
|
|
|
def test_edit(self):
|
|
""" test editing an attachment """
|
|
|
|
# TODO
|
|
pass
|
|
|
|
|
|
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)
|