tool_read_graph_test.go

package main

import (
    "path/filepath"
    "testing"
)

func TestToolReadGraph_Basic(t *testing.T) {
    dir := t.TempDir()
    p := filepath.Join(dir, "memory.db")
    bolt := p + ".bolt"
    m := NewGraphManager(p, bolt)

    if _, err := m.CreateEntities([]Entity{{Name: "A", EntityType: "t", Observations: []string{"o1"}}, {Name: "B", EntityType: "t"}}); err != nil {
        t.Fatalf("CreateEntities: %v", err)
    }
    if _, err := m.CreateRelations([]Relation{{From: "A", To: "B", RelationType: "likes"}}); err != nil {
        t.Fatalf("CreateRelations: %v", err)
    }

    g, err := m.ReadGraph()
    if err != nil {
        t.Fatalf("ReadGraph: %v", err)
    }
    if len(g.Entities) != 2 {
        t.Fatalf("want 2 entities, got %d", len(g.Entities))
    }
    if len(g.Relations) != 1 {
        t.Fatalf("want 1 relation, got %d", len(g.Relations))
    }
}