Merge pull request cosmos#437 from CosmWasm/wasm-migrate-msg

Encode WasmMsg::Migrate and test
notional-labs · Mar 5, 2021 · d51a1b8 · d51a1b8
2 parents b98353e + 27d5351
commit d51a1b8
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/x/wasm/internal/keeper/handler_plugin.go b/x/wasm/internal/keeper/handler_plugin.go
@@ -226,6 +226,14 @@ func EncodeWasmMsg(sender sdk.AccAddress, msg *wasmvmtypes.WasmMsg) ([]sdk.Msg,
 			Funds:   coins,
 		}
 		return []sdk.Msg{&sdkMsg}, nil
+	case msg.Migrate != nil:
+		sdkMsg := types.MsgMigrateContract{
+			Sender:     sender.String(),
+			Contract:   msg.Migrate.ContractAddr,
+			CodeID:     msg.Migrate.NewCodeID,
+			MigrateMsg: msg.Migrate.Msg,
+		}
+		return []sdk.Msg{&sdkMsg}, nil
 	default:
 		return nil, sdkerrors.Wrap(types.ErrInvalidMsg, "Unknown variant of Wasm")
 	}

diff --git a/x/wasm/internal/keeper/handler_plugin_test.go b/x/wasm/internal/keeper/handler_plugin_test.go
@@ -172,6 +172,26 @@ func TestEncoding(t *testing.T) {
 				},
 			},
 		},
+		"wasm migrate": {
+			sender: addr2,
+			srcMsg: wasmvmtypes.CosmosMsg{
+				Wasm: &wasmvmtypes.WasmMsg{
+					Migrate: &wasmvmtypes.MigrateMsg{
+						ContractAddr: addr1.String(),
+						NewCodeID:    12,
+						Msg:          jsonMsg,
+					},
+				},
+			},
+			output: []sdk.Msg{
+				&types.MsgMigrateContract{
+					Sender:     addr2.String(),
+					Contract:   addr1.String(),
+					CodeID:     12,
+					MigrateMsg: jsonMsg,
+				},
+			},
+		},
 		"staking delegate": {
 			sender: addr1,
 			srcMsg: wasmvmtypes.CosmosMsg{