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))