f51f5e6784
Files returned by WriteToNewTempFile are cleaned up automatically at the end of a test case execution. WriteToNewTempFile now relies on the TB.TempDir() function provided by the testing std package. TempFile returns a temporary file that can be used within a test case and is automatically removed at the end of the test execution.
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
package testutil_test
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/cosmos/cosmos-sdk/testutil"
|
|
)
|
|
|
|
func TestApplyMockIO(t *testing.T) {
|
|
cmd := &cobra.Command{}
|
|
oldStdin := cmd.InOrStdin()
|
|
oldStdout := cmd.OutOrStdout()
|
|
oldStderr := cmd.ErrOrStderr()
|
|
|
|
testutil.ApplyMockIO(cmd)
|
|
|
|
require.NotEqual(t, cmd.InOrStdin(), oldStdin)
|
|
require.NotEqual(t, cmd.OutOrStdout(), oldStdout)
|
|
require.NotEqual(t, cmd.ErrOrStderr(), oldStderr)
|
|
require.Equal(t, cmd.ErrOrStderr(), cmd.OutOrStdout())
|
|
}
|
|
|
|
func TestWriteToNewTempFile(t *testing.T) {
|
|
tempfile := testutil.WriteToNewTempFile(t, "test string")
|
|
tempfile.Close()
|
|
|
|
bs, err := ioutil.ReadFile(tempfile.Name())
|
|
require.NoError(t, err)
|
|
require.Equal(t, "test string", string(bs))
|
|
}
|
|
|
|
func TestApplyMockIODiscardOutErr(t *testing.T) {
|
|
cmd := &cobra.Command{}
|
|
oldStdin := cmd.InOrStdin()
|
|
|
|
testutil.ApplyMockIODiscardOutErr(cmd)
|
|
require.NotEqual(t, cmd.InOrStdin(), oldStdin)
|
|
require.Equal(t, cmd.OutOrStdout(), ioutil.Discard)
|
|
require.Equal(t, cmd.ErrOrStderr(), ioutil.Discard)
|
|
}
|