mirror of
https://github.com/planetmint/planetmint-go.git
synced 2025-03-30 15:08:28 +00:00
85 lines
3.0 KiB
Go
85 lines
3.0 KiB
Go
package util_test
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"math/rand"
|
|
"strconv"
|
|
"sync"
|
|
"testing"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/planetmint/planetmint-go/clients/shamir/coordinator"
|
|
"github.com/planetmint/planetmint-go/config"
|
|
"github.com/planetmint/planetmint-go/testutil/keeper"
|
|
clientmocks "github.com/planetmint/planetmint-go/testutil/mocks"
|
|
"github.com/planetmint/planetmint-go/testutil/moduleobject"
|
|
"github.com/planetmint/planetmint-go/testutil/sample"
|
|
"github.com/planetmint/planetmint-go/util"
|
|
"github.com/planetmint/planetmint-go/util/mocks"
|
|
"github.com/planetmint/planetmint-go/x/machine/types"
|
|
elements "github.com/rddl-network/elements-rpc"
|
|
elementsmocks "github.com/rddl-network/elements-rpc/utils/mocks"
|
|
scctypes "github.com/rddl-network/shamir-coordinator-service/types"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRegisterNFT(t *testing.T) {
|
|
params := types.DefaultParams()
|
|
url := params.AssetRegistryScheme + "://" + params.AssetRegistryDomain + "/" + params.AssetRegistryPath
|
|
entity := types.Entity{
|
|
Domain: params.AssetRegistryDomain,
|
|
}
|
|
c := types.Contract{
|
|
Entity: entity,
|
|
IssuerPubkey: "020000000000000000000000000000000000000000000000000000000000000000",
|
|
MachineAddr: "plmnt10mq5nj8jhh27z7ejnz2ql3nh0qhzjnfvy50877",
|
|
Name: "machine",
|
|
Precision: 0,
|
|
Version: 0,
|
|
}
|
|
contractBytes, err := json.Marshal(c)
|
|
assert.NoError(t, err)
|
|
|
|
contract := string(contractBytes)
|
|
asset := "0000000000000000000000000000000000000000000000000000000000000000"
|
|
goctx := context.Background()
|
|
|
|
util.RegisterAssetServiceHTTPClient = &mocks.MockClient{}
|
|
err = util.RegisterAsset(goctx, asset, contract, url)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestMachineNFTIssuance(t *testing.T) {
|
|
t.Setenv(config.ValAddr, "plmnt10mq5nj8jhh27z7ejnz2ql3nh0qhzjnfvy50877")
|
|
ctrl := gomock.NewController(t)
|
|
elements.Client = &elementsmocks.MockClient{}
|
|
shamirMock := clientmocks.NewMockISCClient(ctrl)
|
|
shamirMock.EXPECT().IssueMachineNFT(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(scctypes.IssueMachineNFTResponse{
|
|
HexTX: "0000000000000000000000000000000000000000000000000000000000000000",
|
|
Contract: `{"entity":{"domain":"testnet-assets.rddl.io"}, "issuer_pubkey":"02", "machine_addr":"addr","name":"machine","precicion":8,"version":1}`,
|
|
Asset: "0000000000000000000000000000000000000000000000000000000000000000",
|
|
}, nil)
|
|
coordinator.SCClient = shamirMock
|
|
util.RegisterAssetServiceHTTPClient = &mocks.MockClient{}
|
|
_, ctx := keeper.MachineKeeper(t)
|
|
params := types.DefaultParams()
|
|
var wg sync.WaitGroup
|
|
|
|
for i := 0; i < 10; i++ {
|
|
wg.Add(1)
|
|
go func() {
|
|
randomInt := rand.Int()
|
|
sk, pk := sample.KeyPair(randomInt)
|
|
machine := moduleobject.MachineRandom(pk, pk, sk, "address "+strconv.Itoa(randomInt), randomInt)
|
|
goCtx := sdk.WrapSDKContext(ctx)
|
|
|
|
err := util.IssueMachineNFT(goCtx, &machine, params.AssetRegistryScheme, params.AssetRegistryDomain, params.AssetRegistryPath)
|
|
assert.NoError(t, err)
|
|
wg.Done()
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
}
|