cli/commands/mr/mr_test.go

41 lines
877 B
Go

package mr
import (
"bytes"
"io"
"os"
"testing"
"github.com/stretchr/testify/assert"
"gitlab.com/gitlab-org/cli/commands/cmdtest"
"gitlab.com/gitlab-org/cli/commands/cmdutils"
)
func TestMain(m *testing.M) {
cmdtest.InitTest(m, "mr_cmd_test")
cmdtest.InitTest(m, "mr_cmd_autofill")
}
func TestMrCmd_noARgs(t *testing.T) {
old := os.Stdout // keep backup of the real stdout
r, w, _ := os.Pipe()
os.Stdout = w
assert.Nil(t, NewCmdMR(&cmdutils.Factory{}).Execute())
outC := make(chan string)
// copy the output in a separate goroutine so printing can't block indefinitely
go func() {
var buf bytes.Buffer
_, _ = io.Copy(&buf, r)
outC <- buf.String()
}()
// back to normal state
w.Close()
os.Stdout = old // restoring the real stdout
out := <-outC
assert.Contains(t, out, "Use \"mr [command] --help\" for more information about a command.\n")
}