mirror of
https://github.com/inventree/InvenTree.git
synced 2025-05-07 07:48:50 +00:00
89ac0a623b
* Add migration test Looking at older data migration which removes stock items which are "scheduled_for_deletion" * Account for parent / child relationships
132 lines
4.1 KiB
Python
132 lines
4.1 KiB
Python
"""Unit tests for data migrations in the 'stock' app"""
|
|
|
|
from django_test_migrations.contrib.unittest_case import MigratorTestCase
|
|
|
|
from InvenTree import helpers
|
|
|
|
|
|
class TestSerialNumberMigration(MigratorTestCase):
|
|
"""Test data migration which updates serial numbers"""
|
|
|
|
migrate_from = ('stock', '0067_alter_stockitem_part')
|
|
migrate_to = ('stock', helpers.getNewestMigrationFile('stock'))
|
|
|
|
def prepare(self):
|
|
"""Create initial data for this migration"""
|
|
|
|
Part = self.old_state.apps.get_model('part', 'part')
|
|
StockItem = self.old_state.apps.get_model('stock', 'stockitem')
|
|
|
|
# Create a base part
|
|
my_part = Part.objects.create(
|
|
name='PART-123',
|
|
description='Some part',
|
|
active=True,
|
|
trackable=True,
|
|
level=0,
|
|
tree_id=0,
|
|
lft=0, rght=0
|
|
)
|
|
|
|
# Create some serialized stock items
|
|
for sn in range(10, 20):
|
|
StockItem.objects.create(
|
|
part=my_part,
|
|
quantity=1,
|
|
serial=sn,
|
|
level=0,
|
|
tree_id=0,
|
|
lft=0, rght=0
|
|
)
|
|
|
|
# Create a stock item with a very large serial number
|
|
item = StockItem.objects.create(
|
|
part=my_part,
|
|
quantity=1,
|
|
serial='9999999999999999999999999999999999999999999999999999999999999',
|
|
level=0,
|
|
tree_id=0,
|
|
lft=0, rght=0
|
|
)
|
|
|
|
self.big_ref_pk = item.pk
|
|
|
|
def test_migrations(self):
|
|
"""Test that the migrations have been applied correctly"""
|
|
|
|
StockItem = self.new_state.apps.get_model('stock', 'stockitem')
|
|
|
|
# Check that the serial number integer conversion has been applied correctly
|
|
for sn in range(10, 20):
|
|
item = StockItem.objects.get(serial_int=sn)
|
|
|
|
self.assertEqual(item.serial, str(sn))
|
|
|
|
big_ref_item = StockItem.objects.get(pk=self.big_ref_pk)
|
|
|
|
# Check that the StockItem maximum serial number
|
|
self.assertEqual(big_ref_item.serial, '9999999999999999999999999999999999999999999999999999999999999')
|
|
self.assertEqual(big_ref_item.serial_int, 0x7fffffff)
|
|
|
|
|
|
class TestScheduledForDeletionMigration(MigratorTestCase):
|
|
"""Test data migration for removing 'scheduled_for_deletion' field"""
|
|
|
|
migrate_from = ('stock', '0066_stockitem_scheduled_for_deletion')
|
|
migrate_to = ('stock', helpers.getNewestMigrationFile('stock'))
|
|
|
|
def prepare(self):
|
|
"""Create some initial stock items"""
|
|
|
|
Part = self.old_state.apps.get_model('part', 'part')
|
|
StockItem = self.old_state.apps.get_model('stock', 'stockitem')
|
|
|
|
for idx in range(5):
|
|
part = Part.objects.create(
|
|
name=f'Part_{idx}',
|
|
description='Just a part, nothing to see here',
|
|
active=True,
|
|
level=0, tree_id=0,
|
|
lft=0, rght=0,
|
|
)
|
|
|
|
for jj in range(5):
|
|
StockItem.objects.create(
|
|
part=part,
|
|
quantity=jj + 5,
|
|
level=0, tree_id=0,
|
|
lft=0, rght=0,
|
|
scheduled_for_deletion=True
|
|
)
|
|
|
|
# For extra points, create some parent-child relationships between stock items
|
|
part = Part.objects.first()
|
|
|
|
item_1 = StockItem.objects.create(
|
|
part=part,
|
|
quantity=100,
|
|
level=0, tree_id=0,
|
|
lft=0, rght=0,
|
|
scheduled_for_deletion=True,
|
|
)
|
|
|
|
for ii in range(3):
|
|
StockItem.objects.create(
|
|
part=part,
|
|
quantity=200,
|
|
level=0, tree_id=0,
|
|
lft=0, rght=0,
|
|
scheduled_for_deletion=False,
|
|
parent=item_1,
|
|
)
|
|
|
|
self.assertEqual(StockItem.objects.count(), 29)
|
|
|
|
def test_migration(self):
|
|
"""Test that all stock items were actually removed"""
|
|
|
|
StockItem = self.new_state.apps.get_model('stock', 'stockitem')
|
|
|
|
# All the "scheduled for deletion" items have been removed
|
|
self.assertEqual(StockItem.objects.count(), 3)
|