You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
1.2 KiB
Go
87 lines
1.2 KiB
Go
package cmdline
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/tal-tech/go-zero/core/lang"
|
|
)
|
|
|
|
func TestEnterToContinue(t *testing.T) {
|
|
r, w, err := os.Pipe()
|
|
assert.Nil(t, err)
|
|
ow := os.Stdout
|
|
os.Stdout = w
|
|
or := os.Stdin
|
|
os.Stdin = r
|
|
defer func() {
|
|
os.Stdin = or
|
|
os.Stdout = ow
|
|
}()
|
|
|
|
var wg sync.WaitGroup
|
|
wg.Add(2)
|
|
go func() {
|
|
defer wg.Done()
|
|
fmt.Println()
|
|
}()
|
|
go func() {
|
|
defer wg.Done()
|
|
EnterToContinue()
|
|
}()
|
|
|
|
wait := make(chan lang.PlaceholderType)
|
|
go func() {
|
|
wg.Wait()
|
|
close(wait)
|
|
}()
|
|
|
|
select {
|
|
case <-time.After(time.Second):
|
|
t.Error("timeout")
|
|
case <-wait:
|
|
}
|
|
}
|
|
|
|
func TestReadLine(t *testing.T) {
|
|
r, w, err := os.Pipe()
|
|
assert.Nil(t, err)
|
|
ow := os.Stdout
|
|
os.Stdout = w
|
|
or := os.Stdin
|
|
os.Stdin = r
|
|
defer func() {
|
|
os.Stdin = or
|
|
os.Stdout = ow
|
|
}()
|
|
|
|
const message = "hello"
|
|
var wg sync.WaitGroup
|
|
wg.Add(2)
|
|
go func() {
|
|
defer wg.Done()
|
|
fmt.Println(message)
|
|
}()
|
|
go func() {
|
|
defer wg.Done()
|
|
input := ReadLine("")
|
|
assert.Equal(t, message, input)
|
|
}()
|
|
|
|
wait := make(chan lang.PlaceholderType)
|
|
go func() {
|
|
wg.Wait()
|
|
close(wait)
|
|
}()
|
|
|
|
select {
|
|
case <-time.After(time.Second):
|
|
t.Error("timeout")
|
|
case <-wait:
|
|
}
|
|
}
|