mirror of
https://github.com/inventree/InvenTree.git
synced 2025-06-16 12:05:53 +00:00
115 lines
3.3 KiB
Python
115 lines
3.3 KiB
Python
"""Unit tests specific to the SupplierPart model"""
|
|
|
|
from decimal import Decimal
|
|
|
|
from django.core.exceptions import ValidationError
|
|
|
|
from company.models import Company, SupplierPart
|
|
from InvenTree.unit_test import InvenTreeTestCase
|
|
from part.models import Part
|
|
|
|
|
|
class SupplierPartPackUnitsTests(InvenTreeTestCase):
|
|
"""Unit tests for the SupplierPart pack_units field"""
|
|
|
|
def test_pack_units_dimensionless(self):
|
|
"""Test valid values for the 'pack_units' field"""
|
|
|
|
# Create a part without units (dimensionless)
|
|
part = Part.objects.create(name='Test Part', description='Test part description', component=True)
|
|
|
|
# Create a supplier (company)
|
|
company = Company.objects.create(name='Test Company', is_supplier=True)
|
|
|
|
# Create a supplier part for this part
|
|
sp = SupplierPart.objects.create(
|
|
part=part,
|
|
supplier=company,
|
|
SKU='TEST-SKU'
|
|
)
|
|
|
|
# All these values are valid for a dimensionless part
|
|
pass_tests = {
|
|
'': 1,
|
|
'1': 1,
|
|
'1.01': 1.01,
|
|
'12.000001': 12.000001,
|
|
'99.99': 99.99,
|
|
}
|
|
|
|
# All these values are invalid for a dimensionless part
|
|
fail_tests = [
|
|
'1.2m',
|
|
'-1',
|
|
'0',
|
|
'0.0',
|
|
'100 feet',
|
|
'0 amps'
|
|
]
|
|
|
|
for test, expected in pass_tests.items():
|
|
sp.pack_units = test
|
|
sp.full_clean()
|
|
self.assertEqual(sp.pack_units_native, expected)
|
|
|
|
for test in fail_tests:
|
|
sp.pack_units = test
|
|
with self.assertRaises(ValidationError):
|
|
sp.full_clean()
|
|
|
|
def test_pack_units(self):
|
|
"""Test pack_units for a part with a specified dimension"""
|
|
|
|
# Create a part with units 'm'
|
|
part = Part.objects.create(name='Test Part', description='Test part description', component=True, units='m')
|
|
|
|
# Create a supplier (company)
|
|
company = Company.objects.create(name='Test Company', is_supplier=True)
|
|
|
|
# Create a supplier part for this part
|
|
sp = SupplierPart.objects.create(
|
|
part=part,
|
|
supplier=company,
|
|
SKU='TEST-SKU'
|
|
)
|
|
|
|
# All these values are valid for a part with dimesion 'm'
|
|
pass_tests = {
|
|
'': 1,
|
|
'1': 1,
|
|
'1m': 1,
|
|
'1.01m': 1.01,
|
|
'1.01': 1.01,
|
|
'5 inches': 0.127,
|
|
'27 cm': 0.27,
|
|
'3km': 3000,
|
|
'14 feet': 4.2672,
|
|
'0.5 miles': 804.672,
|
|
}
|
|
|
|
# All these values are invalid for a part with dimension 'm'
|
|
# Either the values are invalid, or the units are incomaptible
|
|
fail_tests = [
|
|
'-1',
|
|
'-1m',
|
|
'0',
|
|
'0m',
|
|
'12 deg',
|
|
'57 amps',
|
|
'-12 oz',
|
|
'17 yaks',
|
|
]
|
|
|
|
for test, expected in pass_tests.items():
|
|
sp.pack_units = test
|
|
sp.full_clean()
|
|
self.assertEqual(
|
|
round(Decimal(sp.pack_units_native), 10),
|
|
round(Decimal(str(expected)), 10)
|
|
)
|
|
|
|
for test in fail_tests:
|
|
sp.pack_units = test
|
|
with self.assertRaises(ValidationError):
|
|
sp.full_clean()
|