33 lines
715 B
Go
33 lines
715 B
Go
package testutil
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client"
|
|
distrcli "github.com/cosmos/cosmos-sdk/x/distribution/client/cli"
|
|
)
|
|
|
|
func MsgWithdrawDelegatorRewardExec(clientCtx client.Context, valAddr fmt.Stringer, extraArgs ...string) ([]byte, error) {
|
|
buf := new(bytes.Buffer)
|
|
clientCtx = clientCtx.WithOutput(buf)
|
|
|
|
ctx := context.Background()
|
|
ctx = context.WithValue(ctx, client.ClientContextKey, &clientCtx)
|
|
|
|
args := []string{valAddr.String()}
|
|
args = append(args, extraArgs...)
|
|
|
|
cmd := distrcli.NewWithdrawRewardsCmd()
|
|
cmd.SetErr(buf)
|
|
cmd.SetOut(buf)
|
|
cmd.SetArgs(args)
|
|
|
|
if err := cmd.ExecuteContext(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return buf.Bytes(), nil
|
|
}
|