91 lines
2 KiB
Go
91 lines
2 KiB
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, true),
|
||
|
|
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, true),
|
||
|
|
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
|
||
|
|
}
|