mirror of
https://github.com/inventree/InvenTree.git
synced 2025-12-16 01:08:12 +00:00
fix test
This commit is contained in:
@@ -32,7 +32,7 @@ from InvenTree.unit_test import (
|
||||
PluginMixin,
|
||||
addUserPermission,
|
||||
)
|
||||
from part.models import Part, PartParameterTemplate
|
||||
from part.models import Part
|
||||
from plugin import registry
|
||||
|
||||
from .api import WebhookView
|
||||
@@ -45,6 +45,7 @@ from .models import (
|
||||
NotesImage,
|
||||
NotificationEntry,
|
||||
NotificationMessage,
|
||||
ParameterTemplate,
|
||||
ProjectCode,
|
||||
SelectionList,
|
||||
SelectionListEntry,
|
||||
@@ -2055,27 +2056,37 @@ class SelectionListTest(InvenTreeAPITestCase):
|
||||
|
||||
# Add to parameter
|
||||
part = Part.objects.get(pk=1)
|
||||
template = PartParameterTemplate.objects.create(
|
||||
template = ParameterTemplate.objects.create(
|
||||
name='test_parameter', units='', selectionlist=self.list
|
||||
)
|
||||
rsp = self.get(
|
||||
reverse('api-part-parameter-template-detail', kwargs={'pk': template.pk})
|
||||
reverse('api-parameter-template-detail', kwargs={'pk': template.pk})
|
||||
)
|
||||
self.assertEqual(rsp.data['name'], 'test_parameter')
|
||||
self.assertEqual(rsp.data['choices'], '')
|
||||
|
||||
# Add to part
|
||||
url = reverse('api-part-parameter-list')
|
||||
url = reverse('api-parameter-list')
|
||||
response = self.post(
|
||||
url,
|
||||
{'part': part.pk, 'template': template.pk, 'data': 70},
|
||||
{
|
||||
'model_id': part.pk,
|
||||
'model_type': 'part.part',
|
||||
'template': template.pk,
|
||||
'data': 70,
|
||||
},
|
||||
expected_code=400,
|
||||
)
|
||||
self.assertIn('Invalid choice for parameter value', response.data['data'])
|
||||
|
||||
response = self.post(
|
||||
url,
|
||||
{'part': part.pk, 'template': template.pk, 'data': self.entry1.value},
|
||||
{
|
||||
'model_id': part.pk,
|
||||
'model_type': 'part.part',
|
||||
'template': template.pk,
|
||||
'data': self.entry1.value,
|
||||
},
|
||||
expected_code=201,
|
||||
)
|
||||
self.assertEqual(response.data['data'], self.entry1.value)
|
||||
|
||||
Reference in New Issue
Block a user