def test_cluster_status_cli(libraries_api_mock):
libraries_api_mock.cluster_status.return_value = CLUSTER_STATUS_RETURN
runner = CliRunner()
res = runner.invoke(cli.cluster_status_cli,
['--cluster-id', TEST_CLUSTER_ID])
libraries_api_mock.cluster_status.assert_called_with(TEST_CLUSTER_ID)
assert_cli_output(res.output, pretty_format(CLUSTER_STATUS_RETURN))
def test_list_cli_with_cluster_id():
with mock.patch('databricks_cli.libraries.cli.cluster_status'
) as cluster_status_mock:
cluster_status_mock.return_value = CLUSTER_STATUS_RETURN
runner = CliRunner()
res = runner.invoke(cli.list_cli, ['--cluster-id', TEST_CLUSTER_ID])
cluster_status_mock.assert_called_with(TEST_CLUSTER_ID)
assert_cli_output(res.output, pretty_format(CLUSTER_STATUS_RETURN))
def test_list_cli_without_cluster_id():
with mock.patch('databricks_cli.libraries.cli.all_cluster_statuses') as \
all_cluster_statuses_mock:
all_cluster_statuses_mock.return_value = ALL_CLUSTER_STATUSES_RETURN
runner = CliRunner()
res = runner.invoke(cli.list_cli)
all_cluster_statuses_mock.assert_called_once()
assert_cli_output(res.output,
pretty_format(ALL_CLUSTER_STATUSES_RETURN))
def test_edit_cli_no_args():
runner = CliRunner()
res = runner.invoke(cli.edit_cli, [])
assert_cli_output(
res.output,
'Error: RuntimeError: Either --json-file or --json should be provided')
def test_list_cli_without_cluster_id(libraries_api_mock):
libraries_api_mock.all_cluster_statuses.return_value = ALL_CLUSTER_STATUSES_RETURN
runner = CliRunner()
res = runner.invoke(cli.list_cli)
libraries_api_mock.all_cluster_statuses.assert_called_once()
assert_cli_output(res.output, pretty_format(ALL_CLUSTER_STATUSES_RETURN))
