def test_structure_2_with_package(self):
images = {'image': [self.IMAGE], 'packages': [self.PACKAGE_1]}
image_handler = ImageHandler(self.device, images)
self.assertEqual(image_handler.image, self.EXPECTED_IMAGE)
self.assertEqual(image_handler.packages, self.EXPECTED_SINGLE_PKG)
def test_structure_1_with_packages(self):
images = [self.IMAGE, self.PACKAGE_1, self.PACKAGE_2]
image_handler = ImageHandler(self.device, images)
self.assertEqual(image_handler.image, self.EXPECTED_IMAGE)
self.assertEqual(image_handler.packages, self.EXPECTED_DOUBLE_PKG)
def test_structure_2_without_packages(self):
images = {
'image': [self.IMAGE],
}
image_handler = ImageHandler(self.device, images)
self.assertEqual(image_handler.image, self.EXPECTED_IMAGE)
def test_structure_1_without_package(self):
images = [
self.IMAGE,
]
image_handler = ImageHandler(self.device, images)
self.assertEqual(image_handler.image, self.EXPECTED_IMAGE)
def test_structure_3_with_packages(self):
images = {
'image': {
'file': [self.IMAGE]
},
'packages': {
'file': [self.PACKAGE_1, self.PACKAGE_2]
}
}
image_handler = ImageHandler(self.device, images)
self.assertEqual(image_handler.image, self.EXPECTED_IMAGE)
self.assertEqual(image_handler.packages, self.EXPECTED_DOUBLE_PKG)
def test_structure_1_missing_entry(self):
images = []
with self.assertRaises(Exception):
ImageHandler(self.device, images)