package store_test

import (
	"testing"

	storepkg "github.com/cirruslabs/orchard/internal/controller/store"
	"github.com/cirruslabs/orchard/internal/controller/store/badger"
	"github.com/cirruslabs/orchard/pkg/resource/v1"
	"github.com/stretchr/testify/require"
	"go.uber.org/zap"
)

func TestListEventsPage(t *testing.T) {
	logger := zap.NewNop().Sugar()
	store, err := badger.NewBadgerStore(t.TempDir(), true, logger)
	require.NoError(t, err)

	events := []v1.Event{
		{Kind: v1.EventKindLogLine, Timestamp: 1, Payload: "one"},
		{Kind: v1.EventKindLogLine, Timestamp: 2, Payload: "two"},
		{Kind: v1.EventKindLogLine, Timestamp: 3, Payload: "three"},
		{Kind: v1.EventKindLogLine, Timestamp: 4, Payload: "four"},
	}

	err = store.Update(func(txn storepkg.Transaction) error {
		return txn.AppendEvents(events, "vms", "vm-uid")
	})
	require.NoError(t, err)

	var page storepkg.Page[v1.Event]
	err = store.View(func(txn storepkg.Transaction) error {
		page, err = txn.ListEventsPage(storepkg.ListOptions{
			Limit: 2,
		}, "vms", "vm-uid")
		return err
	})
	require.NoError(t, err)
	require.Equal(t, events[:2], page.Items)
	require.NotEmpty(t, page.NextCursor)

	var page2 storepkg.Page[v1.Event]
	err = store.View(func(txn storepkg.Transaction) error {
		page2, err = txn.ListEventsPage(storepkg.ListOptions{
			Limit:  2,
			Cursor: page.NextCursor,
		}, "vms", "vm-uid")
		return err
	})
	require.NoError(t, err)
	require.Equal(t, events[2:], page2.Items)
	require.Empty(t, page2.NextCursor)

	var descPage storepkg.Page[v1.Event]
	err = store.View(func(txn storepkg.Transaction) error {
		descPage, err = txn.ListEventsPage(storepkg.ListOptions{
			Limit: 2,
			Order: storepkg.ListOrderDesc,
		}, "vms", "vm-uid")
		return err
	})
	require.NoError(t, err)
	require.Equal(t, []v1.Event{events[3], events[2]}, descPage.Items)
	require.NotEmpty(t, descPage.NextCursor)

	var descPage2 storepkg.Page[v1.Event]
	err = store.View(func(txn storepkg.Transaction) error {
		descPage2, err = txn.ListEventsPage(storepkg.ListOptions{
			Limit:  2,
			Cursor: descPage.NextCursor,
			Order:  storepkg.ListOrderDesc,
		}, "vms", "vm-uid")
		return err
	})
	require.NoError(t, err)
	require.Equal(t, []v1.Event{events[1], events[0]}, descPage2.Items)
	require.Empty(t, descPage2.NextCursor)
}