mirror of
https://github.com/inventree/InvenTree
synced 2024-08-30 18:33:04 +00:00
152 lines
4.5 KiB
Python
152 lines
4.5 KiB
Python
from django.test import TestCase
|
|
import django.core.exceptions as django_exceptions
|
|
|
|
from part.models import Part
|
|
from .models import PurchaseOrder, PurchaseOrderLineItem
|
|
from stock.models import StockLocation
|
|
from company.models import SupplierPart
|
|
|
|
from InvenTree.status_codes import OrderStatus
|
|
|
|
|
|
class OrderTest(TestCase):
|
|
"""
|
|
Tests to ensure that the order models are functioning correctly.
|
|
"""
|
|
|
|
fixtures = [
|
|
'company',
|
|
'supplier_part',
|
|
'category',
|
|
'part',
|
|
'location',
|
|
'stock',
|
|
'order'
|
|
]
|
|
|
|
def test_basics(self):
|
|
""" Basic tests e.g. repr functions etc """
|
|
|
|
order = PurchaseOrder.objects.get(pk=1)
|
|
|
|
self.assertEqual(order.get_absolute_url(), '/order/purchase-order/1/')
|
|
|
|
self.assertEqual(str(order), 'PO 1')
|
|
|
|
line = PurchaseOrderLineItem.objects.get(pk=1)
|
|
|
|
self.assertEqual(str(line), "100 x ACME0001 from ACME (for PO 1)")
|
|
|
|
def test_on_order(self):
|
|
""" There should be 3 separate items on order for the M2x4 LPHS part """
|
|
|
|
part = Part.objects.get(name='M2x4 LPHS')
|
|
|
|
open_orders = []
|
|
|
|
for supplier in part.supplier_parts.all():
|
|
open_orders += supplier.open_orders()
|
|
|
|
self.assertEqual(len(open_orders), 4)
|
|
|
|
# Test the total on-order quantity
|
|
self.assertEqual(part.on_order, 1400)
|
|
|
|
def test_add_items(self):
|
|
""" Test functions for adding line items to an order """
|
|
|
|
order = PurchaseOrder.objects.get(pk=1)
|
|
|
|
self.assertEqual(order.status, OrderStatus.PENDING)
|
|
self.assertEqual(order.lines.count(), 3)
|
|
|
|
sku = SupplierPart.objects.get(SKU='ACME-WIDGET')
|
|
part = sku.part
|
|
|
|
# Try to order some invalid things
|
|
with self.assertRaises(django_exceptions.ValidationError):
|
|
order.add_line_item(sku, -999)
|
|
|
|
with self.assertRaises(django_exceptions.ValidationError):
|
|
order.add_line_item(sku, 'not a number')
|
|
|
|
# Order the part
|
|
self.assertEqual(part.on_order, 0)
|
|
|
|
order.add_line_item(sku, 100)
|
|
|
|
self.assertEqual(part.on_order, 100)
|
|
self.assertEqual(order.lines.count(), 4)
|
|
|
|
# Order the same part again (it should be merged)
|
|
order.add_line_item(sku, 50)
|
|
self.assertEqual(order.lines.count(), 4)
|
|
self.assertEqual(part.on_order, 150)
|
|
|
|
# Try to order a supplier part from the wrong supplier
|
|
sku = SupplierPart.objects.get(SKU='ZERG-WIDGET')
|
|
|
|
with self.assertRaises(django_exceptions.ValidationError):
|
|
order.add_line_item(sku, 99)
|
|
|
|
def test_receive(self):
|
|
""" Test order receiving functions """
|
|
|
|
part = Part.objects.get(name='M2x4 LPHS')
|
|
|
|
# Receive some items
|
|
line = PurchaseOrderLineItem.objects.get(id=1)
|
|
|
|
order = line.order
|
|
loc = StockLocation.objects.get(id=1)
|
|
|
|
# There should be two lines against this order
|
|
self.assertEqual(len(order.pending_line_items()), 3)
|
|
|
|
# Should fail, as order is 'PENDING' not 'PLACED"
|
|
self.assertEqual(order.status, OrderStatus.PENDING)
|
|
|
|
with self.assertRaises(django_exceptions.ValidationError):
|
|
order.receive_line_item(line, loc, 50, user=None)
|
|
|
|
order.place_order()
|
|
|
|
self.assertEqual(order.status, OrderStatus.PLACED)
|
|
|
|
order.receive_line_item(line, loc, 50, user=None)
|
|
|
|
self.assertEqual(line.remaining(), 50)
|
|
|
|
self.assertEqual(part.on_order, 1350)
|
|
|
|
# Try to order some invalid things
|
|
with self.assertRaises(django_exceptions.ValidationError):
|
|
order.receive_line_item(line, loc, -10, user=None)
|
|
|
|
with self.assertRaises(django_exceptions.ValidationError):
|
|
order.receive_line_item(line, loc, 'not a number', user=None)
|
|
|
|
# Receive the rest of the items
|
|
order.receive_line_item(line, loc, 50, user=None)
|
|
|
|
line = PurchaseOrderLineItem.objects.get(id=2)
|
|
order.receive_line_item(line, loc, 2 * line.quantity, user=None)
|
|
|
|
self.assertEqual(part.on_order, 1100)
|
|
self.assertEqual(order.status, OrderStatus.PLACED)
|
|
|
|
for line in order.pending_line_items():
|
|
order.receive_line_item(line, loc, line.quantity, user=None)
|
|
|
|
self.assertEqual(order.status, OrderStatus.COMPLETE)
|
|
|
|
def test_export(self):
|
|
""" Test order exporting """
|
|
|
|
order = PurchaseOrder.objects.get(pk=1)
|
|
|
|
output = order.export_to_file(format='csv')
|
|
|
|
self.assertIn('M2x4 LPHS', output)
|
|
self.assertIn('Line,Part,Description', output)
|