def test_report_status_obeys_cluster_whitelist(
mock_stdout,
mock_report_invalid_whitelist_values,
mock_report_status_for_cluster,
):
service = 'fake_service'
cluster_whitelist = ['cluster1']
instance_whitelist = []
deploy_pipeline = actual_deployments = [
'cluster1.main', 'cluster2.main', 'cluster3.main']
report_status(
service=service,
deploy_pipeline=deploy_pipeline,
actual_deployments=actual_deployments,
cluster_whitelist=cluster_whitelist,
instance_whitelist=instance_whitelist,
)
mock_report_invalid_whitelist_values.assert_called_once_with(
cluster_whitelist, ['cluster1', 'cluster2', 'cluster3'], 'cluster')
mock_report_status_for_cluster.assert_called_once_with(
service=service,
cluster='cluster1',
deploy_pipeline=deploy_pipeline,
actual_deployments=actual_deployments,
instance_whitelist=instance_whitelist,
verbose=False
)
def test_report_status_obeys_filter(
mock_stdout,
mock_report_bogus_filters,
mock_report_status_for_cluster,
):
service = 'fake_service'
cluster_filter = ['cluster1']
deploy_pipeline = actual_deployments = [
'cluster1.main', 'cluster2.main', 'cluster3.main']
report_status(
service, deploy_pipeline, actual_deployments, cluster_filter)
mock_report_bogus_filters.assert_called_once_with(
cluster_filter, ['cluster1', 'cluster2', 'cluster3'])
mock_report_status_for_cluster.assert_called_once_with(service, 'cluster1', deploy_pipeline,
actual_deployments, False)
def test_report_status_handle_none_filter(
mock_stdout,
mock_report_bogus_filters,
mock_report_status_for_cluster,
):
service = 'fake_service'
cluster_filter = None
deploy_pipeline = actual_deployments = [
'cluster1.main', 'cluster2.main', 'cluster3.main']
report_status(
service, deploy_pipeline, actual_deployments, cluster_filter)
mock_report_status_for_cluster.assert_any_call(
service, 'cluster1', deploy_pipeline, actual_deployments, False)
mock_report_status_for_cluster.assert_any_call(
service, 'cluster2', deploy_pipeline, actual_deployments, False)
mock_report_status_for_cluster.assert_any_call(
service, 'cluster3', deploy_pipeline, actual_deployments, False)
def test_report_status_handle_none_whitelist(
mock_stdout,
mock_report_invalid_whitelist_values,
mock_report_status_for_cluster,
):
service = 'fake_service'
cluster_whitelist = []
instance_whitelist = []
deploy_pipeline = actual_deployments = [
'cluster1.main', 'cluster2.main', 'cluster3.main']
report_status(
service=service,
deploy_pipeline=deploy_pipeline,
actual_deployments=actual_deployments,
cluster_whitelist=cluster_whitelist,
instance_whitelist=instance_whitelist,
)
mock_report_status_for_cluster.assert_any_call(
service=service,
cluster='cluster1',
deploy_pipeline=deploy_pipeline,
actual_deployments=actual_deployments,
instance_whitelist=instance_whitelist,
verbose=False
)
mock_report_status_for_cluster.assert_any_call(
service=service,
cluster='cluster2',
deploy_pipeline=deploy_pipeline,
actual_deployments=actual_deployments,
instance_whitelist=instance_whitelist,
verbose=False
)
mock_report_status_for_cluster.assert_any_call(
service=service,
cluster='cluster3',
deploy_pipeline=deploy_pipeline,
actual_deployments=actual_deployments,
instance_whitelist=instance_whitelist,
verbose=False
)
