mirror of
https://github.com/planetmint/planetmint-go.git
synced 2025-11-24 06:25:47 +00:00
55 lines
1.5 KiB
Go
55 lines
1.5 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
keepertest "planetmint-go/testutil/keeper"
|
|
"planetmint-go/x/machine/keeper"
|
|
"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)
|
|
keeper.StoreMachine(ctx, items[i])
|
|
}
|
|
return items
|
|
}
|
|
|
|
func TestGetMachineById(t *testing.T) {
|
|
keeper, ctx := keepertest.MachineKeeper(t)
|
|
items := createNMachine(keeper, ctx, 10)
|
|
for _, item := range items {
|
|
machineById, found := keeper.GetMachine(ctx, item.MachineId)
|
|
assert.True(t, found)
|
|
assert.Equal(t, item, machineById)
|
|
}
|
|
}
|
|
|
|
func TestGetMachineByIssuerPlanetmint(t *testing.T) {
|
|
keeper, ctx := keepertest.MachineKeeper(t)
|
|
items := createNMachine(keeper, ctx, 10)
|
|
for _, item := range items {
|
|
machineById, found := keeper.GetMachine(ctx, item.IssuerPlanetmint)
|
|
assert.True(t, found)
|
|
assert.Equal(t, item, machineById)
|
|
}
|
|
}
|
|
|
|
func TestGetMachineByIssuerLiquid(t *testing.T) {
|
|
keeper, ctx := keepertest.MachineKeeper(t)
|
|
items := createNMachine(keeper, ctx, 10)
|
|
for _, item := range items {
|
|
machineById, found := keeper.GetMachine(ctx, item.IssuerLiquid)
|
|
assert.True(t, found)
|
|
assert.Equal(t, item, machineById)
|
|
}
|
|
}
|