def test_generate_driver_called_on_tosca_based_shell(self):
# Arrange
self.fs.CreateFile('nut_shell/shell.yml',
contents="""
shell:
name: nut_shell
driver: NutShellDriver
""")
self.fs.CreateFile('nut_shell/TOSCA-Metadata/TOSCA.meta',
contents="""
TOSCA-Meta-File-Version: 1.0
CSAR-Version: 0.1.0
Created-By: Anonymous
Entry-Definitions: shell-definition.yml
""")
os.chdir('nut_shell')
driver_generator = MagicMock()
driver_generator.generate_driver = MagicMock()
command_executor = GenerateCommandExecutor(
driver_generator=driver_generator)
# Act
command_executor.generate()
# Assert
self.assertTrue(driver_generator.generate_driver.called)
self.assertEqual(
driver_generator.generate_driver.call_args[1]['shell_filename'],
'NutShell.zip')
def test_generate_driver_not_called_on_non_tosca_based_shell(self):
# Arrange
self.fs.CreateFile('nut_shell/shell.yml',
contents="""
shell:
name: nut_shell
driver: NutShellDriver
""")
os.chdir('nut_shell')
driver_generator = MagicMock()
command_executor = GenerateCommandExecutor(
driver_generator=driver_generator)
# Act
command_executor.generate()
# Assert
driver_generator.generate_driver.assert_not_called()
def generate():
"""
Generates Python driver data model to be used in driver code
"""
PackCommandExecutor().pack()
GenerateCommandExecutor().generate()