コード例 #1
0
ファイル: test_main.py プロジェクト: bhyvex/tsuru-unit-agent
 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])
コード例 #2
0
ファイル: test_main.py プロジェクト: bhyvex/tsuru-unit-agent
    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'})
コード例 #3
0
ファイル: test_main.py プロジェクト: bhyvex/tsuru-unit-agent
 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])
コード例 #4
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'})
コード例 #5
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'})
コード例 #6
0
ファイル: test_main.py プロジェクト: bhyvex/tsuru-unit-agent
 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'})