Mukan Erkin Törük
c32551b6f7
Some checks failed
Docs Deploy / build_and_deploy (push) Has been cancelled
Generate Docs / cli (push) Has been cancelled
Generate Config Doc / cli (push) Has been cancelled
Go formatting / go-formatting (push) Has been cancelled
Check links / markdown-link-check (push) Has been cancelled
Integration / pre-test (push) Has been cancelled
Integration / test on (push) Has been cancelled
Integration / status (push) Has been cancelled
Lint / Lint Go code (push) Has been cancelled
Test / test (ubuntu-latest) (push) Has been cancelled
135 lines
2.8 KiB
Go
135 lines
2.8 KiB
Go
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)
|
|
}
|