def test_tosca_based_shell_packed_when_artifacts_missing_in_yaml_file(
self):
# Arrange
self.fs.CreateFile('nut-shell/TOSCA-Metadata/TOSCA.meta',
contents='TOSCA-Meta-File-Version: 1.0 \n'
'CSAR-Version: 1.1 \n'
'Created-By: Anonymous\n'
'Entry-Definitions: shell-definition.yml')
self.fs.CreateFile(
'nut-shell/shell-definition.yml',
contents='tosca_definitions_version: tosca_simple_yaml_1_0\n'
'metadata:\n'
' template_name: NutShell\n'
' template_author: Anonymous\n'
' template_version: 1.0.0\n'
'node_types:\n'
' vendor.switch.NXOS:\n'
' derived_from: cloudshell.nodes.Switch')
os.chdir('nut-shell')
shell_package_builder = ShellPackageBuilder()
# Act
with patch('click.echo'):
shell_package_builder.pack('/nut-shell')
# Assert
assertFileExists(self, 'dist/NutShell.zip')
TestPackageBuilder.unzip('dist/NutShell.zip', 'dist/package_content')
assertFileExists(self,
'dist/package_content/TOSCA-Metadata/TOSCA.meta')
assertFileExists(self, 'dist/package_content/shell-definition.yml')
class PackCommandExecutor(object):
def __init__(self):
self.config_reader = ShellConfigReader()
self.package_builder = PackageBuilder()
self.shell_package_builder = ShellPackageBuilder()
def pack(self):
current_path = os.getcwd()
shell_package = ShellPackage(current_path)
if shell_package.is_tosca():
self.shell_package_builder.pack(current_path)
else:
self._pack_old_school_shell(current_path)
def _pack_old_school_shell(self, current_path):
try:
config = self.config_reader.read()
self.package_builder.build_package(current_path, config.name,
config.driver_name)
except ShellYmlMissingException:
click.echo(u'shell.yml file is missing')
except WrongShellYmlException:
click.echo(u'shell.yml format is wrong')
class PackCommandExecutor(object):
def __init__(self):
self.config_reader = ShellConfigReader()
self.package_builder = PackageBuilder()
self.shell_package_builder = ShellPackageBuilder()
def pack(self):
current_path = os.getcwd()
shell_package = ShellPackage(current_path)
if shell_package.is_layer_one():
click.secho(
"Packaging a L1 shell directly via shellfoundry is not supported.",
fg="yellow")
elif shell_package.is_tosca():
self.shell_package_builder.pack(current_path)
else:
self._pack_old_school_shell(current_path)
def _pack_old_school_shell(self, current_path):
try:
config = self.config_reader.read()
self.package_builder.build_package(current_path, config.name,
config.driver_name)
except ShellYmlMissingException:
click.echo('shell.yml file is missing')
except WrongShellYmlException:
click.echo('shell.yml format is wrong')
def test_tosca_based_shell_with_icon_packed(self):
# Arrange
self.fs.CreateFile('nut-shell/TOSCA-Metadata/TOSCA.meta',
contents='TOSCA-Meta-File-Version: 1.0 \n'
'CSAR-Version: 1.1 \n'
'Created-By: Anonymous\n'
'Entry-Definitions: shell-definition.yml')
self.fs.CreateFile(
'nut-shell/shell-definition.yml',
contents='tosca_definitions_version: tosca_simple_yaml_1_0\n'
'metadata:\n'
' template_name: NutShell\n'
' template_author: Anonymous\n'
' template_version: 1.0.0\n'
' template_icon: shell.png\n'
'node_types:\n'
' vendor.switch.NXOS:\n'
' derived_from: cloudshell.nodes.Switch\n'
' artifacts:\n'
' icon:\n'
' file: nxos.png\n'
' type: tosca.artifacts.File\n'
' driver:\n'
' file: NutShellDriver.zip\n'
' type: tosca.artifacts.File')
self.fs.CreateFile('nut-shell/shell.png', contents='SHELL_IMAGE')
self.fs.CreateFile('nut-shell/nxos.png', contents='IMAGE')
self.fs.CreateFile('nut-shell/src/driver.py',
contents='DRIVER CONTENT')
os.chdir('nut-shell')
shell_package_builder = ShellPackageBuilder()
# Act
with patch('click.echo'):
shell_package_builder.pack('/nut-shell')
# Assert
assertFileExists(self, 'dist/NutShell.zip')
TestPackageBuilder.unzip('dist/NutShell.zip', 'dist/package_content')
assertFileExists(self,
'dist/package_content/TOSCA-Metadata/TOSCA.meta')
assertFileExists(self, 'dist/package_content/shell-definition.yml')
assertFileExists(self, 'dist/package_content/nxos.png')
assertFileExists(self, 'dist/package_content/shell.png')
assertFileExists(self, 'dist/package_content/NutShellDriver.zip')
def test_tosca_based_shell_failed(self):
# Arrange
self.fs.CreateFile('nut-shell/TOSCA-Metadata/TOSCA.meta',
contents='TOSCA-Meta-File-Version: 1.0 \n'
'CSAR-Version: 1.1 \n'
'Created-By: Anonymous\n'
'Entry-Definitions: shell-definition.yml')
self.fs.CreateFile(
'nut-shell/shell-definition.yml',
contents='tosca_definitions_version: tosca_simple_yaml_1_0\n'
'metadata:\n'
' template_name: NutShell\n'
' template_author: Anonymous\n'
' template_version: 1.0.0\n'
'node_types:\n'
' vendor.switch.NXOS:\n'
' derived_from: cloudshell.nodes.Switch\n'
' artifacts:\n'
' icon:\n'
' file: nxos.png\n'
' type: tosca.artifacts.File\n'
' driver:\n'
' file: NutShellDriver.zip\n'
' type: tosca.artifacts.File')
self.fs.CreateFile('nut-shell/nxos.png', contents='IMAGE')
os.chdir('nut-shell')
shell_package_builder = ShellPackageBuilder()
# Act
with self.assertRaisesRegexp(Exception, "Invalid driver structure."):
with patch('click.echo'):
shell_package_builder.pack('/nut-shell')
def __init__(self):
self.config_reader = ShellConfigReader()
self.package_builder = PackageBuilder()
self.shell_package_builder = ShellPackageBuilder()