import pytest
from pathlib import Path
import os
import sys

# Add project root to sys.path
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

from src.mcp_client import ts_cpp_get_skeleton, ts_cpp_get_code_outline

def test_ts_cpp_get_skeleton(tmp_path):
 cpp_code = """#include <iostream>
#include <vector>

template<typename T>
class Box {
public:
 Box(T val) : value(val) {}
 T getValue() {
  return value;
 }
private:
 T value;
};

void globalFunc() {
 std::cout << "Global" << std::endl;
}
"""
 cpp_file = tmp_path / "test.cpp"
 cpp_file.write_text(cpp_code)
 
 from src import mcp_client
 original_resolve = mcp_client._resolve_and_check
 mcp_client._resolve_and_check = lambda path: (Path(path), None)
 
 try:
  skeleton = ts_cpp_get_skeleton(str(cpp_file))
  assert "class Box" in skeleton
  assert "Box(T val) ..." in skeleton
  assert "T getValue() ..." in skeleton
  assert "void globalFunc() ..." in skeleton
  assert "std::cout" not in skeleton
 finally:
  mcp_client._resolve_and_check = original_resolve

def test_ts_cpp_get_code_outline(tmp_path):
 cpp_code = """
class MyClass {
 void method1() {
 }
};

template<typename T>
void templateFunc(T t) {
}
"""
 cpp_file = tmp_path / "test.cpp"
 cpp_file.write_text(cpp_code)
 
 from src import mcp_client
 original_resolve = mcp_client._resolve_and_check
 mcp_client._resolve_and_check = lambda path: (Path(path), None)
 
 try:
  outline = ts_cpp_get_code_outline(str(cpp_file))
  assert "[Class] MyClass (Lines 2-5)" in outline
  assert "[Method] method1 (Lines 3-4)" in outline
  assert "[Func] templateFunc (Lines 8-9)" in outline
 finally:
  mcp_client._resolve_and_check = original_resolve