Add support for broadcasting arbitrary messages as test users #186
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,40 @@ | ||
| package ibctest | ||
|
|
||
| import ( | ||
| "context" | ||
|
|
||
| "github.com/cosmos/cosmos-sdk/client/tx" | ||
| sdk "github.com/cosmos/cosmos-sdk/types" | ||
| "github.com/strangelove-ventures/ibctest/broadcast" | ||
| ) | ||
|
|
||
| // BroadcastTx uses the provided Broadcaster to broadcast all the provided messages which will be signed | ||
| // by the User provided. The sdk.TxResponse and an error are returned. | ||
| func BroadcastTx(ctx context.Context, broadcaster broadcast.Broadcaster, broadcastingUser broadcast.User, msgs ...sdk.Msg) (sdk.TxResponse, error) { | ||
| for _, msg := range msgs { | ||
| if err := msg.ValidateBasic(); err != nil { | ||
| return sdk.TxResponse{}, err | ||
| } | ||
| } | ||
|
|
||
| f, err := broadcaster.GetFactory(ctx, broadcastingUser) | ||
| if err != nil { | ||
| return sdk.TxResponse{}, err | ||
| } | ||
|
|
||
| cc, err := broadcaster.GetClientContext(ctx, broadcastingUser) | ||
| if err != nil { | ||
| return sdk.TxResponse{}, err | ||
| } | ||
|
|
||
| if err := tx.BroadcastTx(cc, f, msgs...); err != nil { | ||
| return sdk.TxResponse{}, err | ||
| } | ||
|
|
||
| txBytes, err := broadcaster.GetTxResponseBytes(ctx, broadcastingUser) | ||
| if err != nil { | ||
| return sdk.TxResponse{}, err | ||
| } | ||
|
|
||
| return broadcaster.UnmarshalTxResponseBytes(ctx, txBytes) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,28 @@ | ||
| package broadcast | ||
|
|
||
| import ( | ||
| "context" | ||
|
|
||
| "github.com/cosmos/cosmos-sdk/client" | ||
| "github.com/cosmos/cosmos-sdk/client/tx" | ||
| sdk "github.com/cosmos/cosmos-sdk/types" | ||
| ) | ||
|
|
||
| type ClientContextOpt func(clientContext client.Context) client.Context | ||
|
|
||
| type FactoryOpt func(factory tx.Factory) tx.Factory | ||
|
|
||
| type User interface { | ||
| GetKeyName() string | ||
| Bech32Address(bech32Prefix string) string | ||
| } | ||
|
|
||
| // Broadcaster implementations can broadcast messages as the provided user. | ||
| type Broadcaster interface { | ||
| ConfigureFactoryOptions(opts ...FactoryOpt) | ||
| ConfigureClientContextOptions(opts ...ClientContextOpt) | ||
| GetFactory(ctx context.Context, user User) (tx.Factory, error) | ||
| GetClientContext(ctx context.Context, user User) (client.Context, error) | ||
| GetTxResponseBytes(ctx context.Context, user User) ([]byte, error) | ||
| UnmarshalTxResponseBytes(ctx context.Context, bytes []byte) (sdk.TxResponse, error) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,170 @@ | ||
| package cosmos | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "context" | ||
| "fmt" | ||
| "testing" | ||
|
|
||
| "github.com/cosmos/cosmos-sdk/client" | ||
| "github.com/cosmos/cosmos-sdk/client/flags" | ||
| "github.com/cosmos/cosmos-sdk/client/tx" | ||
| "github.com/cosmos/cosmos-sdk/crypto/keyring" | ||
| sdk "github.com/cosmos/cosmos-sdk/types" | ||
| "github.com/cosmos/cosmos-sdk/types/tx/signing" | ||
| authtypes "github.com/cosmos/cosmos-sdk/x/auth/types" | ||
| "github.com/strangelove-ventures/ibctest/broadcast" | ||
| "github.com/strangelove-ventures/ibctest/internal/dockerutil" | ||
| ) | ||
|
|
||
| var _ broadcast.Broadcaster = &Broadcaster{} | ||
|
|
||
| type Broadcaster struct { | ||
| // buf stores the output sdk.TxResponse when broadcast.Tx is invoked. | ||
| buf *bytes.Buffer | ||
| // keyrings is a mapping of keyrings which point to a temporary test directory. The contents | ||
| // of this directory are copied from the node container for the specific user. | ||
| keyrings map[broadcast.User]keyring.Keyring | ||
|
|
||
| // chain is a reference to the CosmosChain instance which will be the target of the messages. | ||
| chain *CosmosChain | ||
| // t is the testing.T for the current test. | ||
| t *testing.T | ||
|
|
||
| // factoryOptions is a slice of broadcast.FactoryOpt which enables arbitrary configuration of the tx.Factory. | ||
| factoryOptions []broadcast.FactoryOpt | ||
| // clientContextOptions is a slice of broadcast.ClientContextOpt which enables arbitrary configuration of the client.Context. | ||
| clientContextOptions []broadcast.ClientContextOpt | ||
| } | ||
|
|
||
| // NewBroadcaster returns a instance of Broadcaster which can be used with broadcast.Tx to | ||
| // broadcast messages sdk messages. | ||
| func NewBroadcaster(t *testing.T, chain *CosmosChain) *Broadcaster { | ||
| return &Broadcaster{ | ||
| t: t, | ||
| chain: chain, | ||
| buf: &bytes.Buffer{}, | ||
| keyrings: map[broadcast.User]keyring.Keyring{}, | ||
| } | ||
| } | ||
|
|
||
| // ConfigureFactoryOptions ensure the given configuration functions are run when calling GetFactory | ||
| // after all default options have been applied. | ||
| func (b *Broadcaster) ConfigureFactoryOptions(opts ...broadcast.FactoryOpt) { | ||
| b.factoryOptions = append(b.factoryOptions, opts...) | ||
| } | ||
|
|
||
| // ConfigureClientContextOptions ensure the given configuration functions are run when calling GetClientContext | ||
| // after all default options have been applied. | ||
| func (b *Broadcaster) ConfigureClientContextOptions(opts ...broadcast.ClientContextOpt) { | ||
| b.clientContextOptions = append(b.clientContextOptions, opts...) | ||
| } | ||
|
|
||
| // GetFactory returns an instance of tx.Factory that is configured with this Broadcaster's CosmosChain | ||
| // and the provided user. | ||
| func (b *Broadcaster) GetFactory(ctx context.Context, user broadcast.User) (tx.Factory, error) { | ||
| clientContext, err := b.GetClientContext(ctx, user) | ||
| if err != nil { | ||
| return tx.Factory{}, err | ||
| } | ||
|
|
||
| sdkAdd, err := sdk.AccAddressFromBech32(user.Bech32Address(b.chain.Config().Bech32Prefix)) | ||
| if err != nil { | ||
| return tx.Factory{}, err | ||
| } | ||
|
|
||
| accNumber, err := clientContext.AccountRetriever.GetAccount(clientContext, sdkAdd) | ||
| if err != nil { | ||
| return tx.Factory{}, err | ||
| } | ||
|
|
||
| f := b.defaultTxFactory(clientContext, accNumber.GetAccountNumber()) | ||
| for _, opt := range b.factoryOptions { | ||
| f = opt(f) | ||
| } | ||
| return f, nil | ||
| } | ||
|
|
||
| // GetClientContext returns a client context that is configured with this Broadcaster's CosmosChain and | ||
| // the provided user. | ||
| func (b *Broadcaster) GetClientContext(ctx context.Context, user broadcast.User) (client.Context, error) { | ||
mark-rushakoff marked this conversation as resolved.
Show resolved
Hide resolved
|
||
| chain := b.chain | ||
| cn := chain.getFullNode() | ||
|
|
||
| _, ok := b.keyrings[user] | ||
| if !ok { | ||
| localDir := b.t.TempDir() | ||
| containerKeyringDir := fmt.Sprintf("%s/keyring-test", cn.HomeDir()) | ||
|
|
||
| kr, err := dockerutil.NewLocalKeyringFromDockerContainer(ctx, cn.DockerClient, localDir, containerKeyringDir, cn.containerID) | ||
| if err != nil { | ||
| return client.Context{}, err | ||
| } | ||
| b.keyrings[user] = kr | ||
| } | ||
|
|
||
| sdkAdd, err := sdk.AccAddressFromBech32(user.Bech32Address(chain.Config().Bech32Prefix)) | ||
| if err != nil { | ||
| return client.Context{}, err | ||
| } | ||
|
|
||
| clientContext := b.defaultClientContext(user, sdkAdd) | ||
| for _, opt := range b.clientContextOptions { | ||
| clientContext = opt(clientContext) | ||
| } | ||
| return clientContext, nil | ||
| } | ||
|
|
||
| // GetTxResponseBytes returns the sdk.TxResponse bytes which returned from broadcast.Tx. | ||
| func (b *Broadcaster) GetTxResponseBytes(ctx context.Context, user broadcast.User) ([]byte, error) { | ||
| if b.buf == nil || b.buf.Len() == 0 { | ||
| return nil, fmt.Errorf("empty buffer, transaction has not be executed yet") | ||
|
|
||
| } | ||
| return b.buf.Bytes(), nil | ||
| } | ||
|
|
||
| // UnmarshalTxResponseBytes accepts the sdk.TxResponse bytes and unmarshalls them into an | ||
| // instance of sdk.TxResponse. | ||
| func (b *Broadcaster) UnmarshalTxResponseBytes(ctx context.Context, bytes []byte) (sdk.TxResponse, error) { | ||
| resp := sdk.TxResponse{} | ||
| if err := defaultEncoding.Marshaler.UnmarshalJSON(bytes, &resp); err != nil { | ||
| return sdk.TxResponse{}, err | ||
| } | ||
| return resp, nil | ||
| } | ||
|
|
||
| // defaultClientContext returns a default client context configured with the user as the sender. | ||
| func (b *Broadcaster) defaultClientContext(fromUser broadcast.User, sdkAdd sdk.AccAddress) client.Context { | ||
| // initialize a clean buffer each time | ||
| b.buf.Reset() | ||
| kr := b.keyrings[fromUser] | ||
| cn := b.chain.getFullNode() | ||
| return cn.CliContext(). | ||
| WithOutput(b.buf). | ||
| WithFrom(fromUser.Bech32Address(b.chain.Config().Bech32Prefix)). | ||
| WithFromAddress(sdkAdd). | ||
| WithFromName(fromUser.GetKeyName()). | ||
| WithSkipConfirmation(true). | ||
| WithAccountRetriever(authtypes.AccountRetriever{}). | ||
| WithKeyring(kr). | ||
| WithBroadcastMode(flags.BroadcastBlock). | ||
| WithCodec(defaultEncoding.Marshaler). | ||
| WithHomeDir(cn.Home) | ||
|
|
||
| } | ||
|
|
||
| // defaultTxFactory creates a new Factory with default configuration. | ||
| func (b *Broadcaster) defaultTxFactory(clientCtx client.Context, accountNumber uint64) tx.Factory { | ||
| chainConfig := b.chain.Config() | ||
| return tx.Factory{}. | ||
| WithAccountNumber(accountNumber). | ||
| WithSignMode(signing.SignMode_SIGN_MODE_DIRECT). | ||
| WithGasAdjustment(chainConfig.GasAdjustment). | ||
| WithGas(flags.DefaultGasLimit). | ||
| WithGasPrices(chainConfig.GasPrices). | ||
| WithMemo("ibctest"). | ||
| WithTxConfig(clientCtx.TxConfig). | ||
| WithAccountRetriever(clientCtx.AccountRetriever). | ||
| WithKeybase(clientCtx.Keyring). | ||
| WithChainID(clientCtx.ChainID). | ||
| WithSimulateAndExecute(false) | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Wouldn't hurt to mention
ConfigureFactoryOptionsin the godoc for this method.