Exemplo n.º 1
0
        def setup_and_execute_app(self, argv, configuration):
            with self.a_temp_file() as config_location:
                with self.modified_env(HARPOON_CONFIG=config_location):
                    app = App()
                    args_obj, args_dict, extra_args = app.make_cli_parser().interpret_args(argv, app.cli_categories)

                    collector = mock.Mock(name="collector")
                    collector.configuration = configuration
                    FakeCollector = mock.Mock(name="Collector", return_value=collector)
                    docker_context = mock.Mock(name="docker_context")
                    logging_handler = mock.Mock(name="logging_handler")
                    setup_logging_theme = mock.Mock(name="setup_logging_theme")
                    docker_context_maker = mock.Mock(name="docker_context_maker", return_value=docker_context)

                    with mock.patch("harpoon.executor.Collector", FakeCollector):
                        with mock.patch("harpoon.executor.docker_context", docker_context_maker):
                            with mock.patch.object(app, "setup_logging_theme", setup_logging_theme):
                                yield collector, docker_context_maker, docker_context, app, setup_logging_theme, args_dict
                                app.execute(args_obj, args_dict, extra_args, logging_handler)

            FakeCollector.assert_called_once()
Exemplo n.º 2
0
from input_algorithms.spec_base import NotSpecified
from contextlib import contextmanager
from tests.helpers import HarpoonCase
from six import StringIO
import mock
import six
import sys
import os

describe HarpoonCase, "App":
    describe "Cli parsing":
        it "replaces task and image with positional arguments":
            with self.a_temp_file() as config_location:
                argv = ["list_tasks", "blah", "--harpoon-config", config_location]
                app = App()
                args_obj, args_dict, extra_args = app.make_cli_parser().interpret_args(argv, app.cli_categories)
                self.assertEqual(args_obj.harpoon_chosen_task, "list_tasks")
                self.assertEqual(args_obj.harpoon_chosen_image, "blah")
                self.assertEqual(args_dict["harpoon"]["chosen_task"], "list_tasks")
                self.assertEqual(args_dict["harpoon"]["chosen_image"], "blah")

        it "takes in HARPOON_CONFIG as the configuration file":
            with self.a_temp_file() as config_location:
                with self.modified_env(HARPOON_CONFIG=config_location):
                    argv = ["list_tasks", "blah"]
                    app = App()
                    args_obj, args_dict, extra_args = app.make_cli_parser().interpret_args(argv, app.cli_categories)
                    self.assertEqual(args_obj.harpoon_config().name, config_location)
                    self.assertEqual(args_dict["harpoon"]["config"]().name, config_location)