github.com/Finschia/finschia-sdk@v0.48.1/types/simulation/account_test.go (about)

     1  package simulation_test
     2  
     3  import (
     4  	"math/rand"
     5  	"testing"
     6  	"time"
     7  
     8  	"github.com/stretchr/testify/require"
     9  
    10  	sdk "github.com/Finschia/finschia-sdk/types"
    11  	"github.com/Finschia/finschia-sdk/types/simulation"
    12  )
    13  
    14  func TestRandomAccounts(t *testing.T) {
    15  	t.Parallel()
    16  	r := rand.New(rand.NewSource(time.Now().Unix()))
    17  	tests := []struct {
    18  		name string
    19  		n    int
    20  		want int
    21  	}{
    22  		{"0-accounts", 0, 0},
    23  		{"1-accounts", 1, 1},
    24  		{"100-accounts", 100, 100},
    25  	}
    26  	for _, tt := range tests {
    27  		tt := tt
    28  		t.Run(tt.name, func(t *testing.T) {
    29  			got := simulation.RandomAccounts(r, tt.n)
    30  			require.Equal(t, tt.want, len(got))
    31  			if tt.n == 0 {
    32  				return
    33  			}
    34  			acc, i := simulation.RandomAcc(r, got)
    35  			require.True(t, acc.Equals(got[i]))
    36  			accFound, found := simulation.FindAccount(got, acc.Address)
    37  			require.True(t, found)
    38  			require.True(t, accFound.Equals(acc))
    39  		})
    40  	}
    41  }
    42  
    43  func TestFindAccountEmptySlice(t *testing.T) {
    44  	t.Parallel()
    45  	r := rand.New(rand.NewSource(time.Now().Unix()))
    46  	accs := simulation.RandomAccounts(r, 1)
    47  	require.Equal(t, 1, len(accs))
    48  	acc, found := simulation.FindAccount(nil, accs[0].Address)
    49  	require.False(t, found)
    50  	require.Nil(t, acc.Address)
    51  	require.Nil(t, acc.PrivKey)
    52  	require.Nil(t, acc.PubKey)
    53  }
    54  
    55  func TestRandomFees(t *testing.T) {
    56  	t.Parallel()
    57  	r := rand.New(rand.NewSource(time.Now().Unix()))
    58  	tests := []struct {
    59  		name           string
    60  		spendableCoins sdk.Coins
    61  		wantEmpty      bool
    62  		wantErr        bool
    63  	}{
    64  		{"0 coins", sdk.Coins{}, true, false},
    65  		{"0 coins", sdk.NewCoins(sdk.NewInt64Coin("aaa", 10), sdk.NewInt64Coin("bbb", 5)), false, false},
    66  	}
    67  	for _, tt := range tests {
    68  		tt := tt
    69  
    70  		t.Run(tt.name, func(t *testing.T) {
    71  			got, err := simulation.RandomFees(r, sdk.Context{}, tt.spendableCoins)
    72  			if (err != nil) != tt.wantErr {
    73  				t.Errorf("RandomFees() error = %v, wantErr %v", err, tt.wantErr)
    74  				return
    75  			}
    76  			require.Equal(t, tt.wantEmpty, got.Empty())
    77  		})
    78  	}
    79  }