Update tests for install onboarding flow

Author: PsYcGoD

tests/test_claude_sage_enforcement.py

@@ -50,11 +50,11 @@ def test_allows_npx_sage_wrapped_shell_command():
     assert result.stderr == ""
 
 
-def test_blocks_direct_file_tool():
+def test_allows_direct_file_tool():
     result = run_hook({"tool_name": "Read", "tool_input": {"file_path": "README.md"}})
 
-    assert result.returncode == 2
-    assert "Use SAGE MCP tools instead" in result.stderr
+    assert result.returncode == 0
+    assert result.stderr == ""
 
 
 def test_blocks_subagent_without_sage_instructions():

tests/test_cli_basic.py

@@ -191,8 +191,8 @@ def test_cli_first_use_auto_installs_global_agent_enforcement(self, tmp_path, mo
         assert "sage run -- <command>" in content
         assert "npx -y psycgod-sage run -- <command>" in content
         assert "start through SAGE automatically" in content
-        assert "mcp__sage__sage_read_file" in content
-        assert "mcp__sage__sage_tree" in content
+        assert "Use native file/search/edit tools normally" in content
+        assert "SAGE MCP tools are unavailable" in content
         assert "Please help me with my general book in this folder" in content
         assert "ð" not in content
         assert "â" not in content
@@ -201,8 +201,8 @@ def test_cli_first_use_auto_installs_global_agent_enforcement(self, tmp_path, mo
         assert "PreToolUse" in settings
         assert "Bash(sage run -- *)" in settings
         assert "Bash(npx -y psycgod-sage run -- *)" in settings
-        assert "Read(*)" in settings
-        assert "mcpServers" in settings
+        assert "Read(*)" not in settings
+        assert "mcpServers" not in settings
         parsed_settings = json.loads(settings)
         hook_config = parsed_settings["hooks"]["PreToolUse"][0]["hooks"][0]
         assert hook_config["type"] == "command"
@@ -229,17 +229,21 @@ def test_project_install_adds_local_claude_and_agents_files(self, tmp_path):
 
         claude = (tmp_path / "CLAUDE.md").read_text(encoding="utf-8")
         assert "SAGE Integration - MANDATORY" in claude
-        assert "mcp__sage__sage_edit_file" in claude
+        assert "Use native file/search/edit tools normally" in claude
         assert "start through SAGE automatically" in claude
 
     def test_cli_install_command_runs_system_enforcement(self, tmp_path, monkeypatch):
-        from sage import install
-        from sage.cli import install_command
+        from sage import cli
 
-        monkeypatch.setattr(install.Path, "home", staticmethod(lambda: tmp_path))
-        assert install_command() == 0
-        assert (tmp_path / ".claude" / "CLAUDE.md").exists()
-        assert (tmp_path / ".codex" / "AGENTS.md").exists()
+        calls: list[tuple[bool, bool]] = []
+
+        def fake_activate_command(*, force: bool = False, project: bool = True) -> int:
+            calls.append((force, project))
+            return 0
+
+        monkeypatch.setattr(cli, "activate_command", fake_activate_command)
+        assert cli.install_command(force=True, project=False, wait=False) == 0
+        assert calls == [(True, False)]
 
     def test_auto_enforcement_warning_does_not_block_commands(self, monkeypatch):
         from sage import install

tests/test_public_release_docs.py

@@ -31,10 +31,11 @@ def test_readme_public_positioning():
 
     assert "# SAGE" in readme
     assert "pip install psycgod-sage" in readme
-    assert "## Start Here: Install, Activate, Then Use Any AI Agent" in readme
+    assert "## Start Here: Install SAGE, Then Use Any AI Agent" in readme
+    assert "sage install" in readme
     assert "sage doctor --activation" in readme
     assert "npx -y psycgod-sage doctor --activation" in readme
-    assert "After activation, restart any open AI-agent sessions" in readme
+    assert "After install, restart any open AI-agent sessions" in readme
     assert "Please help me with my general book in this folder" in readme
     assert "Raw logs" in readme or "raw logs" in readme.lower()
     assert "## Known Limitations" in readme