50 lines
1.5 KiB
Go
50 lines
1.5 KiB
Go
|
// Copyright (C) 2022 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package storjscan_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"storj.io/common/testcontext"
|
||
|
|
"storj.io/common/testrand"
|
||
|
|
"storj.io/storj/private/blockchain"
|
||
|
|
"storj.io/storj/satellite"
|
||
|
|
"storj.io/storj/satellite/satellitedb/satellitedbtest"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestWalletsDB(t *testing.T) {
|
||
|
|
userID1 := testrand.UUID()
|
||
|
|
userID2 := testrand.UUID()
|
||
|
|
userID3 := testrand.UUID()
|
||
|
|
walletAddress1, err := blockchain.BytesToAddress(testrand.Bytes(20))
|
||
|
|
require.NoError(t, err)
|
||
|
|
walletAddress2, err := blockchain.BytesToAddress(testrand.Bytes(20))
|
||
|
|
require.NoError(t, err)
|
||
|
|
walletAddress3, err := blockchain.BytesToAddress(testrand.Bytes(20))
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
t.Run("get wallet", func(t *testing.T) {
|
||
|
|
satellitedbtest.Run(t, func(ctx *testcontext.Context, t *testing.T, db satellite.DB) {
|
||
|
|
err := db.Wallets().Add(ctx, userID1, walletAddress1)
|
||
|
|
require.NoError(t, err)
|
||
|
|
err = db.Wallets().Add(ctx, userID2, walletAddress2)
|
||
|
|
require.NoError(t, err)
|
||
|
|
err = db.Wallets().Add(ctx, userID3, walletAddress3)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
address1, err := db.Wallets().Get(ctx, userID1)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, walletAddress1, address1)
|
||
|
|
address2, err := db.Wallets().Get(ctx, userID2)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, walletAddress2, address2)
|
||
|
|
address3, err := db.Wallets().Get(ctx, userID3)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, walletAddress3, address3)
|
||
|
|
})
|
||
|
|
})
|
||
|
|
}
|