test: add an integration test for UpdateAdmin

x/wasm/keeper/msg_server_integration_test.go

@@ -5,12 +5,15 @@ import (
 	_ "embed"
 	"encoding/hex"
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
+	abci "github.com/tendermint/tendermint/abci/types"
 	tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
 
 	"github.com/Finschia/finschia-sdk/testutil/testdata"
+	sdk "github.com/Finschia/finschia-sdk/types"
 
 	"github.com/Finschia/wasmd/app"
 	"github.com/Finschia/wasmd/x/wasm/types"
@@ -60,3 +63,92 @@ func TestStoreCode(t *testing.T) {
 	assert.Equal(t, sender.String(), info.Creator)
 	assert.Equal(t, types.DefaultParams().InstantiateDefaultPermission.With(sender), info.InstantiateConfig)
 }
+
+func TestUpdateAdmin(t *testing.T) {
+	wasmApp := app.Setup(false)
+	ctx := wasmApp.BaseApp.NewContext(false, tmproto.Header{Time: time.Now()})
+
+	var (
+		myAddress       sdk.AccAddress = make([]byte, types.ContractAddrLen)
+		_, _, otherAddr                = testdata.KeyTestPubAddr()
+	)
+
+	specs := map[string]struct {
+		addr           string
+		expErr         bool
+		assertEventsFn func(*testing.T, []abci.Event, string, string)
+	}{
+		"admin can update admin": {
+			addr:   myAddress.String(),
+			expErr: false,
+			assertEventsFn: func(t *testing.T, events []abci.Event, contractAddress, newAdmin string) {
+				assert.Equal(t, 2, len(events))
+				assert.Equal(t, "message", events[0].Type)
+				assert.Equal(t, 2, len(events[0].Attributes))
+				assert.Equal(t, "module", string(events[0].Attributes[0].Key))
+				assert.Equal(t, "wasm", string(events[0].Attributes[0].Value))
+				assert.Equal(t, "sender", string(events[0].Attributes[1].Key))
+				assert.Equal(t, myAddress.String(), string(events[0].Attributes[1].Value))
+				assert.Equal(t, "update_contract_admin", events[1].Type)
+				assert.Equal(t, 2, len(events[1].Attributes))
+				assert.Equal(t, "_contract_address", string(events[1].Attributes[0].Key))
+				assert.Equal(t, contractAddress, string(events[1].Attributes[0].Value))
+				assert.Equal(t, "new_admin_address", string(events[1].Attributes[1].Key))
+				assert.Equal(t, newAdmin, string(events[1].Attributes[1].Value))
+			},
+		},
+		"other address cannot update admin": {
+			addr:   otherAddr.String(),
+			expErr: true,
+		},
+	}
+	for name, spec := range specs {
+		t.Run(name, func(t *testing.T) {
+			_, _, newAdmin := testdata.KeyTestPubAddr()
+
+			// setup
+			storeMsg := &types.MsgStoreCode{
+				Sender:                myAddress.String(),
+				WASMByteCode:          wasmContract,
+				InstantiatePermission: &types.AllowEverybody,
+			}
+			rsp, err := wasmApp.MsgServiceRouter().Handler(storeMsg)(ctx, storeMsg)
+			require.NoError(t, err)
+			var storeCodeResult types.MsgStoreCodeResponse
+			require.NoError(t, wasmApp.AppCodec().Unmarshal(rsp.Data, &storeCodeResult))
+			codeID := storeCodeResult.CodeID
+
+			initMsg := &types.MsgInstantiateContract{
+				Sender: myAddress.String(),
+				Admin:  myAddress.String(),
+				CodeID: codeID,
+				Label:  "test",
+				Msg:    []byte(`{}`),
+				Funds:  sdk.Coins{},
+			}
+			rsp, err = wasmApp.MsgServiceRouter().Handler(initMsg)(ctx, initMsg)
+			require.NoError(t, err)
+
+			var instantiateContractResult types.MsgInstantiateContractResponse
+			require.NoError(t, wasmApp.AppCodec().Unmarshal(rsp.Data, &instantiateContractResult))
+			contractAddress := instantiateContractResult.Address
+
+			// when
+			msgUpdateAdmin := &types.MsgUpdateAdmin{
+				Sender:   spec.addr,
+				NewAdmin: newAdmin.String(),
+				Contract: contractAddress,
+			}
+			rsp, err = wasmApp.MsgServiceRouter().Handler(msgUpdateAdmin)(ctx, msgUpdateAdmin)
+
+			// then
+			if spec.expErr {
+				require.Error(t, err)
+				return
+			}
+
+			require.NoError(t, err)
+			spec.assertEventsFn(t, rsp.Events, contractAddress, newAdmin.String())
+		})
+	}
+}