diff --git a/testutil/keeper/asset.go b/testutil/keeper/asset.go index 92e1132..9c65177 100644 --- a/testutil/keeper/asset.go +++ b/testutil/keeper/asset.go @@ -47,7 +47,8 @@ func AssetKeeper(t testing.TB) (*keeper.Keeper, sdk.Context) { ctrl := gomock.NewController(t) mk := assettestutils.NewMockMachineKeeper(ctrl) - mk.EXPECT().GetMachine(ctx, "pubkey").Return(sample.Machine(), true) + mk.EXPECT().GetMachine(ctx, "pubkey").Return(sample.Machine(), true).AnyTimes() + mk.EXPECT().GetMachine(ctx, "privkey").Return(sample.Machine(), false).AnyTimes() k := keeper.NewKeeper( cdc, diff --git a/x/asset/keeper/msg_server_test.go b/x/asset/keeper/msg_server_test.go index eba8898..27da179 100644 --- a/x/asset/keeper/msg_server_test.go +++ b/x/asset/keeper/msg_server_test.go @@ -25,3 +25,10 @@ func TestMsgServerNotarizeAsset(t *testing.T) { assert.Equal(t, &types.MsgNotarizeAssetResponse{}, res) } } + +func TestMsgServerNotarizeAssetMachineNotFound(t *testing.T) { + msg := types.NewMsgNotarizeAsset("privkey", "cid", "sign", "pubkey") + msgServer, ctx := setupMsgServer(t) + _, err := msgServer.NotarizeAsset(ctx, msg) + assert.EqualError(t, err, "machine not found") +}