package run_test

import (
	"context"
	"errors"
	"regexp"
	"testing"
	"time"

	"github.com/DATA-DOG/go-sqlmock"
	"github.com/stretchr/testify/require"

	"github.com/tidbcloud/AgentTeam/internal/coredb"
	"github.com/tidbcloud/AgentTeam/internal/store/workflow/run"
)

func TestListRunsQueriesStartedTimeDescendingWithStableCursor(t *testing.T) {
	sqlDB, mock, err := sqlmock.New()
	require.NoError(t, err)
	t.Cleanup(func() { _ = sqlDB.Close() })
	store := run.New(coredb.WrapSQLDB(sqlDB))
	cursorTime := time.Date(2026, 7, 31, 12, 0, 0, 0, time.UTC)
	wantErr := errors.New("query observed")

	mock.ExpectQuery(regexp.QuoteMeta(`WHERE r.workspace_id=? AND r.status=? AND (r.started_at<? OR r.started_at IS NULL OR (r.started_at=? AND r.id<?))
ORDER BY r.started_at DESC, r.id DESC LIMIT ?`)).
		WithArgs("wsp_1", run.RunStatusSucceeded, cursorTime, cursorTime, "wri_20", 21).
		WillReturnError(wantErr)

	_, err = store.ListRuns(context.Background(), run.ListRunsParams{
		WorkspaceID: "wsp_1", Status: run.RunStatusSucceeded,
		BeforeStartedAt: &cursorTime, BeforeID: "wri_20", Limit: 21,
	})
	require.ErrorIs(t, err, wantErr)
	require.NoError(t, mock.ExpectationsWereMet())
}

func TestListRunsContinuesWithinUnstartedRunsByDescendingID(t *testing.T) {
	sqlDB, mock, err := sqlmock.New()
	require.NoError(t, err)
	t.Cleanup(func() { _ = sqlDB.Close() })
	store := run.New(coredb.WrapSQLDB(sqlDB))
	wantErr := errors.New("query observed")

	mock.ExpectQuery(regexp.QuoteMeta(`WHERE r.workspace_id=? AND r.started_at IS NULL AND r.id<?
ORDER BY r.started_at DESC, r.id DESC LIMIT ?`)).
		WithArgs("wsp_1", "wri_05", 20).
		WillReturnError(wantErr)

	_, err = store.ListRuns(context.Background(), run.ListRunsParams{
		WorkspaceID: "wsp_1", BeforeID: "wri_05", Limit: 20,
	})
	require.ErrorIs(t, err, wantErr)
	require.NoError(t, mock.ExpectationsWereMet())
}

func TestListRunsFiltersByPlanID(t *testing.T) {
	sqlDB, mock, err := sqlmock.New()
	require.NoError(t, err)
	t.Cleanup(func() { _ = sqlDB.Close() })
	store := run.New(coredb.WrapSQLDB(sqlDB))
	wantErr := errors.New("query observed")

	mock.ExpectQuery(regexp.QuoteMeta(`WHERE r.workspace_id=? AND r.plan_id=? AND r.status=?
ORDER BY r.started_at DESC, r.id DESC LIMIT ?`)).
		WithArgs("wsp_1", "pln_1", run.RunStatusRunning, 20).
		WillReturnError(wantErr)

	_, err = store.ListRuns(context.Background(), run.ListRunsParams{
		WorkspaceID: "wsp_1", PlanID: " pln_1 ", Status: run.RunStatusRunning, Limit: 20,
	})
	require.ErrorIs(t, err, wantErr)
	require.NoError(t, mock.ExpectationsWereMet())
}

func TestListAwaitingCallbackEffectsFiltersResponderBeforePayloadDecode(t *testing.T) {
	sqlDB, mock, err := sqlmock.New()
	require.NoError(t, err)
	t.Cleanup(func() { _ = sqlDB.Close() })
	store := run.New(coredb.WrapSQLDB(sqlDB))
	wantErr := errors.New("query observed")

	mock.ExpectQuery(`(?s)FROM workflow_run_effects\s+WHERE workspace_id=\?\s+AND callback_responder_agent_id=\?\s+AND status='awaiting'\s+AND id>\?\s+ORDER BY id ASC LIMIT \?`).
		WithArgs("wsp_1", "agt_1", "wfx_after", 51).
		WillReturnError(wantErr)

	_, err = store.ListAwaitingCallbackEffects(context.Background(), run.ListAwaitingCallbackEffectsParams{
		WorkspaceID: " wsp_1 ", AgentID: " agt_1 ", AfterID: " wfx_after ", Limit: 51,
	})
	require.ErrorIs(t, err, wantErr)
	require.NoError(t, mock.ExpectationsWereMet())
}