status-im/status-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)
            }
        })
    }
}