package integration

import (
	"testing"

	"github.com/marte-community/marte-dev-tools/internal/index"
)

func TestNodeMap(t *testing.T) {
	pt := index.NewProjectTree()
	root := pt.Root

	// Create structure: +A -> +B -> +C
	nodeA := &index.ProjectNode{Name: "A", RealName: "+A", Children: make(map[string]*index.ProjectNode), Parent: root}
	root.Children["A"] = nodeA

	nodeB := &index.ProjectNode{Name: "B", RealName: "+B", Children: make(map[string]*index.ProjectNode), Parent: nodeA}
	nodeA.Children["B"] = nodeB

	nodeC := &index.ProjectNode{Name: "C", RealName: "+C", Children: make(map[string]*index.ProjectNode), Parent: nodeB}
	nodeB.Children["C"] = nodeC

	// Rebuild Index
	pt.RebuildIndex()

	// Find by Name
	found := pt.FindNode(root, "C", nil)
	if found != nodeC {
		t.Errorf("FindNode(C) failed. Got %v, want %v", found, nodeC)
	}

	// Find by RealName
	found = pt.FindNode(root, "+C", nil)
	if found != nodeC {
		t.Errorf("FindNode(+C) failed. Got %v, want %v", found, nodeC)
	}

	// Find by Path
	found = pt.FindNode(root, "A.B.C", nil)
	if found != nodeC {
		t.Errorf("FindNode(A.B.C) failed. Got %v, want %v", found, nodeC)
	}

	// Find by Path with RealName
	found = pt.FindNode(root, "+A.+B.+C", nil)
	if found != nodeC {
		t.Errorf("FindNode(+A.+B.+C) failed. Got %v, want %v", found, nodeC)
	}
}

func TestResolveReferencesWithMap(t *testing.T) {
	pt := index.NewProjectTree()
	root := pt.Root

	nodeA := &index.ProjectNode{Name: "A", RealName: "+A", Children: make(map[string]*index.ProjectNode), Parent: root}
	root.Children["A"] = nodeA

	ref := index.Reference{Name: "A", File: "test.marte"}
	pt.References = append(pt.References, ref)

	pt.ResolveReferences()

	if pt.References[0].Target != nodeA {
		t.Error("ResolveReferences failed to resolve A")
	}
}