package tools

import (
	"context"
	"encoding/json"
	"testing"

	"github.com/bnema/zut/packages/core"
)

func TestUpdateGoalToolReturnsPersistableStatus(t *testing.T) {
	tool := &UpdateGoalTool{}
	raw, err := json.Marshal(map[string]string{"status": "complete"})
	if err != nil {
		t.Fatal(err)
	}
	result, err := tool.Execute(context.Background(), raw, nil)
	if err != nil {
		t.Fatal(err)
	}
	if result.IsError {
		t.Fatalf("result = %#v, want success", result)
	}
	update, ok := GoalUpdateFromResult(result)
	if !ok {
		t.Fatalf("details = %#v, want goal update", result.Details)
	}
	if update.Status != core.GoalDone || update.Reason != "" {
		t.Fatalf("update = %#v", update)
	}
}

func TestUpdateGoalToolReturnsPersistableManagerGoal(t *testing.T) {
	tool := &UpdateGoalTool{}
	result, err := tool.Execute(context.Background(), json.RawMessage(`{"status":"active","objective":"reproduce the reported failure"}`), nil)
	if err != nil {
		t.Fatal(err)
	}
	update, ok := GoalUpdateFromResult(result)
	if !ok {
		t.Fatalf("details = %#v, want goal update", result.Details)
	}
	if update.Status != core.GoalActive || update.Objective != "reproduce the reported failure" {
		t.Fatalf("update = %#v", update)
	}
}

func TestUpdateGoalToolReturnsPersistableSupersedingGoal(t *testing.T) {
	tool := &UpdateGoalTool{}
	result, err := tool.Execute(context.Background(), json.RawMessage(`{"status":"superseded","objective":"use the supported API","goal_id":"goal-1","mission_id":"mission-1"}`), nil)
	if err != nil {
		t.Fatal(err)
	}
	update, ok := GoalUpdateFromResult(result)
	if !ok {
		t.Fatalf("details = %#v, want superseding goal update", result.Details)
	}
	if update.Status != core.GoalSuperseded || update.Objective != "use the supported API" || update.GoalID != "goal-1" || update.MissionID != "mission-1" {
		t.Fatalf("update = %#v", update)
	}
}

func TestUpdateGoalToolRejectsUnknownFields(t *testing.T) {
	tool := &UpdateGoalTool{}
	result, err := tool.Execute(context.Background(), json.RawMessage(`{"status":"complete","unexpected":true}`), nil)
	if err != nil {
		t.Fatal(err)
	}
	if !result.IsError {
		t.Fatalf("result = %#v, want tool error", result)
	}
}

func TestGoalUpdateFromResultRejectsBlockedWithoutReason(t *testing.T) {
	result := core.ToolResult{Details: GoalUpdate{Status: core.GoalBlocked}}
	if _, ok := GoalUpdateFromResult(result); ok {
		t.Fatal("blank-reason blocked update was accepted")
	}

	result.Details = GoalUpdate{Status: core.GoalBlocked, Reason: "needs user input"}
	if _, ok := GoalUpdateFromResult(result); !ok {
		t.Fatal("blocked update with reason was rejected")
	}

	result.Details = GoalUpdate{Status: core.GoalDone}
	if _, ok := GoalUpdateFromResult(result); !ok {
		t.Fatal("completed update without reason was rejected")
	}
}

func TestUpdateGoalToolRejectsSupersededWithoutReplacement(t *testing.T) {
	tool := &UpdateGoalTool{}
	result, err := tool.Execute(context.Background(), json.RawMessage(`{"status":"superseded"}`), nil)
	if err != nil {
		t.Fatal(err)
	}
	if !result.IsError {
		t.Fatalf("result = %#v, want tool error", result)
	}
}

func TestUpdateGoalToolRejectsUnsupportedStatus(t *testing.T) {
	tool := &UpdateGoalTool{}
	raw := json.RawMessage(`{"status":"active"}`)
	result, err := tool.Execute(context.Background(), raw, nil)
	if err != nil {
		t.Fatal(err)
	}
	if !result.IsError {
		t.Fatalf("result = %#v, want tool error", result)
	}
	if _, ok := GoalUpdateFromResult(result); ok {
		t.Fatal("invalid result exposed a goal update")
	}
}