def test_image_queryset_existing_device_firmware(self):
(
device,
device_fw,
img_org2,
yuncore,
mesh_image,
) = self._prepare_image_qs_test_env()
# existing DeviceFirmware
# restricts images to category of image in used
form = DeviceFirmwareForm(device=device, instance=device_fw)
self.assertEqual(form.fields['image'].queryset.count(), 1)
self.assertIn(device_fw.image, form.fields['image'].queryset)
self.assertNotIn(img_org2, form.fields['image'].queryset)
def test_image_queryset_new_device_firmware(self):
(
device,
device_fw,
img_org2,
yuncore,
mesh_image,
) = self._prepare_image_qs_test_env()
# new DeviceFirmware
# shows all the categories related to the model
form = DeviceFirmwareForm(device=device)
self.assertEqual(form.fields['image'].queryset.count(), 2)
self.assertIn(device_fw.image, form.fields['image'].queryset)
self.assertIn(mesh_image, form.fields['image'].queryset)
self.assertNotIn(img_org2, form.fields['image'].queryset)
def test_image_queryset_no_model(self):
(
device,
device_fw,
img_org2,
yuncore,
mesh_image,
) = self._prepare_image_qs_test_env()
# existing DeviceFirmware
# if no model specified, get all models available
device.model = ''
device.save()
form = DeviceFirmwareForm(device=device, instance=device_fw)
self.assertEqual(form.fields['image'].queryset.count(), 2)
self.assertIn(yuncore, form.fields['image'].queryset)
self.assertNotIn(img_org2, form.fields['image'].queryset)
def test_image_queryset_no_model_nor_device_firmware(self):
(
device,
device_fw,
img_org2,
yuncore,
mesh_image,
) = self._prepare_image_qs_test_env()
# new DeviceFirmware, no model
# returns all devices of the org
device.model = ''
device.save()
form = DeviceFirmwareForm(device=device)
self.assertEqual(form.fields['image'].queryset.count(), 3)
self.assertIn(device_fw.image, form.fields['image'].queryset)
self.assertIn(mesh_image, form.fields['image'].queryset)
self.assertIn(yuncore, form.fields['image'].queryset)
self.assertNotIn(img_org2, form.fields['image'].queryset)