2022-11-15 12:14:41 +00:00
|
|
|
package history
|
|
|
|
|
|
|
|
import (
|
|
|
|
"math/big"
|
2023-10-04 12:00:12 +00:00
|
|
|
"reflect"
|
2022-11-15 12:14:41 +00:00
|
|
|
"testing"
|
|
|
|
|
2023-10-04 12:00:12 +00:00
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
|
2022-11-15 12:14:41 +00:00
|
|
|
"github.com/ethereum/go-ethereum/common"
|
2023-08-11 11:28:45 +00:00
|
|
|
"github.com/status-im/status-go/t/helpers"
|
|
|
|
"github.com/status-im/status-go/walletdatabase"
|
2022-11-15 12:14:41 +00:00
|
|
|
)
|
|
|
|
|
2023-10-04 12:00:12 +00:00
|
|
|
func newTestDB(t *testing.T) *BalanceDB {
|
2023-08-11 11:28:45 +00:00
|
|
|
db, err := helpers.SetupTestMemorySQLDB(walletdatabase.DbInitializer{})
|
2022-11-15 12:14:41 +00:00
|
|
|
require.NoError(t, err)
|
2023-10-04 12:00:12 +00:00
|
|
|
return NewBalanceDB(db)
|
2022-11-15 12:14:41 +00:00
|
|
|
}
|
|
|
|
|
2023-10-04 12:00:12 +00:00
|
|
|
func dbWithEntries(t *testing.T, entries []*entry) *BalanceDB {
|
|
|
|
db := newTestDB(t)
|
|
|
|
for _, entry := range entries {
|
|
|
|
err := db.add(entry)
|
|
|
|
require.NoError(t, err)
|
2022-11-15 12:14:41 +00:00
|
|
|
}
|
2023-10-04 12:00:12 +00:00
|
|
|
return db
|
2022-11-15 12:14:41 +00:00
|
|
|
}
|
|
|
|
|
2023-10-04 12:00:12 +00:00
|
|
|
func TestBalance_addPaddingPoints(t *testing.T) {
|
|
|
|
type args struct {
|
|
|
|
currency string
|
2023-11-29 11:20:18 +00:00
|
|
|
addresses []common.Address
|
2023-10-04 12:00:12 +00:00
|
|
|
fromTimestamp uint64
|
|
|
|
currentTimestamp uint64
|
|
|
|
data []*entry
|
|
|
|
limit int
|
2022-11-15 12:14:41 +00:00
|
|
|
}
|
2023-10-04 12:00:12 +00:00
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
|
|
|
want []*entry
|
|
|
|
wantErr bool
|
2022-11-15 12:14:41 +00:00
|
|
|
}{
|
2023-10-04 12:00:12 +00:00
|
|
|
{
|
|
|
|
name: "addOnePaddingPointAtMiddle",
|
|
|
|
args: args{
|
|
|
|
currency: "ETH",
|
2023-11-29 11:20:18 +00:00
|
|
|
addresses: []common.Address{common.Address{1}},
|
2023-10-04 12:00:12 +00:00
|
|
|
fromTimestamp: 0,
|
|
|
|
currentTimestamp: 2,
|
|
|
|
data: []*entry{
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 0,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(2),
|
|
|
|
timestamp: 2,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
limit: 3,
|
|
|
|
},
|
|
|
|
want: []*entry{
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 0,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 1,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(2),
|
|
|
|
timestamp: 2,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "noPaddingEqualsLimit",
|
|
|
|
args: args{
|
|
|
|
currency: "ETH",
|
2023-11-29 11:20:18 +00:00
|
|
|
addresses: []common.Address{common.Address{1}},
|
2023-10-04 12:00:12 +00:00
|
|
|
fromTimestamp: 0,
|
|
|
|
currentTimestamp: 2,
|
|
|
|
data: []*entry{
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 0,
|
|
|
|
block: big.NewInt(1),
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(1),
|
|
|
|
timestamp: 2,
|
|
|
|
block: big.NewInt(2),
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
limit: 2,
|
|
|
|
},
|
|
|
|
want: []*entry{
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 0,
|
|
|
|
block: big.NewInt(1),
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(1),
|
|
|
|
timestamp: 2,
|
|
|
|
block: big.NewInt(2),
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "limitLessThanDataSize",
|
|
|
|
args: args{
|
|
|
|
currency: "ETH",
|
2023-11-29 11:20:18 +00:00
|
|
|
addresses: []common.Address{common.Address{1}},
|
2023-10-04 12:00:12 +00:00
|
|
|
fromTimestamp: 0,
|
|
|
|
currentTimestamp: 2,
|
|
|
|
data: []*entry{
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 0,
|
|
|
|
block: big.NewInt(1),
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(1),
|
|
|
|
timestamp: 2,
|
|
|
|
block: big.NewInt(2),
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
limit: 1,
|
|
|
|
},
|
|
|
|
want: []*entry{
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 0,
|
|
|
|
block: big.NewInt(1),
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(1),
|
|
|
|
timestamp: 2,
|
|
|
|
block: big.NewInt(2),
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "addMultiplePaddingPoints",
|
|
|
|
args: args{
|
|
|
|
currency: "ETH",
|
2023-11-29 11:20:18 +00:00
|
|
|
addresses: []common.Address{common.Address{1}},
|
2023-10-04 12:00:12 +00:00
|
|
|
fromTimestamp: 1,
|
|
|
|
currentTimestamp: 5,
|
|
|
|
data: []*entry{
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 1,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(4),
|
|
|
|
timestamp: 4,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(5),
|
|
|
|
timestamp: 5,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
limit: 5,
|
|
|
|
},
|
|
|
|
want: []*entry{
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 1,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 2,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 3,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(4),
|
|
|
|
timestamp: 4,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(5),
|
|
|
|
timestamp: 5,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "addMultiplePaddingPointsDuplicateTimestamps",
|
|
|
|
args: args{
|
|
|
|
currency: "ETH",
|
2023-11-29 11:20:18 +00:00
|
|
|
addresses: []common.Address{common.Address{1}},
|
2023-10-04 12:00:12 +00:00
|
|
|
fromTimestamp: 1,
|
|
|
|
currentTimestamp: 5,
|
|
|
|
data: []*entry{
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 1,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 1,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(4),
|
|
|
|
timestamp: 4,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(5),
|
|
|
|
timestamp: 5,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
limit: 5,
|
|
|
|
},
|
|
|
|
want: []*entry{
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 1,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 1,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 2,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(4),
|
|
|
|
timestamp: 4,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
balance: big.NewInt(5),
|
|
|
|
timestamp: 5,
|
|
|
|
tokenSymbol: "ETH",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
2023-11-29 11:20:18 +00:00
|
|
|
gotRes, err := addPaddingPoints(tt.args.currency, tt.args.addresses, tt.args.currentTimestamp, tt.args.data, tt.args.limit)
|
2023-10-04 12:00:12 +00:00
|
|
|
if (err != nil) != tt.wantErr {
|
|
|
|
t.Errorf("Balance.addPaddingPoints() error = %v, wantErr %v", err, tt.wantErr)
|
|
|
|
return
|
2022-11-15 12:14:41 +00:00
|
|
|
}
|
2023-10-04 12:00:12 +00:00
|
|
|
if !reflect.DeepEqual(gotRes, tt.want) {
|
|
|
|
t.Errorf("Balance.addPaddingPoints() = %v, want %v", gotRes, tt.want)
|
2022-11-15 12:14:41 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-10-04 12:00:12 +00:00
|
|
|
func TestBalance_addEdgePoints(t *testing.T) {
|
2022-11-15 12:14:41 +00:00
|
|
|
|
2023-10-04 12:00:12 +00:00
|
|
|
walletDB := newTestDB(t)
|
2022-11-15 12:14:41 +00:00
|
|
|
|
2023-10-04 12:00:12 +00:00
|
|
|
type fields struct {
|
|
|
|
db *BalanceDB
|
2022-11-15 12:14:41 +00:00
|
|
|
}
|
2023-10-04 12:00:12 +00:00
|
|
|
type args struct {
|
|
|
|
chainID uint64
|
|
|
|
currency string
|
2023-11-29 11:20:18 +00:00
|
|
|
addresses []common.Address
|
2023-10-04 12:00:12 +00:00
|
|
|
fromTimestamp uint64
|
|
|
|
toTimestamp uint64
|
|
|
|
data []*entry
|
2022-11-15 12:14:41 +00:00
|
|
|
}
|
2023-10-04 12:00:12 +00:00
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
fields fields
|
|
|
|
args args
|
|
|
|
wantRes []*entry
|
|
|
|
wantErr bool
|
2022-11-15 12:14:41 +00:00
|
|
|
}{
|
2023-10-04 12:00:12 +00:00
|
|
|
{
|
|
|
|
name: "addToEmptyData",
|
|
|
|
fields: fields{
|
|
|
|
db: walletDB,
|
|
|
|
},
|
|
|
|
args: args{
|
|
|
|
chainID: 111,
|
|
|
|
currency: "SNT",
|
2023-11-29 11:20:18 +00:00
|
|
|
addresses: []common.Address{common.Address{1}},
|
2023-10-04 12:00:12 +00:00
|
|
|
fromTimestamp: 1,
|
|
|
|
toTimestamp: 2,
|
|
|
|
data: []*entry{},
|
|
|
|
},
|
|
|
|
wantRes: []*entry{
|
|
|
|
{
|
|
|
|
chainID: 111,
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 1,
|
|
|
|
tokenSymbol: "SNT",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
chainID: 111,
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: 2,
|
|
|
|
tokenSymbol: "SNT",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "addToEmptyDataSinceGenesis",
|
|
|
|
fields: fields{
|
|
|
|
db: walletDB,
|
|
|
|
},
|
|
|
|
args: args{
|
|
|
|
chainID: 111,
|
|
|
|
currency: "SNT",
|
2023-11-29 11:20:18 +00:00
|
|
|
addresses: []common.Address{common.Address{1}},
|
2023-10-04 12:00:12 +00:00
|
|
|
fromTimestamp: 0, // will set to genesisTimestamp
|
|
|
|
toTimestamp: genesisTimestamp + 1,
|
|
|
|
data: []*entry{},
|
|
|
|
},
|
|
|
|
wantRes: []*entry{
|
|
|
|
{
|
|
|
|
chainID: 111,
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: genesisTimestamp,
|
|
|
|
tokenSymbol: "SNT",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
chainID: 111,
|
|
|
|
balance: big.NewInt(0),
|
|
|
|
timestamp: genesisTimestamp + 1,
|
|
|
|
tokenSymbol: "SNT",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: "addToNonEmptyDataFromPreviousEntry",
|
|
|
|
fields: fields{
|
|
|
|
db: dbWithEntries(t, []*entry{
|
|
|
|
{
|
|
|
|
chainID: 111,
|
|
|
|
balance: big.NewInt(1),
|
|
|
|
timestamp: 1,
|
|
|
|
block: big.NewInt(1),
|
|
|
|
tokenSymbol: "SNT",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
}),
|
|
|
|
},
|
|
|
|
args: args{
|
|
|
|
chainID: 111,
|
|
|
|
currency: "SNT",
|
2023-11-29 11:20:18 +00:00
|
|
|
addresses: []common.Address{common.Address{1}},
|
2023-10-04 12:00:12 +00:00
|
|
|
fromTimestamp: 2,
|
|
|
|
toTimestamp: 4,
|
|
|
|
data: []*entry{
|
|
|
|
{
|
|
|
|
chainID: 111,
|
|
|
|
balance: big.NewInt(3),
|
|
|
|
timestamp: 3,
|
|
|
|
block: big.NewInt(3),
|
|
|
|
tokenSymbol: "SNT",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
chainID: 111,
|
|
|
|
balance: big.NewInt(2),
|
|
|
|
timestamp: 4,
|
|
|
|
block: big.NewInt(4),
|
|
|
|
tokenSymbol: "SNT",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantRes: []*entry{
|
|
|
|
{
|
|
|
|
chainID: 111,
|
|
|
|
balance: big.NewInt(1),
|
|
|
|
timestamp: 2,
|
|
|
|
tokenSymbol: "SNT",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
chainID: 111,
|
|
|
|
balance: big.NewInt(3),
|
|
|
|
timestamp: 3,
|
|
|
|
block: big.NewInt(3),
|
|
|
|
tokenSymbol: "SNT",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
chainID: 111,
|
|
|
|
balance: big.NewInt(2),
|
|
|
|
timestamp: 4,
|
|
|
|
block: big.NewInt(4),
|
|
|
|
tokenSymbol: "SNT",
|
|
|
|
address: common.Address{1},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
wantErr: false,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
b := &Balance{
|
|
|
|
db: tt.fields.db,
|
2022-11-15 12:14:41 +00:00
|
|
|
}
|
2023-11-29 11:20:18 +00:00
|
|
|
gotRes, err := b.addEdgePoints(tt.args.chainID, tt.args.currency, tt.args.addresses, tt.args.fromTimestamp, tt.args.toTimestamp, tt.args.data)
|
2023-10-04 12:00:12 +00:00
|
|
|
if (err != nil) != tt.wantErr {
|
|
|
|
t.Errorf("Balance.addEdgePoints() error = %v, wantErr %v", err, tt.wantErr)
|
|
|
|
return
|
2022-11-15 12:14:41 +00:00
|
|
|
}
|
2023-10-04 12:00:12 +00:00
|
|
|
if !reflect.DeepEqual(gotRes, tt.wantRes) {
|
|
|
|
t.Errorf("Balance.addEdgePoints() = \n%v,\nwant \n%v", gotRes, tt.wantRes)
|
2022-11-15 12:14:41 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|