def create_pimages(pimage=None):
pimages = uforge.publishImages()
pimages.publishImages = pyxb.BIND()
if pimage:
pimages.publishImages.append(pimage)
return pimages
def test_do_list_check_size(self, mock_table_add_row, mock_api_pimg_getall, mock_api_getall):
# given
i = self.prepare_image()
mock_api_getall.return_value = self.create_images(6000, "users/myuser/whatever/12/testing/18")
new_pimages = uforge.publishImages()
new_pimages.publishImages = pyxb.BIND()
mock_api_pimg_getall.return_value = new_pimages
# when
i.do_list("")
# then
self.assertEquals(mock_table_add_row.call_count, 1)
mock_table_add_row.assert_called_with([ANY, ANY, ANY, ANY, ANY, ANY, size(6000), ANY, ANY])
def prepare_aws_pimages_from_scan(self):
publish_images = uforge.publishImages()
publish_images.publishImages = pyxb.BIND()
publish_image = PublishImageAws()
publish_image.dbId = 1234
publish_image.imageUri = 'users/guest/scannedinstances/5/scans/12/images/1234'
publish_image.status = "complete"
publish_image.status.complete = True
publish_image.targetFormat = uforge.targetFormat()
publish_image.targetFormat.name = "Amazon AWS"
publish_images.publishImages.append(publish_image)
return publish_images
def prepare_aws_pimages_from_scan(self):
publish_images = uforge.publishImages()
publish_images.publishImages = pyxb.BIND()
publish_image = PublishImageAws()
publish_image.dbId = 1234
publish_image.imageUri = 'users/guest/scannedinstances/5/scans/12/images/1234'
publish_image.status = "complete"
publish_image.status.complete = True
publish_image.targetFormat = uforge.targetFormat()
publish_image.targetFormat.name = "Amazon AWS"
publish_images.publishImages.append(publish_image)
return publish_images
def prepare_azure_pimages_from_scan(self):
publish_images = uforge.publishImages()
publish_images.publishImages = pyxb.BIND()
publish_image = PublishImageAzureResourceManager()
publish_image.dbId = 1234
publish_image.imageUri = 'users/guest/scannedinstances/5/scans/12/images/1234'
publish_image.status = "complete"
publish_image.status.complete = True
publish_image.targetFormat = uforge.targetFormat()
publish_image.targetFormat.name = "Microsoft Azure"
publish_images.publishImages.append(publish_image)
return publish_images
def prepare_azure_pimages_from_app(self):
publish_images = uforge.publishImages()
publish_images.publishImages = pyxb.BIND()
publish_image = PublishImageAzure()
publish_image.dbId = 1234
publish_image.imageUri = 'users/guest/appliances/5/images/1234'
publish_image.parentUri = 'users/guest/appliances/5'
publish_image.status = "complete"
publish_image.status.complete = True
publish_image.targetFormat = uforge.targetFormat()
publish_image.targetFormat.name = "Microsoft Azure"
publish_images.publishImages.append(publish_image)
return publish_images
def prepare_azure_pimages_from_app(self):
publish_images = uforge.publishImages()
publish_images.publishImages = pyxb.BIND()
publish_image = PublishImageAzure()
publish_image.dbId = 1234
publish_image.imageUri = 'users/guest/appliances/5/images/1234'
publish_image.parentUri = 'users/guest/appliances/5'
publish_image.status = "complete"
publish_image.status.complete = True
publish_image.targetFormat = uforge.targetFormat()
publish_image.targetFormat.name = "Microsoft Azure"
publish_images.publishImages.append(publish_image)
return publish_images
def test_do_list_check_size(self, mock_table_add_row, mock_api_pimg_getall, mock_api_getall):
# given
i = image.Image()
i.api = Api("url", username="******", password="******", headers=None,
disable_ssl_certificate_validation=False, timeout=constants.HTTP_TIMEOUT)
i.login = "******"
i.password = "******"
self.create_image(6000, mock_api_getall)
new_pimages = uforge.publishImages()
new_pimages.publishImages = pyxb.BIND()
mock_api_pimg_getall.return_value = new_pimages
# when
i.do_list("")
# then
self.assertEquals(mock_table_add_row.call_count, 1)
mock_table_add_row.assert_called_with([ANY, ANY, ANY, ANY, ANY, ANY, size(6000), ANY, ANY])
def prepare_openstack_publish_images_from_scan(self):
publish_images = uforge.publishImages()
publish_images.publishImages = pyxb.BIND()
publish_image = PublishImageOpenStack()
publish_image.dbId = 1234
publish_image.imageUri = 'users/guest/scannedinstances/5/scans/12/images/116'
publish_image.status = "complete"
publish_image.status.complete = True
publish_image.targetFormat = uforge.targetFormat()
publish_image.targetFormat.name = "OpenStack"
publish_image.uri = "users/guest/scannedinstances/5/scans/12/images/116/pimages/1234"
publish_image.credAccount = self.prepare_openstack_credaccount();
publish_images.publishImages.append(publish_image)
return publish_images
def test_do_list_check_size(self, mock_table_add_row, mock_api_pimg_getall,
mock_api_getall):
# given
i = image.Image()
i.api = Api("url",
username="******",
password="******",
headers=None,
disable_ssl_certificate_validation=False,
timeout=constants.HTTP_TIMEOUT)
i.login = "******"
i.password = "******"
self.create_image(6000, mock_api_getall)
new_pimages = uforge.publishImages()
new_pimages.publishImages = pyxb.BIND()
mock_api_pimg_getall.return_value = new_pimages
# when
i.do_list("")
# then
self.assertEquals(mock_table_add_row.call_count, 1)
mock_table_add_row.assert_called_with(
[ANY, ANY, ANY, ANY, ANY, ANY,
size(6000), ANY, ANY])
