mirror of
https://github.com/inventree/InvenTree.git
synced 2025-04-29 03:56:43 +00:00
60e44700a9
* Fix for barcode URL generation * Refactor the fix entirely - move the URL generation out of MakeBarcode function - Improved unit testing - actually render a label template * Updates for CI * CI fix
132 lines
4.0 KiB
Python
132 lines
4.0 KiB
Python
"""Tests for labels"""
|
|
|
|
import io
|
|
|
|
from django.apps import apps
|
|
from django.conf import settings
|
|
from django.core.exceptions import ValidationError
|
|
from django.core.files.base import ContentFile
|
|
from django.urls import reverse
|
|
|
|
from common.models import InvenTreeSetting
|
|
from InvenTree.api_tester import InvenTreeAPITestCase
|
|
from InvenTree.helpers import validateFilterString
|
|
from part.models import Part
|
|
from stock.models import StockItem
|
|
|
|
from .models import PartLabel, StockItemLabel, StockLocationLabel
|
|
|
|
|
|
class LabelTest(InvenTreeAPITestCase):
|
|
"""Unit test class for label models"""
|
|
|
|
fixtures = [
|
|
'category',
|
|
'part',
|
|
'location',
|
|
'stock'
|
|
]
|
|
|
|
def setUp(self) -> None:
|
|
"""Ensure that some label instances exist as part of init routine"""
|
|
super().setUp()
|
|
apps.get_app_config('label').create_labels()
|
|
|
|
def test_default_labels(self):
|
|
"""Test that the default label templates are copied across."""
|
|
labels = StockItemLabel.objects.all()
|
|
|
|
self.assertTrue(labels.count() > 0)
|
|
|
|
labels = StockLocationLabel.objects.all()
|
|
|
|
self.assertTrue(labels.count() > 0)
|
|
|
|
def test_default_files(self):
|
|
"""Test that label files exist in the MEDIA directory."""
|
|
def test_subdir(ref_name):
|
|
item_dir = settings.MEDIA_ROOT.joinpath(
|
|
'label',
|
|
'inventree',
|
|
ref_name,
|
|
)
|
|
self.assertTrue(len([item_dir.iterdir()]) > 0)
|
|
|
|
test_subdir('stockitem')
|
|
test_subdir('stocklocation')
|
|
test_subdir('part')
|
|
|
|
def test_filters(self):
|
|
"""Test the label filters."""
|
|
filter_string = "part__pk=10"
|
|
|
|
filters = validateFilterString(filter_string, model=StockItem)
|
|
|
|
self.assertEqual(type(filters), dict)
|
|
|
|
bad_filter_string = "part_pk=10"
|
|
|
|
with self.assertRaises(ValidationError):
|
|
validateFilterString(bad_filter_string, model=StockItem)
|
|
|
|
def test_label_rendering(self):
|
|
"""Test label rendering."""
|
|
labels = PartLabel.objects.all()
|
|
part = Part.objects.first()
|
|
|
|
for label in labels:
|
|
url = reverse('api-part-label-print', kwargs={'pk': label.pk})
|
|
self.get(f'{url}?parts={part.pk}', expected_code=200)
|
|
|
|
def test_print_part_label(self):
|
|
"""Actually 'print' a label, and ensure that the correct information is contained."""
|
|
|
|
label_data = """
|
|
{% load barcode %}
|
|
{% load report %}
|
|
|
|
<html>
|
|
<!-- Test that the part instance is supplied -->
|
|
part: {{ part.pk }} - {{ part.name }}
|
|
<!-- Test qr data -->
|
|
data: {{ qr_data|safe }}
|
|
<!-- Test InvenTree URL -->
|
|
url: {{ qr_url|safe }}
|
|
<!-- Test image URL generation -->
|
|
image: {% part_image part %}
|
|
<!-- Test InvenTree logo -->
|
|
logo: {% logo_image %}
|
|
</html>
|
|
"""
|
|
|
|
buffer = io.StringIO()
|
|
buffer.write(label_data)
|
|
|
|
template = ContentFile(buffer.getvalue(), "label.html")
|
|
|
|
# Construct a label template
|
|
label = PartLabel.objects.create(
|
|
name='test',
|
|
description='Test label',
|
|
enabled=True,
|
|
label=template,
|
|
)
|
|
|
|
# Ensure we are in "debug" mode (so the report is generated as HTML)
|
|
InvenTreeSetting.set_setting('REPORT_ENABLE', True, None)
|
|
InvenTreeSetting.set_setting('REPORT_DEBUG_MODE', True, None)
|
|
|
|
# Print via the API
|
|
url = reverse('api-part-label-print', kwargs={'pk': label.pk})
|
|
|
|
response = self.get(f'{url}?parts=1', expected_code=200)
|
|
|
|
content = str(response.content)
|
|
|
|
# Test that each element has been rendered correctly
|
|
self.assertIn("part: 1 - M2x4 LPHS", content)
|
|
self.assertIn('data: {"part": 1}', content)
|
|
self.assertIn("http://testserver/part/1/", content)
|
|
self.assertIn("image: /static/img/blank_image.png", content)
|
|
self.assertIn("logo: /static/img/inventree.png", content)
|