package cosmostxcollector_test

import (
	"context"
	"testing"

	"github.com/stretchr/testify/mock"
	"github.com/stretchr/testify/require"

	"git.cw.tr/mukan-network/mukan-ignite/ignite/pkg/cosmosclient"
	"git.cw.tr/mukan-network/mukan-ignite/ignite/pkg/cosmostxcollector"
	"git.cw.tr/mukan-network/mukan-ignite/ignite/pkg/cosmostxcollector/mocks"
	"git.cw.tr/mukan-network/mukan-ignite/ignite/pkg/errors"
)

func TestCollector(t *testing.T) {
	// Arrange
	var (
		savedTXs [][]cosmosclient.TX

		fromHeight int64 = 1
	)

	txs := [][]cosmosclient.TX{{}, {}}

	client := mocks.NewTXsCollector(t)
	client.EXPECT().
		CollectTXs(
			mock.Anything,
			fromHeight,
			mock.AnythingOfType("chan<- []cosmosclient.TX"),
		).
		Run(func(_ context.Context, _ int64, tc chan<- []cosmosclient.TX) {
			defer close(tc)

			// Send the collected block transactions
			tc <- txs[0]
			tc <- txs[1]
		}).
		Return(nil).
		Times(1)

	db := mocks.NewSaver(t)
	db.EXPECT().
		Save(
			mock.Anything,
			mock.AnythingOfType("[]cosmosclient.TX"),
		).
		Run(func(_ context.Context, txs []cosmosclient.TX) {
			// Save the transactions
			savedTXs = append(savedTXs, txs)
		}).
		Return(nil).
		Times(2)

	c := cosmostxcollector.New(db, client)
	ctx := context.Background()

	// Act
	err := c.Collect(ctx, fromHeight)

	// Assert
	require.NoError(t, err)
	require.Equal(t, savedTXs, txs)
}

func TestCollectorWithCollectError(t *testing.T) {
	// Arrange
	wantErr := errors.New("expected error")

	client := mocks.NewTXsCollector(t)
	client.EXPECT().
		CollectTXs(
			mock.Anything,
			mock.AnythingOfType("int64"),
			mock.AnythingOfType("chan<- []cosmosclient.TX"),
		).
		Run(func(_ context.Context, _ int64, tc chan<- []cosmosclient.TX) {
			close(tc)
		}).
		Return(wantErr).
		Times(1)

	db := mocks.NewSaver(t)
	c := cosmostxcollector.New(db, client)
	ctx := context.Background()

	// Act
	err := c.Collect(ctx, 1)

	// Assert
	require.ErrorIs(t, err, wantErr)

	db.AssertNotCalled(t, "Save", mock.Anything, mock.AnythingOfType("[]cosmosclient.TX"))
}

func TestCollectorWithSaveError(t *testing.T) {
	// Arrange
	wantErr := errors.New("expected error")
	var txs []cosmosclient.TX

	client := mocks.NewTXsCollector(t)
	client.EXPECT().
		CollectTXs(
			mock.Anything,
			mock.AnythingOfType("int64"),
			mock.AnythingOfType("chan<- []cosmosclient.TX"),
		).
		Run(func(_ context.Context, _ int64, tc chan<- []cosmosclient.TX) {
			defer close(tc)

			// Send the collected block transactions
			tc <- txs
		}).
		Return(nil).
		Times(1)

	db := mocks.NewSaver(t)
	db.EXPECT().
		Save(
			mock.Anything,
			mock.AnythingOfType("[]cosmosclient.TX"),
		).
		Return(wantErr).
		Times(1)

	c := cosmostxcollector.New(db, client)
	ctx := context.Background()

	// Act
	err := c.Collect(ctx, 1)

	// Assert
	require.ErrorIs(t, err, wantErr)
}