package handlers

import (
	"context"
	"testing"

	"task-dashboard/internal/models"
	"task-dashboard/internal/store"
)

type mockClaudomatorClient struct {
	stories []models.ClaudomatorStory
}

func (m *mockClaudomatorClient) GetActiveStories(ctx context.Context) ([]models.ClaudomatorStory, error) {
	return m.stories, nil
}

func TestBuildUnifiedAtomList_WithNativeTasks(t *testing.T) {
	s, err := store.New(":memory:", "../../migrations")
	if err != nil {
		t.Fatalf("failed to create in-memory store: %v", err)
	}
	defer s.Close()

	if err := s.CreateNativeTask(models.Task{ID: "native-1", Content: "Buy milk"}); err != nil {
		t.Fatalf("failed to create native task: %v", err)
	}

	atoms, _, err := BuildUnifiedAtomList(s, nil)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}

	found := false
	for _, a := range atoms {
		if a.Source == "doot" && a.Title == "Buy milk" {
			found = true
			break
		}
	}
	if !found {
		t.Errorf("expected atom with Source='doot' and Title='Buy milk'; got atoms: %+v", atoms)
	}
}

func TestBuildUnifiedAtomList_WithClaudomator(t *testing.T) {
	s, err := store.New(":memory:", "../../migrations")
	if err != nil {
		t.Fatalf("failed to create in-memory store: %v", err)
	}
	defer s.Close()

	mock := &mockClaudomatorClient{
		stories: []models.ClaudomatorStory{
			{ID: "s1", Title: "My story", Status: "IN_PROGRESS", ProjectID: "nav"},
		},
	}

	atoms, _, err := BuildUnifiedAtomList(s, mock)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}

	found := false
	for _, a := range atoms {
		if a.Source == "claudomator" && a.Title == "My story" {
			found = true
			break
		}
	}
	if !found {
		t.Error("expected atom with Source='claudomator' and Title='My story'")
	}
}