mirror of
https://github.com/inventree/InvenTree.git
synced 2026-09-15 08:52:50 +00:00
miggland
807784d9e4
* Add metadata for ManufacturerPart * Add Metadata for SupplierPart * Add metadata to label models * Add metadata to order line items * Add metadata to shipment * Add metadata to Build and BuildItem * Add metadata to BomItem * Add metadata to PartParameterTemplate * Syntax, lint * Fix isort style * Lint * Correction of model name * Metadata for Reports * Fix silly error * Fix silly error * Correct model name * Correct model name * Correction * Correct company urls * Apply generic model to Report metadat * Rename/remove redundant import * Remove shadowing of report in loop variable * Update import ordering * More corrections * better docstrings * Correct names for API endpoints * Default to PO, required for api-doc to work * Changes by @matmair * Suppress metadata from Bom export * Add migration files * Increment API version * Add tests for all Metadata models, even previously existing ones * Update tests * Fix * Delay tests * Fix imports * Fix tests * API Version number * Remove unused import * isort * Revent unintended change of cache
243 lines
8.0 KiB
Python
243 lines
8.0 KiB
Python
"""Unit tests for the models in the 'company' app"""
|
|
|
|
import os
|
|
from decimal import Decimal
|
|
|
|
from django.core.exceptions import ValidationError
|
|
from django.test import TestCase
|
|
|
|
from part.models import Part
|
|
|
|
from .models import (Company, Contact, ManufacturerPart, SupplierPart,
|
|
rename_company_image)
|
|
|
|
|
|
class CompanySimpleTest(TestCase):
|
|
"""Unit tests for the Company model"""
|
|
|
|
fixtures = [
|
|
'company',
|
|
'category',
|
|
'part',
|
|
'location',
|
|
'bom',
|
|
'manufacturer_part',
|
|
'supplier_part',
|
|
'price_breaks',
|
|
]
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
"""Perform initialization for the tests in this class"""
|
|
|
|
super().setUpTestData()
|
|
|
|
Company.objects.create(name='ABC Co.',
|
|
description='Seller of ABC products',
|
|
website='www.abc-sales.com',
|
|
address='123 Sales St.',
|
|
is_customer=False,
|
|
is_supplier=True)
|
|
|
|
cls.acme0001 = SupplierPart.objects.get(SKU='ACME0001')
|
|
cls.acme0002 = SupplierPart.objects.get(SKU='ACME0002')
|
|
cls.zerglphs = SupplierPart.objects.get(SKU='ZERGLPHS')
|
|
cls.zergm312 = SupplierPart.objects.get(SKU='ZERGM312')
|
|
|
|
def test_company_model(self):
|
|
"""Tests for the company model data"""
|
|
c = Company.objects.get(name='ABC Co.')
|
|
self.assertEqual(c.name, 'ABC Co.')
|
|
self.assertEqual(str(c), 'ABC Co. - Seller of ABC products')
|
|
|
|
def test_company_url(self):
|
|
"""Test the detail URL for a company"""
|
|
c = Company.objects.get(pk=1)
|
|
self.assertEqual(c.get_absolute_url(), '/company/1/')
|
|
|
|
def test_image_renamer(self):
|
|
"""Test the company image upload functionality"""
|
|
c = Company.objects.get(pk=1)
|
|
rn = rename_company_image(c, 'test.png')
|
|
self.assertEqual(rn, 'company_images' + os.path.sep + 'company_1_img.png')
|
|
|
|
rn = rename_company_image(c, 'test2')
|
|
self.assertEqual(rn, 'company_images' + os.path.sep + 'company_1_img')
|
|
|
|
def test_price_breaks(self):
|
|
"""Unit tests for price breaks"""
|
|
self.assertTrue(self.acme0001.has_price_breaks)
|
|
self.assertTrue(self.acme0002.has_price_breaks)
|
|
self.assertTrue(self.zergm312.has_price_breaks)
|
|
self.assertFalse(self.zerglphs.has_price_breaks)
|
|
|
|
self.assertEqual(self.acme0001.price_breaks.count(), 3)
|
|
self.assertEqual(self.acme0002.price_breaks.count(), 2)
|
|
self.assertEqual(self.zerglphs.price_breaks.count(), 0)
|
|
self.assertEqual(self.zergm312.price_breaks.count(), 2)
|
|
|
|
def test_quantity_pricing(self):
|
|
"""Simple test for quantity pricing."""
|
|
p = self.acme0001.get_price
|
|
self.assertEqual(p(1), 10)
|
|
self.assertEqual(p(4), 40)
|
|
self.assertEqual(p(11), 82.5)
|
|
self.assertEqual(p(23), 172.5)
|
|
self.assertEqual(p(100), 350)
|
|
|
|
p = self.acme0002.get_price
|
|
self.assertEqual(p(0.5), 3.5)
|
|
self.assertEqual(p(1), 7)
|
|
self.assertEqual(p(2), 14)
|
|
self.assertEqual(p(5), 35)
|
|
self.assertEqual(p(45), 315)
|
|
self.assertEqual(p(55), 68.75)
|
|
|
|
def test_part_pricing(self):
|
|
"""Unit tests for supplier part pricing"""
|
|
m2x4 = Part.objects.get(name='M2x4 LPHS')
|
|
|
|
self.assertEqual(m2x4.get_price_info(5.5), "38.5 - 41.25")
|
|
self.assertEqual(m2x4.get_price_info(10), "70 - 75")
|
|
self.assertEqual(m2x4.get_price_info(100), "125 - 350")
|
|
|
|
pmin, pmax = m2x4.get_price_range(5)
|
|
self.assertEqual(pmin, 35)
|
|
self.assertEqual(pmax, 37.5)
|
|
|
|
m3x12 = Part.objects.get(name='M3x12 SHCS')
|
|
|
|
self.assertEqual(m3x12.get_price_info(0.3), Decimal('2.4'))
|
|
self.assertEqual(m3x12.get_price_info(3), Decimal('24'))
|
|
self.assertIsNotNone(m3x12.get_price_info(50))
|
|
|
|
def test_currency_validation(self):
|
|
"""Test validation for currency selection."""
|
|
# Create a company with a valid currency code (should pass)
|
|
company = Company.objects.create(
|
|
name='Test',
|
|
description='Toast',
|
|
currency='AUD',
|
|
)
|
|
|
|
company.full_clean()
|
|
|
|
# Create a company with an invalid currency code (should fail)
|
|
company = Company.objects.create(
|
|
name='test',
|
|
description='Toasty',
|
|
currency='XZY',
|
|
)
|
|
|
|
with self.assertRaises(ValidationError):
|
|
company.full_clean()
|
|
|
|
def test_metadata(self):
|
|
"""Unit tests for the metadata field."""
|
|
p = Company.objects.first()
|
|
self.assertIsNone(p.metadata)
|
|
|
|
self.assertIsNone(p.get_metadata('test'))
|
|
self.assertEqual(p.get_metadata('test', backup_value=123), 123)
|
|
|
|
# Test update via the set_metadata() method
|
|
p.set_metadata('test', 3)
|
|
self.assertEqual(p.get_metadata('test'), 3)
|
|
|
|
for k in ['apple', 'banana', 'carrot', 'carrot', 'banana']:
|
|
p.set_metadata(k, k)
|
|
|
|
self.assertEqual(len(p.metadata.keys()), 4)
|
|
|
|
|
|
class ContactSimpleTest(TestCase):
|
|
"""Unit tests for the Contact model"""
|
|
|
|
def setUp(self):
|
|
"""Initialization for the tests in this class"""
|
|
# Create a simple company
|
|
self.c = Company.objects.create(name='Test Corp.', description='We make stuff good')
|
|
|
|
# Add some contacts
|
|
Contact.objects.create(name='Joe Smith', company=self.c)
|
|
Contact.objects.create(name='Fred Smith', company=self.c)
|
|
Contact.objects.create(name='Sally Smith', company=self.c)
|
|
|
|
def test_exists(self):
|
|
"""Test that contacts exist"""
|
|
self.assertEqual(Contact.objects.count(), 3)
|
|
|
|
def test_delete(self):
|
|
"""Test deletion of a Contact instance"""
|
|
# Remove the parent company
|
|
Company.objects.get(pk=self.c.pk).delete()
|
|
self.assertEqual(Contact.objects.count(), 0)
|
|
|
|
|
|
class ManufacturerPartSimpleTest(TestCase):
|
|
"""Unit tests for the ManufacturerPart model"""
|
|
|
|
fixtures = [
|
|
'category',
|
|
'company',
|
|
'location',
|
|
'part',
|
|
'manufacturer_part',
|
|
]
|
|
|
|
def setUp(self):
|
|
"""Initialization for the unit tests in this class"""
|
|
|
|
# Create a manufacturer part
|
|
self.part = Part.objects.get(pk=1)
|
|
manufacturer = Company.objects.get(pk=1)
|
|
|
|
self.mp = ManufacturerPart.create(
|
|
part=self.part,
|
|
manufacturer=manufacturer,
|
|
mpn='PART_NUMBER',
|
|
description='THIS IS A MANUFACTURER PART',
|
|
)
|
|
|
|
# Create a supplier part
|
|
supplier = Company.objects.get(pk=5)
|
|
supplier_part = SupplierPart.objects.create(
|
|
part=self.part,
|
|
supplier=supplier,
|
|
SKU='SKU_TEST',
|
|
)
|
|
|
|
supplier_part.save()
|
|
|
|
def test_exists(self):
|
|
"""That that a ManufacturerPart has been created"""
|
|
self.assertEqual(ManufacturerPart.objects.count(), 4)
|
|
|
|
# Check that manufacturer part was created from supplier part creation
|
|
manufacturer_parts = ManufacturerPart.objects.filter(manufacturer=1)
|
|
self.assertEqual(manufacturer_parts.count(), 1)
|
|
|
|
def test_delete(self):
|
|
"""Test deletion of a ManufacturerPart"""
|
|
Part.objects.get(pk=self.part.id).delete()
|
|
# Check that ManufacturerPart was deleted
|
|
self.assertEqual(ManufacturerPart.objects.count(), 3)
|
|
|
|
def test_metadata(self):
|
|
"""Unit tests for the metadata field."""
|
|
for model in [ManufacturerPart, SupplierPart]:
|
|
p = model.objects.first()
|
|
self.assertIsNone(p.metadata)
|
|
|
|
self.assertIsNone(p.get_metadata('test'))
|
|
self.assertEqual(p.get_metadata('test', backup_value=123), 123)
|
|
|
|
# Test update via the set_metadata() method
|
|
p.set_metadata('test', 3)
|
|
self.assertEqual(p.get_metadata('test'), 3)
|
|
|
|
for k in ['apple', 'banana', 'carrot', 'carrot', 'banana']:
|
|
p.set_metadata(k, k)
|
|
|
|
self.assertEqual(len(p.metadata.keys()), 4)
|