2
0
mirror of https://github.com/inventree/InvenTree.git synced 2026-03-15 16:40:56 +00:00

Fix unit test for data migration

This commit is contained in:
Oliver Walters
2025-12-01 04:40:07 +00:00
parent 0c17f44711
commit 6d5458ec77

View File

@@ -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: