65 lines
1.1 KiB
Go
65 lines
1.1 KiB
Go
package print
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/alcionai/clues"
|
|
"github.com/spf13/cobra"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/alcionai/canario/src/internal/tester"
|
|
)
|
|
|
|
type PrintUnitSuite struct {
|
|
tester.Suite
|
|
}
|
|
|
|
func TestPrintUnitSuite(t *testing.T) {
|
|
suite.Run(t, &PrintUnitSuite{Suite: tester.NewUnitSuite(t)})
|
|
}
|
|
|
|
func (suite *PrintUnitSuite) TestOnly() {
|
|
t := suite.T()
|
|
|
|
ctx, flush := tester.NewContext(t)
|
|
defer flush()
|
|
|
|
c := &cobra.Command{}
|
|
ctx = SetRootCmd(ctx, c)
|
|
|
|
err := Only(ctx, nil)
|
|
assert.NoError(t, err, clues.ToCore(err))
|
|
assert.True(t, c.SilenceUsage)
|
|
}
|
|
|
|
func (suite *PrintUnitSuite) TestOut() {
|
|
t := suite.T()
|
|
|
|
ctx, flush := tester.NewContext(t)
|
|
defer flush()
|
|
|
|
b := bytes.Buffer{}
|
|
msg := "I have seen the fnords!"
|
|
|
|
out(ctx, &b, msg)
|
|
assert.Contains(t, b.String(), msg)
|
|
}
|
|
|
|
func (suite *PrintUnitSuite) TestOutf() {
|
|
t := suite.T()
|
|
|
|
ctx, flush := tester.NewContext(t)
|
|
defer flush()
|
|
|
|
b := bytes.Buffer{}
|
|
msg := "I have seen the fnords!"
|
|
msg2 := "smarf"
|
|
|
|
outf(ctx, &b, msg, msg2)
|
|
bs := b.String()
|
|
assert.Contains(t, bs, msg)
|
|
assert.Contains(t, bs, msg2)
|
|
}
|