mirror of
https://github.com/inventree/InvenTree.git
synced 2026-01-18 04:29:57 -06:00
Adjust unit test
This commit is contained in:
@@ -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.
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user