mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
d077483577
* v3 import renamings * tidy * fmt * rev * Update beacon-chain/core/epoch/precompute/reward_penalty_test.go * Update beacon-chain/core/helpers/validators_test.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/alias.go * Update beacon-chain/db/iface/BUILD.bazel * Update beacon-chain/db/kv/kv.go * Update beacon-chain/db/kv/state.go * Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go * Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go * Update beacon-chain/sync/initial-sync/service.go * fix deps * fix bad replacements * fix bad replacements * change back * gohashtree version * fix deps Co-authored-by: Nishant Das <nishdas93@gmail.com> Co-authored-by: Potuz <potuz@prysmaticlabs.com>
109 lines
2.2 KiB
Go
109 lines
2.2 KiB
Go
package async_test
|
|
|
|
import (
|
|
"errors"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/v3/async"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/require"
|
|
)
|
|
|
|
func TestDouble(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
inValues int
|
|
err error
|
|
}{
|
|
{
|
|
name: "0",
|
|
inValues: 0,
|
|
err: errors.New("input length must be greater than 0"),
|
|
},
|
|
{
|
|
name: "1",
|
|
inValues: 1,
|
|
},
|
|
{
|
|
name: "1023",
|
|
inValues: 1023,
|
|
},
|
|
{
|
|
name: "1024",
|
|
inValues: 1024,
|
|
},
|
|
{
|
|
name: "1025",
|
|
inValues: 1025,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
inValues := make([]int, test.inValues)
|
|
for i := 0; i < test.inValues; i++ {
|
|
inValues[i] = i
|
|
}
|
|
outValues := make([]int, test.inValues)
|
|
workerResults, err := async.Scatter(len(inValues), func(offset int, entries int, _ *sync.RWMutex) (interface{}, error) {
|
|
extent := make([]int, entries)
|
|
for i := 0; i < entries; i++ {
|
|
extent[i] = inValues[offset+i] * 2
|
|
}
|
|
return extent, nil
|
|
})
|
|
if test.err != nil {
|
|
assert.ErrorContains(t, test.err.Error(), err)
|
|
} else {
|
|
require.NoError(t, err)
|
|
for _, result := range workerResults {
|
|
copy(outValues[result.Offset:], result.Extent.([]int))
|
|
}
|
|
|
|
for i := 0; i < test.inValues; i++ {
|
|
require.Equal(t, inValues[i]*2, outValues[i], "Outvalue at %d mismatch", i)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestMutex(t *testing.T) {
|
|
totalRuns := 1048576
|
|
val := 0
|
|
_, err := async.Scatter(totalRuns, func(offset int, entries int, mu *sync.RWMutex) (interface{}, error) {
|
|
for i := 0; i < entries; i++ {
|
|
mu.Lock()
|
|
val++
|
|
mu.Unlock()
|
|
}
|
|
return nil, nil
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
if val != totalRuns {
|
|
t.Fatalf("Unexpected value: expected \"%v\", found \"%v\"", totalRuns, val)
|
|
}
|
|
}
|
|
|
|
func TestError(t *testing.T) {
|
|
totalRuns := 1024
|
|
val := 0
|
|
_, err := async.Scatter(totalRuns, func(offset int, entries int, mu *sync.RWMutex) (interface{}, error) {
|
|
for i := 0; i < entries; i++ {
|
|
mu.Lock()
|
|
val++
|
|
if val == 1011 {
|
|
mu.Unlock()
|
|
return nil, errors.New("bad number")
|
|
}
|
|
mu.Unlock()
|
|
}
|
|
return nil, nil
|
|
})
|
|
if err == nil {
|
|
t.Fatalf("Missing expected error")
|
|
}
|
|
}
|