1
0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00

Fixes #3878: Fix database migration for cable status field

This commit is contained in:
Jeremy Stretch
2020-01-09 16:16:24 -05:00
parent 0527626709
commit dc5f5efcfe
3 changed files with 7 additions and 4 deletions

View File

@@ -24,8 +24,8 @@ CABLE_TYPE_CHOICES = (
)
CABLE_STATUS_CHOICES = (
(True, 'connected'),
(False, 'planned'),
('true', 'connected'),
('false', 'planned'),
)
CABLE_LENGTH_UNIT_CHOICES = (
@@ -44,8 +44,8 @@ def cable_type_to_slug(apps, schema_editor):
def cable_status_to_slug(apps, schema_editor):
Cable = apps.get_model('dcim', 'Cable')
for bool, slug in CABLE_STATUS_CHOICES:
Cable.objects.filter(status=str(bool)).update(status=slug)
for bool_str, slug in CABLE_STATUS_CHOICES:
Cable.objects.filter(status=bool_str).update(status=slug)
def cable_length_unit_to_slug(apps, schema_editor):

View File

@@ -2188,6 +2188,8 @@ class CableTestCase(TestCase):
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
def test_status(self):
params = {'status': [CableStatusChoices.STATUS_CONNECTED]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 3)
params = {'status': [CableStatusChoices.STATUS_PLANNED]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 3)