DDNS/tests/test_config_init_multi.py

# coding=utf-8
# type: ignore[index,operator,assignment]
"""
Unit tests for multi-config functionality
@author: GitHub Copilot
"""
from __init__ import unittest, patch
import tempfile
import json
import os
import sys
from ddns.config import load_configs
from ddns.config.file import load_config as load_file_config


class TestMultiConfig(unittest.TestCase):

    def setUp(self):
        self.temp_dir = tempfile.mkdtemp()
        self.original_argv = sys.argv[:]

    def tearDown(self):
        sys.argv = self.original_argv
        # Clean up temp directory
        import shutil

        shutil.rmtree(self.temp_dir, ignore_errors=True)

    def test_file_loader_single_object(self):
        """Test that file loader works with single object configs"""
        config_data = {"dns": "cloudflare", "id": "[email protected]", "token": "secret123"}

        config_path = os.path.join(self.temp_dir, "single_config.json")
        with open(config_path, "w") as f:
            json.dump(config_data, f)

        result = load_file_config(config_path)
        self.assertIsInstance(result, dict)
        self.assertEqual(result["dns"], "cloudflare")
        self.assertEqual(result["id"], "[email protected]")
        self.assertEqual(result["token"], "secret123")

    def test_cli_multiple_configs(self):
        """Test CLI support for multiple config files"""
        # Create two config files
        config1_data = {"dns": "cloudflare", "id": "[email protected]", "token": "secret1"}
        config2_data = {"dns": "dnspod", "id": "[email protected]", "token": "secret2"}

        config1_path = os.path.join(self.temp_dir, "config1.json")
        config2_path = os.path.join(self.temp_dir, "config2.json")

        with open(config1_path, "w") as f:
            json.dump(config1_data, f)
        with open(config2_path, "w") as f:
            json.dump(config2_data, f)

        # Mock CLI args
        sys.argv = ["ddns", "-c", config1_path, "-c", config2_path]

        with patch("ddns.config.cli.load_config") as mock_cli:
            mock_cli.return_value = {"config": [config1_path, config2_path]}

            with patch("ddns.config.env.load_config") as mock_env:
                mock_env.return_value = {}

                configs = load_configs("test", "1.0", "2023-01-01")

                self.assertEqual(len(configs), 2)
                self.assertEqual(configs[0].dns, "cloudflare")
                self.assertEqual(configs[1].dns, "dnspod")

    def test_env_multiple_configs_integration(self):
        """Test environment variable support for multiple config files using real CLI/env loading"""
        # Create two config files
        config1_data = {"dns": "debug", "id": "[email protected]", "token": "secret1"}
        config2_data = {"dns": "debug", "id": "[email protected]", "token": "secret2"}

        config1_path = os.path.join(self.temp_dir, "config1.json")
        config2_path = os.path.join(self.temp_dir, "config2.json")

        with open(config1_path, "w") as f:
            json.dump(config1_data, f)
        with open(config2_path, "w") as f:
            json.dump(config2_data, f)

        # Test via environment variable
        old_env = os.environ.get("DDNS_CONFIG")
        try:
            os.environ["DDNS_CONFIG"] = "{},{}".format(config1_path, config2_path)
            sys.argv = ["ddns"]

            configs = load_configs("test", "1.0", "2023-01-01")

            self.assertEqual(len(configs), 2)
            self.assertEqual(configs[0].dns, "debug")
            self.assertEqual(configs[1].dns, "debug")
            self.assertEqual(configs[0].id, "[email protected]")
            self.assertEqual(configs[1].id, "[email protected]")
        finally:
            if old_env is not None:
                os.environ["DDNS_CONFIG"] = old_env
            elif "DDNS_CONFIG" in os.environ:
                del os.environ["DDNS_CONFIG"]


if __name__ == "__main__":
    unittest.main()