mirror of
https://github.com/planetmint/planetmint-go.git
synced 2025-05-23 07:16:39 +00:00
59 lines
1.7 KiB
Go
59 lines
1.7 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
keepertest "github.com/planetmint/planetmint-go/testutil/keeper"
|
|
"github.com/planetmint/planetmint-go/x/machine/keeper"
|
|
"github.com/planetmint/planetmint-go/x/machine/types"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func createNMachine(keeper *keeper.Keeper, ctx sdk.Context, n int) []types.Machine {
|
|
items := make([]types.Machine, n)
|
|
for i := range items {
|
|
items[i].MachineId = fmt.Sprintf("machineId%v", i)
|
|
items[i].IssuerPlanetmint = fmt.Sprintf("issuerPlanetmint%v", i)
|
|
items[i].IssuerLiquid = fmt.Sprintf("issuerLiquid%v", i)
|
|
items[i].Address = fmt.Sprintf("address%v", i)
|
|
keeper.StoreMachine(ctx, items[i])
|
|
keeper.StoreMachineIndex(ctx, items[i])
|
|
}
|
|
return items
|
|
}
|
|
|
|
func TestGetMachine(t *testing.T) {
|
|
keeper, ctx := keepertest.MachineKeeper(t)
|
|
items := createNMachine(keeper, ctx, 10)
|
|
for _, item := range items {
|
|
index := types.MachineIndex{
|
|
MachineId: item.MachineId,
|
|
IssuerPlanetmint: item.IssuerPlanetmint,
|
|
IssuerLiquid: item.IssuerLiquid,
|
|
Address: item.Address,
|
|
}
|
|
machineById, found := keeper.GetMachine(ctx, index)
|
|
assert.True(t, found)
|
|
assert.Equal(t, item, machineById)
|
|
}
|
|
}
|
|
|
|
func TestGetMachineIndexByPubKey(t *testing.T) {
|
|
keeper, ctx := keepertest.MachineKeeper(t)
|
|
items := createNMachine(keeper, ctx, 10)
|
|
for _, item := range items {
|
|
expectedIndex := types.MachineIndex{
|
|
MachineId: item.MachineId,
|
|
IssuerPlanetmint: item.IssuerPlanetmint,
|
|
IssuerLiquid: item.IssuerLiquid,
|
|
Address: item.Address,
|
|
}
|
|
index, found := keeper.GetMachineIndexByPubKey(ctx, item.MachineId)
|
|
assert.True(t, found)
|
|
assert.Equal(t, expectedIndex, index)
|
|
}
|
|
}
|