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_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 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])