spec-kit/tests/test_merge.py

import stat

import pytest

from specify_cli import merge_json_files
from specify_cli import handle_vscode_settings

# --- Dimension 2: Polite Deep Merge Strategy ---

def test_merge_json_files_type_mismatch_preservation(tmp_path):
    """If user has a string but template wants a dict, PRESERVE user's string."""
    existing_file = tmp_path / "settings.json"
    # User might have overridden a setting with a simple string or different type
    existing_file.write_text('{"chat.editor.fontFamily": "CustomFont"}')

    # Template might expect a dict for the same key (hypothetically)
    new_settings = {
        "chat.editor.fontFamily": {"font": "TemplateFont"}
    }

    merged = merge_json_files(existing_file, new_settings)
    # Result is None because user settings were preserved and nothing else changed
    assert merged is None

def test_merge_json_files_deep_nesting(tmp_path):
    """Verify deep recursive merging of new keys."""
    existing_file = tmp_path / "settings.json"
    existing_file.write_text("""
    {
        "a": {
            "b": {
                "c": 1
            }
        }
    }
    """)

    new_settings = {
        "a": {
            "b": {
                "d": 2  # New nested key
            },
            "e": 3      # New mid-level key
        }
    }

    merged = merge_json_files(existing_file, new_settings)
    assert merged["a"]["b"]["c"] == 1
    assert merged["a"]["b"]["d"] == 2
    assert merged["a"]["e"] == 3

def test_merge_json_files_empty_existing(tmp_path):
    """Merging into an empty/new file."""
    existing_file = tmp_path / "empty.json"
    existing_file.write_text("{}")

    new_settings = {"a": 1}
    merged = merge_json_files(existing_file, new_settings)
    assert merged == {"a": 1}

# --- Dimension 3: Real-world Simulation ---

def test_merge_vscode_realistic_scenario(tmp_path):
    """A realistic VSCode settings.json with many existing preferences, comments, and trailing commas."""
    existing_file = tmp_path / "vscode_settings.json"
    existing_file.write_text("""
    {
        "editor.fontSize": 12,
        "editor.formatOnSave": true, /* block comment */
        "files.exclude": {
            "**/.git": true,
            "**/node_modules": true,
        },
        "chat.promptFilesRecommendations": {
            "existing.tool": true,
        } // User comment
    }
    """)

    template_settings = {
        "chat.promptFilesRecommendations": {
            "speckit.specify": True,
            "speckit.plan": True
        },
        "chat.tools.terminal.autoApprove": {
            ".specify/scripts/bash/": True
        }
    }

    merged = merge_json_files(existing_file, template_settings)

    # Check preservation
    assert merged["editor.fontSize"] == 12
    assert merged["files.exclude"]["**/.git"] is True
    assert merged["chat.promptFilesRecommendations"]["existing.tool"] is True

    # Check additions
    assert merged["chat.promptFilesRecommendations"]["speckit.specify"] is True
    assert merged["chat.tools.terminal.autoApprove"][".specify/scripts/bash/"] is True

# --- Dimension 4: Error Handling & Robustness ---

def test_merge_json_files_with_bom(tmp_path):
    """Test files with UTF-8 BOM (sometimes created on Windows)."""
    existing_file = tmp_path / "bom.json"
    content = '{"a": 1}'
    # Prepend UTF-8 BOM
    existing_file.write_bytes(b'\xef\xbb\xbf' + content.encode('utf-8'))

    new_settings = {"b": 2}
    merged = merge_json_files(existing_file, new_settings)
    assert merged == {"a": 1, "b": 2}

def test_merge_json_files_not_a_dictionary_template(tmp_path):
    """If for some reason new_content is not a dict, PRESERVE existing settings by returning None."""
    existing_file = tmp_path / "ok.json"
    existing_file.write_text('{"a": 1}')

    # Secure fallback: return None to skip writing and avoid clobbering
    assert merge_json_files(existing_file, ["not", "a", "dict"]) is None

def test_merge_json_files_unparseable_existing(tmp_path):
    """If the existing file is unparseable JSON, return None to avoid overwriting it."""
    bad_file = tmp_path / "bad.json"
    bad_file.write_text('{"a": 1, missing_value}') # Invalid JSON

    assert merge_json_files(bad_file, {"b": 2}) is None


def test_merge_json_files_list_preservation(tmp_path):
    """Verify that existing list values are preserved and NOT merged or overwritten."""
    existing_file = tmp_path / "list.json"
    existing_file.write_text('{"my.list": ["user_item"]}')

    template_settings = {
        "my.list": ["template_item"]
    }

    merged = merge_json_files(existing_file, template_settings)
    # The polite merge policy says: keep existing values if they exist and aren't both dicts.
    # Since nothing changed, it returns None.
    assert merged is None

def test_merge_json_files_no_changes(tmp_path):
    """If the merge doesn't introduce any new keys or changes, return None to skip rewrite."""
    existing_file = tmp_path / "no_change.json"
    existing_file.write_text('{"a": 1, "b": {"c": 2}}')

    template_settings = {
        "a": 1,          # Already exists
        "b": {"c": 2}    # Already exists nested
    }

    # Should return None because result == existing
    assert merge_json_files(existing_file, template_settings) is None

def test_merge_json_files_type_mismatch_no_op(tmp_path):
    """If a key exists with different type and we preserve it, it might still result in no change."""
    existing_file = tmp_path / "mismatch_no_op.json"
    existing_file.write_text('{"a": "user_string"}')

    template_settings = {
        "a": {"key": "template_dict"} # Mismatch, will be ignored
    }

    # Should return None because we preserved the user's string and nothing else changed
    assert merge_json_files(existing_file, template_settings) is None


def test_handle_vscode_settings_preserves_mode_on_atomic_write(tmp_path):
    """Atomic rewrite should preserve existing file mode bits."""
    vscode_dir = tmp_path / ".vscode"
    vscode_dir.mkdir()
    dest_file = vscode_dir / "settings.json"
    template_file = tmp_path / "template_settings.json"

    dest_file.write_text('{"a": 1}\n', encoding="utf-8")
    dest_file.chmod(0o640)
    before_mode = stat.S_IMODE(dest_file.stat().st_mode)

    template_file.write_text('{"b": 2}\n', encoding="utf-8")

    handle_vscode_settings(
        template_file,
        dest_file,
        "settings.json",
        verbose=False,
        tracker=None,
    )

    after_mode = stat.S_IMODE(dest_file.stat().st_mode)
    assert after_mode == before_mode


def test_handle_vscode_settings_propagates_programming_errors(tmp_path):
    """Unexpected programming errors (TypeError) must propagate, not be silently swallowed."""
    vscode_dir = tmp_path / ".vscode"
    vscode_dir.mkdir()
    dest_file = vscode_dir / "settings.json"
    dest_file.write_text('{"a": 1}\n', encoding="utf-8")
    template_file = tmp_path / "template_settings.json"
    template_file.write_text('{"b": 2}\n', encoding="utf-8")

    import specify_cli._utils as utils_mod
    original_merge = utils_mod.merge_json_files
    utils_mod.merge_json_files = lambda *a, **kw: (_ for _ in ()).throw(TypeError("boom"))
    try:
        with pytest.raises(TypeError):
            handle_vscode_settings(
                template_file, dest_file, "settings.json",
                verbose=False, tracker=None,
            )
    finally:
        utils_mod.merge_json_files = original_merge