chore(client): use command's configured output (#22334)

This commit is contained in:
Mark Rushakoff
2024-10-22 23:08:45 +00:00
committed by GitHub
parent cd3ce0c7a4
commit 1c949f772f
2 changed files with 27 additions and 1 deletions
+26
View File
@@ -1,6 +1,7 @@
package client_test
import (
"bytes"
"context"
"fmt"
"testing"
@@ -137,3 +138,28 @@ func TestSetCmdClientContextHandler(t *testing.T) {
})
}
}
func TestContext_usesCobraCommandOutput(t *testing.T) {
var initCtx client.Context
cmd := &cobra.Command{
PreRunE: func(cmd *cobra.Command, args []string) error {
return client.SetCmdClientContextHandler(initCtx, cmd)
},
RunE: func(cmd *cobra.Command, _ []string) error {
cctx, err := client.GetClientTxContext(cmd)
if err != nil {
return err
}
return cctx.PrintString("hello")
},
}
var outBuf bytes.Buffer
cmd.SetOutput(&outBuf)
require.NoError(t, cmd.Execute())
require.Equal(t, "hello", outBuf.String())
}