frostfs-node/pkg/innerring/processors/balance/handlers_test.go

package balance

import (
	"testing"
	"time"

	frostfscontract "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client/frostfs"
	balanceEvent "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/event/balance"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger/test"
	"github.com/nspcc-dev/neo-go/pkg/util"
	"github.com/stretchr/testify/require"
)

func TestProcessorCallsFrostFSContractForLockEvent(t *testing.T) {
	t.Parallel()
	as := &testAlphabetState{
		isAlphabet: true,
	}
	conv := &testPresicionConverter{}
	cl := &testFrostFSContractClient{}
	bsc := util.Uint160{100}

	processor, err := New(&Params{
		Log:           test.NewLogger(t),
		PoolSize:      2,
		FrostFSClient: cl,
		BalanceSC:     bsc,
		AlphabetState: as,
		Converter:     conv,
	})
	require.NoError(t, err, "failed to create processor")

	processor.handleLock(balanceEvent.Lock{})

	for processor.pool.Running() > 0 {
		time.Sleep(10 * time.Millisecond)
	}

	require.Equal(t, 1, cl.chequeCalls, "invalid Cheque calls")
}

func TestProcessorDoesntCallFrostFSContractIfNotAlphabet(t *testing.T) {
	t.Parallel()
	as := &testAlphabetState{}
	conv := &testPresicionConverter{}
	cl := &testFrostFSContractClient{}
	bsc := util.Uint160{100}

	processor, err := New(&Params{
		Log:           test.NewLogger(t),
		PoolSize:      2,
		FrostFSClient: cl,
		BalanceSC:     bsc,
		AlphabetState: as,
		Converter:     conv,
	})
	require.NoError(t, err, "failed to create processor")

	processor.handleLock(balanceEvent.Lock{})

	for processor.pool.Running() > 0 {
		time.Sleep(10 * time.Millisecond)
	}

	require.Equal(t, 0, cl.chequeCalls, "invalid Cheque calls")
}

type testAlphabetState struct {
	isAlphabet bool
}

func (s *testAlphabetState) IsAlphabet() bool {
	return s.isAlphabet
}

type testPresicionConverter struct{}

func (c *testPresicionConverter) ToFixed8(v int64) int64 {
	return v
}

type testFrostFSContractClient struct {
	chequeCalls int
}

func (c *testFrostFSContractClient) Cheque(p frostfscontract.ChequePrm) error {
	c.chequeCalls++
	return nil
}