def test_flow_run_unexpected_exception(self, handle_sentry_event):
org_config = mock.Mock(config={})
config = CliRuntime(
config={
"flows": {"test": {"steps": {1: {"task": "test_task"}}}},
"tasks": {
"test_task": {
"class_path": "cumulusci.cli.tests.test_cci.DummyTask",
"description": "Test Task",
}
},
},
load_keychain=False,
)
config.get_org = mock.Mock(return_value=("test", org_config))
DummyTask._run_task = mock.Mock(side_effect=Exception)
with self.assertRaises(Exception):
run_click_command(
cci.flow_run,
config=config,
flow_name="test",
org="test",
delete_org=False,
debug=False,
o=None,
skip=(),
no_prompt=True,
)
handle_sentry_event.assert_called_once()
def test_flow_run_org_delete_error(self, echo):
org_config = mock.Mock(scratch=True, config={})
org_config.delete_org.side_effect = Exception
config = CliRuntime(
config={
"flows": {"test": {"steps": {1: {"task": "test_task"}}}},
"tasks": {
"test_task": {
"class_path": "cumulusci.cli.tests.test_cci.DummyTask",
"description": "Test Task",
}
},
},
load_keychain=False,
)
config.get_org = mock.Mock(return_value=("test", org_config))
DummyTask._run_task = mock.Mock()
run_click_command(
cci.flow_run,
config=config,
flow_name="test",
org="test",
delete_org=True,
debug=False,
o=None,
skip=(),
no_prompt=True,
)
echo.assert_any_call(
"Scratch org deletion failed. Ignoring the error below to complete the flow:"
)
def test_flow_run(self):
org_config = mock.Mock(scratch=True, config={})
config = CliRuntime(
config={
"flows": {"test": {"steps": {1: {"task": "test_task"}}}},
"tasks": {
"test_task": {
"class_path": "cumulusci.cli.tests.test_cci.DummyTask",
"description": "Test Task",
}
},
},
load_keychain=False,
)
config.get_org = mock.Mock(return_value=("test", org_config))
config.get_flow = mock.Mock()
run_click_command(
cci.flow_run,
config=config,
flow_name="test",
org="test",
delete_org=True,
debug=False,
o=[("test_task__color", "blue")],
skip=(),
no_prompt=True,
)
config.get_flow.assert_called_once_with(
"test", options={"test_task": {"color": "blue"}}
)
org_config.delete_org.assert_called_once()
def test_flow_run_usage_error(self):
org_config = mock.Mock(config={})
config = CliRuntime(
config={
"flows": {"test": {"steps": {1: {"task": "test_task"}}}},
"tasks": {
"test_task": {
"class_path": "cumulusci.cli.tests.test_cci.DummyTask",
"description": "Test Task",
}
},
},
load_keychain=False,
)
config.get_org = mock.Mock(return_value=("test", org_config))
DummyTask._run_task = mock.Mock(side_effect=TaskOptionsError)
with self.assertRaises(click.UsageError):
run_click_command(
cci.flow_run,
config=config,
flow_name="test",
org="test",
delete_org=False,
debug=False,
o=None,
skip=(),
no_prompt=True,
)
def test_flow_run(self):
org_config = mock.Mock(scratch=True, config={})
config = CliRuntime(
config={
"flows": {"test": {"steps": {1: {"task": "test_task"}}}},
"tasks": {
"test_task": {
"class_path": "cumulusci.cli.tests.test_cci.DummyTask",
"description": "Test Task",
}
},
},
load_keychain=False,
)
config.get_org = mock.Mock(return_value=("test", org_config))
config.get_flow = mock.Mock()
run_click_command(
cci.flow_run,
config=config,
flow_name="test",
org="test",
delete_org=True,
debug=False,
o=[("test_task__color", "blue")],
skip=(),
no_prompt=True,
)
config.get_flow.assert_called_once_with(
"test", options={"test_task": {"color": "blue"}}
)
org_config.delete_org.assert_called_once()
def test_flow_run_expected_failure(self):
org_config = mock.Mock(config={})
config = CliRuntime(
config={
"flows": {"test": {"steps": {1: {"task": "test_task"}}}},
"tasks": {
"test_task": {
"class_path": "cumulusci.cli.tests.test_cci.DummyTask",
"description": "Test Task",
}
},
},
load_keychain=False,
)
config.get_org = mock.Mock(return_value=("test", org_config))
DummyTask._run_task = mock.Mock(side_effect=ScratchOrgException("msg"))
with self.assertRaises(click.ClickException) as e:
run_click_command(
cci.flow_run,
config=config,
flow_name="test",
org="test",
delete_org=False,
debug=False,
o=None,
skip=(),
no_prompt=True,
)
assert "msg" in str(e)
def test_flow_run_org_delete_error(self, echo):
org_config = mock.Mock(scratch=True, config={})
org_config.delete_org.side_effect = Exception
config = CliRuntime(
config={
"flows": {"test": {"steps": {1: {"task": "test_task"}}}},
"tasks": {
"test_task": {
"class_path": "cumulusci.cli.tests.test_cci.DummyTask",
"description": "Test Task",
}
},
},
load_keychain=False,
)
config.get_org = mock.Mock(return_value=("test", org_config))
DummyTask._run_task = mock.Mock()
run_click_command(
cci.flow_run,
config=config,
flow_name="test",
org="test",
delete_org=True,
debug=False,
o=None,
skip=(),
no_prompt=True,
)
echo.assert_any_call(
"Scratch org deletion failed. Ignoring the error below to complete the flow:"
)
def test_flow_run_unexpected_exception(self, handle_sentry_event):
org_config = mock.Mock(config={})
config = CliRuntime(
config={
"flows": {"test": {"steps": {1: {"task": "test_task"}}}},
"tasks": {
"test_task": {
"class_path": "cumulusci.cli.tests.test_cci.DummyTask",
"description": "Test Task",
}
},
},
load_keychain=False,
)
config.get_org = mock.Mock(return_value=("test", org_config))
DummyTask._run_task = mock.Mock(side_effect=Exception)
with self.assertRaises(Exception):
run_click_command(
cci.flow_run,
config=config,
flow_name="test",
org="test",
delete_org=False,
debug=False,
o=None,
skip=(),
no_prompt=True,
)
handle_sentry_event.assert_called_once()
def test_flow_run_usage_error(self):
org_config = mock.Mock(config={})
config = CliRuntime(
config={
"flows": {"test": {"steps": {1: {"task": "test_task"}}}},
"tasks": {
"test_task": {
"class_path": "cumulusci.cli.tests.test_cci.DummyTask",
"description": "Test Task",
}
},
},
load_keychain=False,
)
config.get_org = mock.Mock(return_value=("test", org_config))
DummyTask._run_task = mock.Mock(side_effect=TaskOptionsError)
with self.assertRaises(click.UsageError):
run_click_command(
cci.flow_run,
config=config,
flow_name="test",
org="test",
delete_org=False,
debug=False,
o=None,
skip=(),
no_prompt=True,
)
def test_get_org_missing(self):
config = CliRuntime()
config.keychain = mock.Mock()
config.keychain.get_org.return_value = None
with self.assertRaises(click.UsageError):
org_name, org_config_result = config.get_org("test", fail_if_missing=True)
def test_get_org(self):
config = CliRuntime()
config.keychain = mock.Mock()
config.keychain.get_org.return_value = org_config = OrgConfig({}, "test")
org_name, org_config_result = config.get_org("test")
self.assertEqual("test", org_name)
self.assertIs(org_config, org_config_result)
