status-go/services/wallet/activity/session_test.go

package activity

import (
	"reflect"
	"testing"

	eth "github.com/ethereum/go-ethereum/common"

	"github.com/status-im/status-go/services/wallet/transfer"
)

// TODO #12120: cover missing cases
func TestFindUpdates(t *testing.T) {
	txIds := []transfer.TransactionIdentity{
		transfer.TransactionIdentity{
			ChainID: 1,
			Hash:    eth.HexToHash("0x1234"),
			Address: eth.HexToAddress("0x1234"),
		},
	}

	type findUpdatesResult struct {
		new     []mixedIdentityResult
		removed []EntryIdentity
	}

	tests := []struct {
		name       string
		identities []EntryIdentity
		updated    []Entry
		want       findUpdatesResult
	}{
		{
			name:       "Empty to single MT update",
			identities: []EntryIdentity{},
			updated: []Entry{
				{payloadType: MultiTransactionPT, id: 1},
			},
			want: findUpdatesResult{
				new: []mixedIdentityResult{{0, EntryIdentity{payloadType: MultiTransactionPT, id: 1}}},
			},
		},
		{
			name: "No updates",
			identities: []EntryIdentity{
				EntryIdentity{
					payloadType: SimpleTransactionPT, transaction: &txIds[0],
				},
			},
			updated: []Entry{
				{payloadType: SimpleTransactionPT, transaction: &txIds[0]},
			},
			want: findUpdatesResult{},
		},
		{
			name:       "Empty to mixed updates",
			identities: []EntryIdentity{},
			updated: []Entry{
				{payloadType: MultiTransactionPT, id: 1},
				{payloadType: PendingTransactionPT, transaction: &txIds[0]},
			},
			want: findUpdatesResult{
				new: []mixedIdentityResult{{0, EntryIdentity{payloadType: MultiTransactionPT, id: 1}},
					{1, EntryIdentity{payloadType: PendingTransactionPT, transaction: &txIds[0]}},
				},
			},
		},
		{
			name: "Add one on top of one",
			identities: []EntryIdentity{
				EntryIdentity{
					payloadType: MultiTransactionPT, id: 1,
				},
			},
			updated: []Entry{
				{payloadType: PendingTransactionPT, transaction: &txIds[0]},
				{payloadType: MultiTransactionPT, id: 1},
			},
			want: findUpdatesResult{
				new: []mixedIdentityResult{{0, EntryIdentity{payloadType: PendingTransactionPT, transaction: &txIds[0]}}},
			},
		},
		{
			name: "Add one on top keep window",
			identities: []EntryIdentity{
				EntryIdentity{payloadType: MultiTransactionPT, id: 1},
				EntryIdentity{payloadType: PendingTransactionPT, transaction: &txIds[0]},
			},
			updated: []Entry{
				{payloadType: MultiTransactionPT, id: 2},
				{payloadType: MultiTransactionPT, id: 1},
			},
			want: findUpdatesResult{
				new:     []mixedIdentityResult{{0, EntryIdentity{payloadType: MultiTransactionPT, id: 2}}},
				removed: []EntryIdentity{EntryIdentity{payloadType: PendingTransactionPT, transaction: &txIds[0]}},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			gotNew, gotRemoved := findUpdates(tt.identities, tt.updated)
			if !reflect.DeepEqual(gotNew, tt.want.new) || !reflect.DeepEqual(gotRemoved, tt.want.removed) {
				t.Errorf("findUpdates() = %v, %v, want %v, %v", gotNew, gotRemoved, tt.want.new, tt.want.removed)
			}
		})
	}
}