def test_compose_timeout_message(): remaining_instances = { "cluster1": ["instance1", "instance2"], "cluster2": ["instance3"], "cluster3": [], } message = mark_for_deployment.compose_timeout_message( remaining_instances, 1, "fake_group", "someservice", DeploymentVersion(sha="some_git_sha", image_version="extrastuff"), ) assert (" paasta status -c cluster1 -s someservice -i instance1,instance2" in message) assert " paasta status -c cluster2 -s someservice -i instance3" in message assert ( " paasta logs -c cluster1 -s someservice -i instance1,instance2 -C deploy -l 1000" in message) assert ( " paasta logs -c cluster2 -s someservice -i instance3 -C deploy -l 1000" in message) assert ( " paasta wait-for-deployment -s someservice -l fake_group -c some_git_sha --image-version extrastuff" in message)
def test_compose_timeout_message(): clusters_data = [] clusters_data.append( mark_for_deployment.ClusterData(cluster='cluster1', service='someservice', git_sha='somesha', instances_queue=Queue())) clusters_data[0].instances_queue.put('instance1') clusters_data[0].instances_queue.put('instance2') clusters_data.append( mark_for_deployment.ClusterData(cluster='cluster2', service='someservice', git_sha='somesha', instances_queue=Queue())) clusters_data[1].instances_queue.put('instance3') clusters_data.append( mark_for_deployment.ClusterData(cluster='cluster3', service='someservice', git_sha='somesha', instances_queue=Queue())) message = mark_for_deployment.compose_timeout_message( clusters_data, 1, 'fake_group', 'someservice', 'some_git_sha') assert ' paasta status -c cluster1 -s someservice -i instance1,instance2' in message assert ' paasta status -c cluster2 -s someservice -i instance3' in message assert ' paasta logs -c cluster1 -s someservice -i instance1,instance2 -C deploy -l 1000' in message assert ' paasta logs -c cluster2 -s someservice -i instance3 -C deploy -l 1000' in message
def test_compose_timeout_message(): clusters_data = [] clusters_data.append( mark_for_deployment.ClusterData( cluster="cluster1", service="someservice", git_sha="somesha", instances_queue=Queue(), )) clusters_data[0].instances_queue.put( mock_marathon_instance_config("instance1")) clusters_data[0].instances_queue.put( mock_marathon_instance_config("instance2")) clusters_data.append( mark_for_deployment.ClusterData( cluster="cluster2", service="someservice", git_sha="somesha", instances_queue=Queue(), )) clusters_data[1].instances_queue.put( mock_marathon_instance_config("instance3")) clusters_data.append( mark_for_deployment.ClusterData( cluster="cluster3", service="someservice", git_sha="somesha", instances_queue=Queue(), )) message = mark_for_deployment.compose_timeout_message( clusters_data, 1, "fake_group", "someservice", "some_git_sha") assert (" paasta status -c cluster1 -s someservice -i instance1,instance2" in message) assert " paasta status -c cluster2 -s someservice -i instance3" in message assert ( " paasta logs -c cluster1 -s someservice -i instance1,instance2 -C deploy -l 1000" in message) assert ( " paasta logs -c cluster2 -s someservice -i instance3 -C deploy -l 1000" in message)