def test_main_deploy_action_no_apprc(self, unlink_mock, client_mock, tasks_mock):
register_mock = client_mock.return_value.register_unit
register_mock.return_value = ({'env1': 'val1', 'port': '8888', 'PORT': '8888'}, "0.17.0")
exec_script_mock = tasks_mock.execute_start_script
load_yaml_mock = tasks_mock.load_app_yaml
load_yaml_mock.return_value = {'hooks': {'build': ['cmd_1', 'cmd_2']}}
post_app_yaml_mock = client_mock.return_value.post_app_yaml
load_procfile_mock = tasks_mock.load_procfile
load_procfile_mock.return_value = 'web: python myproject.py\nworker: ./startworker'
parse_procfile_mock = tasks_mock.parse_procfile
parse_procfile_mock.return_value = {'web': 'python myproject.py',
'worker': './startworker'}
run_build_hooks_mock = tasks_mock.run_build_hooks
write_circus_conf_mock = tasks_mock.write_circus_conf
save_apprc_mock = tasks_mock.save_apprc_file
main()
call_count = len(client_mock.mock_calls) + len(tasks_mock.mock_calls)
self.assertEqual(call_count, 12)
client_mock.assert_called_once_with('http://localhost', 'token')
register_mock.assert_any_call('app1')
v = load_yaml_mock.return_value
v['procfile'] = load_procfile_mock.return_value
v['processes'] = parse_procfile_mock.return_value
register_mock.assert_any_call('app1', v)
expected_calls = [mock.call({'port': '8888', 'PORT': '8888'}),
mock.call({'env1': 'val1', 'port': '8888', 'PORT': '8888'},
file_path=TEMP_ENV_FILE)]
self.assertEqual(expected_calls, save_apprc_mock.mock_calls)
unlink_mock.assert_called_once_with(TEMP_ENV_FILE)
exec_script_mock.assert_called_once_with('mycmd')
load_yaml_mock.assert_called_once_with()
write_circus_conf_mock.assert_called_once_with(envs=register_mock.return_value[0])
post_app_yaml_mock.assert_called_once_with('app1', load_yaml_mock.return_value)
run_build_hooks_mock.assert_called_once_with(load_yaml_mock.return_value,
envs=register_mock.return_value[0])
def test_main_run_action_api_error(self, client_mock, tasks_mock):
register_mock = client_mock.return_value.register_unit
def fail(*args):
raise ConnectionError()
register_mock.side_effect = fail
save_apprc_mock = tasks_mock.save_apprc_file
parse_apprc_mock = tasks_mock.parse_apprc_file
parse_apprc_mock.return_value = {'env1': 'val1'}
exec_script_mock = tasks_mock.execute_start_script
run_restart_hooks_mock = tasks_mock.run_restart_hooks
write_circus_conf_mock = tasks_mock.write_circus_conf
load_yaml_mock = tasks_mock.load_app_yaml
load_yaml_mock.return_value = {'hooks': {'build': ['cmd_1', 'cmd_2']}}
main()
call_count = len(client_mock.mock_calls) + len(tasks_mock.mock_calls)
self.assertEqual(call_count, 8)
write_circus_conf_mock.assert_called_once_with(envs={'env1': 'val1'})
client_mock.assert_called_once_with('http://localhost', 'token')
register_mock.assert_called_once_with('app1')
parse_apprc_mock.assert_called_once_with()
self.assertEqual(save_apprc_mock.call_count, 0)
exec_script_mock.assert_called_once_with('mycmd', envs={'env1': 'val1'}, with_shell=False)
load_yaml_mock.assert_called_once_with()
run_restart_hooks_mock.assert_any_call('before', load_yaml_mock.return_value,
envs={'env1': 'val1'})
run_restart_hooks_mock.assert_any_call('after', load_yaml_mock.return_value,
envs={'env1': 'val1'})
def test_main_run_action_no_apprc(self, unlink_mock, client_mock, tasks_mock):
register_mock = client_mock.return_value.register_unit
register_mock.return_value = ({'env1': 'val1', 'port': '8888', 'PORT': '8888'}, "0.17.1")
save_apprc_mock = tasks_mock.save_apprc_file
exec_script_mock = tasks_mock.execute_start_script
run_restart_hooks_mock = tasks_mock.run_restart_hooks
write_circus_conf_mock = tasks_mock.write_circus_conf
load_yaml_mock = tasks_mock.load_app_yaml
load_yaml_mock.return_value = {'hooks': {'build': ['cmd_1', 'cmd_2']}}
main()
call_count = len(client_mock.mock_calls) + len(tasks_mock.mock_calls)
self.assertEqual(call_count, 9)
write_circus_conf_mock.assert_called_once_with(envs=register_mock.return_value[0])
client_mock.assert_called_once_with('http://localhost', 'token')
register_mock.assert_called_once_with('app1')
expected_calls = [mock.call({'port': '8888', 'PORT': '8888'}),
mock.call({'env1': 'val1', 'port': '8888', 'PORT': '8888'},
file_path=TEMP_ENV_FILE)]
self.assertEqual(expected_calls, save_apprc_mock.mock_calls)
unlink_mock.assert_called_once_with(TEMP_ENV_FILE)
exec_script_mock.assert_called_once_with('mycmd', envs=register_mock.return_value[0],
with_shell=False)
load_yaml_mock.assert_called_once_with()
run_restart_hooks_mock.assert_any_call('before', load_yaml_mock.return_value,
envs=register_mock.return_value[0])
run_restart_hooks_mock.assert_any_call('after', load_yaml_mock.return_value,
envs=register_mock.return_value[0])
def test_main_deploy_action(self, client_mock, tasks_mock):
register_mock = client_mock.return_value.register_unit
register_mock.return_value = {'env1': 'val1'}
exec_script_mock = tasks_mock.execute_start_script
load_yaml_mock = tasks_mock.load_app_yaml
load_yaml_mock.return_value = {'hooks': {'build': ['cmd_1', 'cmd_2']}}
post_app_yaml_mock = client_mock.return_value.post_app_yaml
load_procfile_mock = tasks_mock.load_procfile
load_procfile_mock.return_value = 'web: python myproject.py\nworker: ./startworker'
run_build_hooks_mock = tasks_mock.run_build_hooks
write_circus_conf_mock = tasks_mock.write_circus_conf
save_apprc_mock = tasks_mock.save_apprc_file
main()
call_count = len(client_mock.mock_calls) + len(tasks_mock.mock_calls)
self.assertEqual(call_count, 10)
client_mock.assert_called_once_with('http://localhost', 'token')
register_mock.assert_any_call('app1')
v = load_yaml_mock.return_value
v['procfile'] = load_procfile_mock.return_value
register_mock.assert_any_call('app1', v)
save_apprc_mock.assert_called_once_with(register_mock.return_value)
exec_script_mock.assert_called_once_with('mycmd')
load_yaml_mock.assert_called_once_with()
write_circus_conf_mock.assert_called_once_with(envs={'env1': 'val1'})
post_app_yaml_mock.assert_called_once_with('app1',
load_yaml_mock.return_value)
run_build_hooks_mock.assert_called_once_with(
load_yaml_mock.return_value, envs={'env1': 'val1'})
def test_main_deploy_action(self, client_mock, tasks_mock):
register_mock = client_mock.return_value.register_unit
register_mock.return_value = {'env1': 'val1'}
exec_script_mock = tasks_mock.execute_start_script
load_yaml_mock = tasks_mock.load_app_yaml
load_yaml_mock.return_value = {'hooks': {'build': ['cmd_1', 'cmd_2']}}
post_app_yaml_mock = client_mock.return_value.post_app_yaml
load_procfile_mock = tasks_mock.load_procfile
load_procfile_mock.return_value = 'web: python myproject.py\nworker: ./startworker'
run_build_hooks_mock = tasks_mock.run_build_hooks
write_circus_conf_mock = tasks_mock.write_circus_conf
save_apprc_mock = tasks_mock.save_apprc_file
main()
call_count = len(client_mock.mock_calls) + len(tasks_mock.mock_calls)
self.assertEqual(call_count, 10)
client_mock.assert_called_once_with('http://localhost', 'token')
register_mock.assert_any_call('app1')
v = load_yaml_mock.return_value
v['procfile'] = load_procfile_mock.return_value
register_mock.assert_any_call('app1', v)
save_apprc_mock.assert_called_once_with(register_mock.return_value)
exec_script_mock.assert_called_once_with('mycmd')
load_yaml_mock.assert_called_once_with()
write_circus_conf_mock.assert_called_once_with(envs={'env1': 'val1'})
post_app_yaml_mock.assert_called_once_with('app1', load_yaml_mock.return_value)
run_build_hooks_mock.assert_called_once_with(load_yaml_mock.return_value,
envs={'env1': 'val1'})
def test_main_run_action(self, client_mock, tasks_mock):
register_mock = client_mock.return_value.register_unit
register_mock.return_value = ({'env1': 'val1'}, "0.16.0")
save_apprc_mock = tasks_mock.save_apprc_file
exec_script_mock = tasks_mock.execute_start_script
run_restart_hooks_mock = tasks_mock.run_restart_hooks
write_circus_conf_mock = tasks_mock.write_circus_conf
load_yaml_mock = tasks_mock.load_app_yaml
load_yaml_mock.return_value = {'hooks': {'build': ['cmd_1', 'cmd_2']}}
main()
call_count = len(client_mock.mock_calls) + len(tasks_mock.mock_calls)
self.assertEqual(call_count, 8)
write_circus_conf_mock.assert_called_once_with(envs={'env1': 'val1'})
client_mock.assert_called_once_with('http://localhost', 'token')
register_mock.assert_called_once_with('app1')
save_apprc_mock.assert_called_once_with(register_mock.return_value[0])
exec_script_mock.assert_called_once_with('mycmd', envs={'env1': 'val1'}, with_shell=False)
load_yaml_mock.assert_called_once_with()
run_restart_hooks_mock.assert_any_call('before', load_yaml_mock.return_value,
envs={'env1': 'val1'})
run_restart_hooks_mock.assert_any_call('after', load_yaml_mock.return_value,
envs={'env1': 'val1'})