Adjust unit test

This commit is contained in:
Oliver Walters
2025-12-01 03:35:07 +00:00
parent b32ba71318
commit b901224e21
2 changed files with 7 additions and 3 deletions

View File

@@ -605,7 +605,11 @@ class InvenTreeParameterMixin(InvenTreePermissionCheckMixin, models.Model):
def get_parameters(self) -> QuerySet:
"""Return all Parameter instances for this model."""
return self.parameters_list.all().order_by('template__name')
return (
self.parameters_list.all()
.prefetch_related('template', 'model_type')
.order_by('template__name')
)
def parameters_map(self) -> dict:
"""Return a map (dict) of parameter values associated with this Part instance, of the form.

View File

@@ -225,8 +225,8 @@ class ParameterAPITests(InvenTreeAPITestCase):
common.models.ParameterTemplate.objects.bulk_create(templates)
# Create a parameter for every company against every template
for company in companies:
for template in templates:
for company in Company.objects.all():
for template in common.models.ParameterTemplate.objects.all():
parameters.append(
common.models.Parameter(
template=template,