NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-16 04:28:17 -06:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

5509 update tests to use CustomFieldTypeChoices

Browse Source
This commit is contained in:
Arthur 2023-09-13 15:43:40 -07:00
parent a146051c44
commit 7dcddc3012
1 changed files with 28 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
netbox/utilities/testing/views.py
View File

@ -8,7 +8,7 @@ from django.test import override_settings
from django.urls import reverse from django.urls import reverse
from extras.choices import ObjectChangeActionChoices, CustomFieldTypeChoices from extras.choices import ObjectChangeActionChoices, CustomFieldTypeChoices
from extras.models import ObjectChange, CustomField from extras.models import CustomField, CustomFieldChoiceSet, ObjectChange
from netbox.models import CustomFieldsMixin from netbox.models import CustomFieldsMixin
from netbox.models.features import ChangeLoggingMixin from netbox.models.features import ChangeLoggingMixin
from users.models import ObjectPermission from users.models import ObjectPermission
@ -35,27 +35,36 @@ def add_custom_field_data(form_data, model):
CustomField(type=CustomFieldTypeChoices.TYPE_DATE, name='date_field', default='2020-01-01'), CustomField(type=CustomFieldTypeChoices.TYPE_DATE, name='date_field', default='2020-01-01'),
CustomField(type=CustomFieldTypeChoices.TYPE_URL, name='url_field', default='http://example.com/1'), CustomField(type=CustomFieldTypeChoices.TYPE_URL, name='url_field', default='http://example.com/1'),
CustomField(type=CustomFieldTypeChoices.TYPE_JSON, name='json_field', default='{"x": "y"}'), CustomField(type=CustomFieldTypeChoices.TYPE_JSON, name='json_field', default='{"x": "y"}'),
CustomField(
type=CustomFieldTypeChoices.TYPE_SELECT,
name='select_field',
default='Foo',
choices=(
'Foo', 'Bar', 'Baz'
)
),
CustomField(
type=CustomFieldTypeChoices.TYPE_MULTISELECT,
name='multiselect_field',
default=['Foo'],
choices=(
'Foo', 'Bar', 'Baz'
)
),
) )
for cf in custom_fields: for cf in custom_fields:
cf.save() cf.save()
cf.content_types.set([ContentType.objects.get_for_model(model)]) cf.content_types.set([ContentType.objects.get_for_model(model)])
choice_set = CustomFieldChoiceSet.objects.create(
name='Choice Set 1',
extra_choices=(('foo', 'Foo'), ('bar', 'Bar'), ('baz', 'Baz'))
)
# Create a custom field on the Site model
ct = ContentType.objects.get_for_model(model)
cf_select = CustomField(
type=CustomFieldTypeChoices.TYPE_SELECT,
name='select_field',
choice_set=choice_set
)
cf_select.save()
cf_select.content_types.set([ct])
cf_select = CustomField(
type=CustomFieldTypeChoices.TYPE_MULTISELECT,
name='multiselect_field',
default=['foo'],
choice_set=choice_set
)
cf_select.save()
cf_select.content_types.set([ct])
form_data['cf_text_field'] = 'foo123' form_data['cf_text_field'] = 'foo123'
form_data['cf_longtext_field'] = 'ABC123' form_data['cf_longtext_field'] = 'ABC123'
form_data['cf_integer_field'] = 456 form_data['cf_integer_field'] = 456
@ -64,8 +73,8 @@ def add_custom_field_data(form_data, model):
form_data['cf_date_field'] = '2022-02-02' form_data['cf_date_field'] = '2022-02-02'
form_data['cf_url_field'] = 'http://example2.com/1' form_data['cf_url_field'] = 'http://example2.com/1'
form_data['cf_json_field'] = '{"x": "z"}' form_data['cf_json_field'] = '{"x": "z"}'
form_data['cf_select_field'] = 'Bar' form_data['cf_select_field'] = 'bar'
form_data['cf_multiselect_field'] = ['Bar'] form_data['cf_multiselect_field'] = ['bar']
return form_data return form_data