tool_delete_relations_test.go

package main

import (
    "path/filepath"
    "testing"
)

func TestToolDeleteRelations_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"}, {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)
    }

    if err := m.DeleteRelations([]Relation{{From: "A", To: "B", RelationType: "likes"}}); err != nil {
        t.Fatalf("DeleteRelations: %v", err)
    }

    g, err := m.ReadGraph()
    if err != nil {
        t.Fatalf("ReadGraph: %v", err)
    }
    if len(g.Relations) != 0 {
        t.Fatalf("expected no relations after delete, got %#v", g.Relations)
    }
}

func TestToolDeleteRelations_MissingEntity(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"}}); err != nil {
        t.Fatalf("CreateEntities: %v", err)
    }

    if err := m.DeleteRelations([]Relation{{From: "A", To: "B", RelationType: "likes"}}); err == nil {
        t.Fatalf("expected error for missing entity")
    }
}