18245 fix tests

This commit is contained in:
Arthur 2025-03-21 10:17:42 -07:00
parent a5dc0807db
commit a028aad849
2 changed files with 8 additions and 1 deletions

View File

@ -1149,7 +1149,7 @@ class InventoryItemTemplateTest(APIViewTestCases.APIViewTestCase):
class DeviceRoleGroupTest(APIViewTestCases.APIViewTestCase):
model = DeviceRoleGroup
brief_fields = ['_depth', 'description', 'display', 'id', 'name', 'slug', 'role_count', 'url']
brief_fields = ['_depth', 'description', 'display', 'id', 'name', 'role_count', 'slug', 'url']
bulk_update_data = {
'description': 'New description',
'comments': 'New Comment',

View File

@ -2298,6 +2298,13 @@ class DeviceRoleTestCase(TestCase, ChangeLoggedFilterSetTests):
params = {'vm_role': 'false'}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
def test_group(self):
group = DeviceRoleGroup.objects.all()[:2]
params = {'group_id': [group[0].pk, group[1].pk]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
params = {'group': [group[0].slug, group[1].slug]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_description(self):
params = {'description': ['foobar1', 'foobar2']}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)