def test_make_mutate_refs_func(): mutate_refs = start_stop_restart.make_mutate_refs_func( service_config=MarathonServiceConfig( cluster='fake_cluster', instance='fake_instance', service='fake_service', config_dict={'deploy_group': 'a'}, branch_dict=None, ), force_bounce='FORCEBOUNCE', desired_state='stop', ) old_refs = { 'refs/tags/paasta-a-20160308T053933-deploy': 'hash_for_a', 'refs/tags/paasta-b-20160308T053933-deploy': 'hash_for_b', 'refs/tags/paasta-c-20160308T053933-deploy': 'hash_for_c', 'refs/tags/paasta-d-20160308T053933-deploy': 'hash_for_d', } expected = dict(old_refs) expected.update({ 'refs/tags/paasta-fake_cluster.fake_instance-FORCEBOUNCE-stop': 'hash_for_a', }) actual = mutate_refs(old_refs) assert actual == expected
def test_make_mutate_refs_func(): mutate_refs = start_stop_restart.make_mutate_refs_func( service_config=MarathonServiceConfig( cluster="fake_cluster", instance="fake_instance", service="fake_service", config_dict={"deploy_group": "a"}, branch_dict=None, ), force_bounce="FORCEBOUNCE", desired_state="stop", ) old_refs = { "refs/tags/paasta-a-20160308T053933-deploy": "hash_for_a", "refs/tags/paasta-b-20160308T053933-deploy": "hash_for_b", "refs/tags/paasta-c-20160308T053933-deploy": "hash_for_c", "refs/tags/paasta-d-20160308T053933-deploy": "hash_for_d", } expected = dict(old_refs) expected.update( {"refs/tags/paasta-fake_cluster.fake_instance-FORCEBOUNCE-stop": "hash_for_a"} ) actual = mutate_refs(old_refs) assert actual == expected
def test_make_mutate_refs_func(): mutate_refs = start_stop_restart.make_mutate_refs_func( service_config=MarathonServiceConfig( cluster='fake_cluster', instance='fake_instance', service='fake_service', config_dict={'deploy_group': 'a'}, branch_dict={}, ), force_bounce='FORCEBOUNCE', desired_state='stop', ) old_refs = { 'refs/tags/paasta-a-20160308T053933-deploy': 'hash_for_a', 'refs/tags/paasta-b-20160308T053933-deploy': 'hash_for_b', 'refs/tags/paasta-c-20160308T053933-deploy': 'hash_for_c', 'refs/tags/paasta-d-20160308T053933-deploy': 'hash_for_d', } expected = dict(old_refs) expected.update({ 'refs/tags/paasta-fake_cluster.fake_instance-FORCEBOUNCE-stop': 'hash_for_a', }) actual = mutate_refs(old_refs) assert actual == expected
def test_make_mutate_refs_func(): mutate_refs = start_stop_restart.make_mutate_refs_func( branches=['a', 'b'], force_bounce='FORCEBOUNCE', desired_state='stop', ) old_refs = { 'refs/heads/a': 'hash_for_a', 'refs/heads/b': 'hash_for_b', 'refs/heads/c': 'hash_for_c', 'refs/heads/d': 'hash_for_d', } expected = dict(old_refs) expected.update({ 'refs/tags/paasta-a-FORCEBOUNCE-stop': 'hash_for_a', 'refs/tags/paasta-b-FORCEBOUNCE-stop': 'hash_for_b', }) actual = mutate_refs(old_refs) assert actual == expected