def test_error_displayed_when_driver_generation_returns_error_code(self):
self.fs.CreateFile("nut-shell/dist/NutShell.zip", contents="ZIP")
self.fs.CreateFile("nut-shell/temp/data_model.py",
contents="python data model content")
ArchiveCreator.make_archive("nut-shell/temp/data-model", "zip",
"nut-shell/temp")
driver_generator = DriverGenerator()
config = InstallConfig("TEST-HOST", 9000, "user", "pwd", "Global",
"author", "online_mode", "template_location",
"github_login", "github_password")
with patch(
"shellfoundry.utilities.driver_generator.PackagingRestApiClient"
) as mock_rest:
rest_client_mock = Mock()
rest_client_mock.token = "TEST-TOKEN"
mock_rest.return_value = rest_client_mock
with patch("shellfoundry.utilities.driver_generator.post"
) as post_mock:
response = Mock()
response.status_code = 500
response.content = "Error occurred"
post_mock.return_value = response
with patch("shellfoundry.utilities.driver_generator.click"
) as click_mock:
click_mock.echo = MagicMock()
# Act
driver_generator.generate_driver(
cloudshell_config=config,
destination_path="nut-shell/src",
package_full_path="nut-shell/dist/NutShell.zip",
shell_filename="NutShell.zip",
shell_name="NutShell")
self.assertTrue(click_mock.echo.called,
"click should have been called")
# Assert
assertFileDoesNotExist(self, "nut-shell/src/data_model.py")
def test_error_displayed_when_failed_to_connect_to_cloudshell_server(self):
self.fs.CreateFile("nut-shell/dist/NutShell.zip", contents="ZIP")
self.fs.CreateFile("nut-shell/temp/data_model.py",
contents="python data model content")
ArchiveCreator.make_archive("nut-shell/temp/data-model", "zip",
"nut-shell/temp")
driver_generator = DriverGenerator()
config = InstallConfig("TEST-HOST", 9000, "user", "pwd", "Global",
"author", "online_mode", "template_location",
"github_login", "github_password")
with patch(
"shellfoundry.utilities.driver_generator.PackagingRestApiClient"
) as mock_rest:
mock_rest.side_effect = urllib2.URLError("connected failed")
with patch("shellfoundry.utilities.driver_generator.click"
) as click_mock:
echo_mock = MagicMock()
click_mock.echo = echo_mock
# Act
try:
driver_generator.generate_driver(
cloudshell_config=config,
destination_path="nut-shell/src",
package_full_path="nut-shell/dist/NutShell.zip",
shell_filename="NutShell.zip",
shell_name="NutShell")
except urllib2.URLError:
pass
self.assertTrue(echo_mock.called,
"click should have been called")
self.assertEqual(
echo_mock.call_args[0][0],
u"Login to CloudShell failed. Please verify the credentials in cloudshell_config.yml"
)
# Assert
assertFileDoesNotExist(self, "nut-shell/src/data_model.py")
def test_error_displayed_when_driver_generation_returns_error_code(self):
self.fs.CreateFile('nut-shell/dist/NutShell.zip', contents='ZIP')
self.fs.CreateFile('nut-shell/temp/data_model.py',
contents='python data model content')
ArchiveCreator.make_archive('nut-shell/temp/data-model', 'zip',
'nut-shell/temp')
driver_generator = DriverGenerator()
config = InstallConfig('TEST-HOST', 9000, 'user', 'pwd', 'Global')
with patch(
'shellfoundry.utilities.driver_generator.PackagingRestApiClient'
) as mock_rest:
rest_client_mock = Mock()
rest_client_mock.token = 'TEST-TOKEN'
mock_rest.return_value = rest_client_mock
with patch('shellfoundry.utilities.driver_generator.post'
) as post_mock:
response = Mock()
response.status_code = 500
response.content = 'Error occurred'
post_mock.return_value = response
with patch('shellfoundry.utilities.driver_generator.click'
) as click_mock:
click_mock.echo = MagicMock()
# Act
driver_generator.generate_driver(
cloudshell_config=config,
destination_path='nut-shell/src',
package_full_path='nut-shell/dist/NutShell.zip',
shell_filename='NutShell.zip',
shell_name='NutShell')
self.assertTrue(click_mock.echo.called,
'click should have been called')
# Assert
assertFileDoesNotExist(self, 'nut-shell/src/data_model.py')
def test_error_displayed_when_failed_to_connect_to_cloudshell_server(self):
self.fs.CreateFile('nut-shell/dist/NutShell.zip', contents='ZIP')
self.fs.CreateFile('nut-shell/temp/data_model.py',
contents='python data model content')
ArchiveCreator.make_archive('nut-shell/temp/data-model', 'zip',
'nut-shell/temp')
driver_generator = DriverGenerator()
config = InstallConfig('TEST-HOST', 9000, 'user', 'pwd', 'Global')
with patch(
'shellfoundry.utilities.driver_generator.PackagingRestApiClient'
) as mock_rest:
mock_rest.side_effect = urllib2.URLError('connected failed')
with patch('shellfoundry.utilities.driver_generator.click'
) as click_mock:
echo_mock = MagicMock()
click_mock.echo = echo_mock
# Act
try:
driver_generator.generate_driver(
cloudshell_config=config,
destination_path='nut-shell/src',
package_full_path='nut-shell/dist/NutShell.zip',
shell_filename='NutShell.zip',
shell_name='NutShell')
except urllib2.URLError:
pass
self.assertTrue(echo_mock.called,
'click should have been called')
self.assertEqual(
echo_mock.call_args[0][0],
u'Login to CloudShell failed. Please verify the credentials in cloudshell_config.yml'
)
# Assert
assertFileDoesNotExist(self, 'nut-shell/src/data_model.py')