def test_deploy_this_environment_allowed_with_environment(
self, deployed, status):
self.deploy_environment = tables.DeployThisEnvironment()
deployed.return_value = True
status.return_value = consts.STATUS_ID_READY, "version"
self.deploy_environment.table = mock.Mock()
self.deploy_environment.table.kwargs = {'environment_id': 'id'}
self.assertFalse(self.deploy_environment.allowed(None, None))
self.assertEqual('Update This Environment',
self.deploy_environment.verbose_name)
deployed.return_value = False
self.assertFalse(self.deploy_environment.allowed(None, None))
self.assertEqual('Deploy This Environment',
self.deploy_environment.verbose_name)
status.return_value = "", 0
self.deploy_environment.table.data = None
self.assertFalse(self.deploy_environment.allowed(None, None))
status.return_value = "", 0
self.deploy_environment.table.data = 'data'
self.assertTrue(self.deploy_environment.allowed(None, None))
def test_deploy_this_environment_single_exception(self, mock_deploy,
mock_messages, reverse):
self.deploy_environment = tables.DeployThisEnvironment()
data_table = mock.Mock()
data_table.kwargs = {'environment_id': 'id'}
mock_deploy.side_effect = Exception("test")
self.assertRaises(BaseException, self.deploy_environment.single,
data_table, None, None)
def test_deploy_this_environment_single(self, mock_deploy, mock_messages,
reverse):
self.deploy_environment = tables.DeployThisEnvironment()
data_table = mock.Mock()
data_table.kwargs = {'environment_id': 'id'}
mock_deploy.side_effect = None
self.deploy_environment.single(data_table, None, None)
self.assertTrue(mock_messages.success.called)
