mirror/erigon-pulse
1
0
Fork 0
mirror of https://gitlab.com/pulsechaincom/erigon-pulse.git synced 2025-01-19 00:54:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

test success flow

Browse Source
This commit is contained in:
alex.sharov 2021-08-09 14:14:02 +07:00
parent 5f05c1fcbd
commit fe9a2ef810
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
txpool/fetch_test.go
View File

@ -19,6 +19,7 @@ package txpool
import ( import (
"context" "context"
"fmt" "fmt"
"io"
"sync" "sync"
"testing" "testing"
@ -33,8 +34,8 @@ import (
) )
func TestFetch(t *testing.T) { func TestFetch(t *testing.T) {
ctx, cancelFn := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancelFn() defer cancel()
var genesisHash [32]byte var genesisHash [32]byte
var networkId uint64 = 1 var networkId uint64 = 1
@ -135,13 +136,18 @@ func TestSendTxPropagate(t *testing.T) {
} }
func TestOnNewBlock(t *testing.T) { func TestOnNewBlock(t *testing.T) {
ctx, cancelFn := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancelFn() defer cancel()
var genesisHash [32]byte var genesisHash [32]byte
var networkId uint64 = 1 var networkId uint64 = 1
i := 0
stream := &remote.KV_StateChangesClientMock{ stream := &remote.KV_StateChangesClientMock{
RecvFunc: func() (*remote.StateChange, error) { RecvFunc: func() (*remote.StateChange, error) {
if i > 0 {
return nil, io.EOF
}
i++
return &remote.StateChange{Txs: [][]byte{decodeHex(txParseTests[0].payloadStr), decodeHex(txParseTests[1].payloadStr), decodeHex(txParseTests[2].payloadStr)}}, nil return &remote.StateChange{Txs: [][]byte{decodeHex(txParseTests[0].payloadStr), decodeHex(txParseTests[1].payloadStr), decodeHex(txParseTests[2].payloadStr)}}, nil
}, },
} }
@ -152,11 +158,7 @@ func TestOnNewBlock(t *testing.T) {
} }
pool := &PoolMock{} pool := &PoolMock{}
fetch := NewFetch(ctx, nil, genesisHash, networkId, nil, pool, stateChanges) fetch := NewFetch(ctx, nil, genesisHash, networkId, nil, pool, stateChanges)
var wg sync.WaitGroup fetch.stateChangesStream(ctx, stateChanges)
wg.Add(1)
fetch.SetWaitGroup(&wg)
fetch.ConnectCore()
fetch.wg.Wait()
assert.Equal(t, 1, len(pool.OnNewBlockCalls())) assert.Equal(t, 1, len(pool.OnNewBlockCalls()))
assert.Equal(t, 3, len(pool.OnNewBlockCalls()[0].MinedTxs.txs)) assert.Equal(t, 3, len(pool.OnNewBlockCalls()[0].MinedTxs.txs))
} }