sentinel/tests/sentry/runner/commands/test_config.py

from collections.abc import Generator

import pytest

from sentry import options
from sentry.options.manager import FLAG_AUTOMATOR_MODIFIABLE, FLAG_IMMUTABLE, UpdateChannel
from sentry.runner.commands.config import config
from sentry.testutils.cases import CliTestCase

MSG = "Option: %s. Set: %r. Set in settings: %r. Last channel: %s. Value: %s"


class ConfigOptionsTest(CliTestCase):
    command = config

    @pytest.fixture(autouse=True, scope="class")
    def register_options(self) -> Generator[None, None, None]:
        options.register("readonly_option", default=10, flags=FLAG_IMMUTABLE)
        options.register("int_option", default=20, flags=FLAG_AUTOMATOR_MODIFIABLE)
        options.register("str_option", default="blabla", flags=FLAG_AUTOMATOR_MODIFIABLE)
        options.register("map_option", default={}, flags=FLAG_AUTOMATOR_MODIFIABLE)
        options.register("list_option", default=[1, 2], flags=FLAG_AUTOMATOR_MODIFIABLE)
        options.register("drifted_option", default=[], flags=FLAG_AUTOMATOR_MODIFIABLE)
        options.register("change_channel_option", default=[], flags=FLAG_AUTOMATOR_MODIFIABLE)
        options.register("to_unset_option", default=[], flags=FLAG_AUTOMATOR_MODIFIABLE)

        yield

        options.unregister("readonly_option")
        options.unregister("int_option")
        options.unregister("str_option")
        options.unregister("map_option")
        options.unregister("list_option")
        options.unregister("drifted_option")
        options.unregister("change_channel_option")
        options.unregister("to_unset_option")

    def test_dump(self) -> None:
        options.set("int_option", 30, channel=UpdateChannel.AUTOMATOR)
        options.set("str_option", "blabla2", channel=UpdateChannel.CLI)

        rv = self.invoke("dump", "-f 2048", "-s")
        assert rv.exit_code == 0, rv.output

        output = "\n".join(
            [
                MSG % ("int_option", True, False, "automator", 30),
                MSG % ("str_option", True, False, "cli", "blabla2"),
            ]
        )
        assert output in rv.output