lib/cli/tests/test_settings_command.py - main - githubmirror/crewai

lib/cli/tests/test_settings_command.py
88 строк3 KB
Greyson LaLonde
chore(cli): drop self-explanatory comments
26 май 2026, 11:05
26 май 2026, 11:05bad64b1
import tempfile
import unittest
from pathlib import Path
from unittest.mock import patch, MagicMock, call

from crewai_cli.settings.main import SettingsCommand
from crewai_cli.config import (
    Settings,
    USER_SETTINGS_KEYS,
    CLI_SETTINGS_KEYS,
    DEFAULT_CLI_SETTINGS,
    HIDDEN_SETTINGS_KEYS,
    READONLY_SETTINGS_KEYS,
)
import shutil


class TestSettingsCommand(unittest.TestCase):
    def setUp(self):
        self.test_dir = Path(tempfile.mkdtemp())
        self.config_path = self.test_dir / "settings.json"
        self.settings = Settings(config_path=self.config_path)
        self.settings_command = SettingsCommand(
            settings_kwargs={"config_path": self.config_path}
        )

    def tearDown(self):
        shutil.rmtree(self.test_dir)

    @patch("crewai_cli.settings.main.console")
    @patch("crewai_cli.settings.main.Table")
    def test_list_settings(self, mock_table_class, mock_console):
        mock_table_instance = MagicMock()
        mock_table_class.return_value = mock_table_instance

        self.settings_command.list()

        mock_table_instance.add_row.assert_has_calls(
            [
                call(
                    field_name,
                    getattr(self.settings, field_name) or "Not set",
                    field_info.description,
                )
                for field_name, field_info in Settings.model_fields.items()
                if field_name not in HIDDEN_SETTINGS_KEYS
            ]
        )

        mock_console.print.assert_called_once_with(mock_table_instance)

    def test_set_valid_keys(self):
        valid_keys = Settings.model_fields.keys() - (
            READONLY_SETTINGS_KEYS + HIDDEN_SETTINGS_KEYS
        )
        for key in valid_keys:
            test_value = f"some_value_for_{key}"
            self.settings_command.set(key, test_value)
            self.assertEqual(getattr(self.settings_command.settings, key), test_value)

    def test_set_invalid_key(self):
        with self.assertRaises(SystemExit):
            self.settings_command.set("invalid_key", "value")

    def test_set_readonly_keys(self):
        for key in READONLY_SETTINGS_KEYS:
            with self.assertRaises(SystemExit):
                self.settings_command.set(key, "some_readonly_key_value")

    def test_set_hidden_keys(self):
        for key in HIDDEN_SETTINGS_KEYS:
            with self.assertRaises(SystemExit):
                self.settings_command.set(key, "some_hidden_key_value")

    def test_reset_all_settings(self):
        for key in USER_SETTINGS_KEYS + CLI_SETTINGS_KEYS:
            setattr(self.settings_command.settings, key, f"custom_value_for_{key}")
        self.settings_command.settings.dump()

        self.settings_command.reset_all_settings()

        for key in USER_SETTINGS_KEYS:
            self.assertEqual(getattr(self.settings_command.settings, key), None)

        for key in CLI_SETTINGS_KEYS:
            self.assertEqual(
                getattr(self.settings_command.settings, key), DEFAULT_CLI_SETTINGS.get(key)
            )