From 6d5458ec77def7e2b7eb05330907c7a5e40c5c14 Mon Sep 17 00:00:00 2001 From: Oliver Walters Date: Mon, 1 Dec 2025 04:40:07 +0000 Subject: [PATCH] Fix unit test for data migration --- src/backend/InvenTree/part/test_migrations.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/backend/InvenTree/part/test_migrations.py b/src/backend/InvenTree/part/test_migrations.py index 85a5477100..4f4c726788 100644 --- a/src/backend/InvenTree/part/test_migrations.py +++ b/src/backend/InvenTree/part/test_migrations.py @@ -340,13 +340,19 @@ class TestPartParameterDeletion(MigratorTestCase): ParameterTemplate = self.new_state.apps.get_model('common', 'parametertemplate') Parameter = self.new_state.apps.get_model('common', 'parameter') Part = self.new_state.apps.get_model('part', 'part') + ContentType = self.new_state.apps.get_model('contenttypes', 'contenttype') self.assertEqual(ParameterTemplate.objects.count(), 3) self.assertEqual(Parameter.objects.count(), 9) self.assertEqual(Part.objects.count(), 3) + content_type, _created = ContentType.objects.get_or_create( + app_label='part', model='part' + ) + for p in Part.objects.all(): - params = p.parameters_list.all() + params = Parameter.objects.filter(model_type=content_type, model_id=p.id) + self.assertEqual(len(params), 3) for unit in self.UNITS: