cliui/cliui_test.go at main · gookit/cliui · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
package cliui_test

import (
	"bytes"
	"context"
	"testing"

	"github.com/gookit/cliui"
	"github.com/gookit/cliui/interact"
	"github.com/gookit/cliui/interact/backend/plain"
	"github.com/gookit/cliui/interact/ui"
	"github.com/gookit/goutil/testutil/assert"
)

func TestCustomIOAffectsInteractReadInput(t *testing.T) {
	is := assert.New(t)

	in := bytes.NewBufferString("tom\n")
	out := new(bytes.Buffer)
	cliui.CustomIO(in, out)
	defer cliui.ResetIO()

	got, err := interact.ReadInput("name:")
	is.Nil(err)
	is.Eq("tom", got)
	is.Contains(out.String(), "name:")
}

func TestCustomIOAffectsUIDefaultIO(t *testing.T) {
	is := assert.New(t)

	in := bytes.NewBufferString("tom\n")
	out := new(bytes.Buffer)
	cliui.CustomIO(in, out)
	defer cliui.ResetIO()

	got, err := ui.NewInput("name").Run(context.Background(), plain.New())
	is.Nil(err)
	is.Eq("tom", got)
	is.Contains(out.String(), "name:")
}