1
0
Fork 0
mirror of https://github.com/moby/moby.git synced 2022-11-09 12:21:53 -05:00
moby--moby/pkg/term/term_linux_test.go
Daniel Nephin 4f0d95fa6e Add canonical import comment
Signed-off-by: Daniel Nephin <dnephin@docker.com>
2018-02-05 16:51:57 -05:00

115 lines
2.9 KiB
Go

//+build linux
package term // import "github.com/docker/docker/pkg/term"
import (
"io/ioutil"
"os"
"testing"
"github.com/stretchr/testify/require"
)
// RequiresRoot skips tests that require root, unless the test.root flag has
// been set
func RequiresRoot(t *testing.T) {
if os.Getuid() != 0 {
t.Skip("skipping test that requires root")
return
}
}
func newTtyForTest(t *testing.T) (*os.File, error) {
RequiresRoot(t)
return os.OpenFile("/dev/tty", os.O_RDWR, os.ModeDevice)
}
func newTempFile() (*os.File, error) {
return ioutil.TempFile(os.TempDir(), "temp")
}
func TestGetWinsize(t *testing.T) {
tty, err := newTtyForTest(t)
defer tty.Close()
require.NoError(t, err)
winSize, err := GetWinsize(tty.Fd())
require.NoError(t, err)
require.NotNil(t, winSize)
require.NotNil(t, winSize.Height)
require.NotNil(t, winSize.Width)
newSize := Winsize{Width: 200, Height: 200, x: winSize.x, y: winSize.y}
err = SetWinsize(tty.Fd(), &newSize)
require.NoError(t, err)
winSize, err = GetWinsize(tty.Fd())
require.NoError(t, err)
require.Equal(t, *winSize, newSize)
}
func TestSetWinsize(t *testing.T) {
tty, err := newTtyForTest(t)
defer tty.Close()
require.NoError(t, err)
winSize, err := GetWinsize(tty.Fd())
require.NoError(t, err)
require.NotNil(t, winSize)
newSize := Winsize{Width: 200, Height: 200, x: winSize.x, y: winSize.y}
err = SetWinsize(tty.Fd(), &newSize)
require.NoError(t, err)
winSize, err = GetWinsize(tty.Fd())
require.NoError(t, err)
require.Equal(t, *winSize, newSize)
}
func TestGetFdInfo(t *testing.T) {
tty, err := newTtyForTest(t)
defer tty.Close()
require.NoError(t, err)
inFd, isTerminal := GetFdInfo(tty)
require.Equal(t, inFd, tty.Fd())
require.Equal(t, isTerminal, true)
tmpFile, err := newTempFile()
require.NoError(t, err)
defer tmpFile.Close()
inFd, isTerminal = GetFdInfo(tmpFile)
require.Equal(t, inFd, tmpFile.Fd())
require.Equal(t, isTerminal, false)
}
func TestIsTerminal(t *testing.T) {
tty, err := newTtyForTest(t)
defer tty.Close()
require.NoError(t, err)
isTerminal := IsTerminal(tty.Fd())
require.Equal(t, isTerminal, true)
tmpFile, err := newTempFile()
require.NoError(t, err)
defer tmpFile.Close()
isTerminal = IsTerminal(tmpFile.Fd())
require.Equal(t, isTerminal, false)
}
func TestSaveState(t *testing.T) {
tty, err := newTtyForTest(t)
defer tty.Close()
require.NoError(t, err)
state, err := SaveState(tty.Fd())
require.NoError(t, err)
require.NotNil(t, state)
tty, err = newTtyForTest(t)
require.NoError(t, err)
defer tty.Close()
err = RestoreTerminal(tty.Fd(), state)
require.NoError(t, err)
}
func TestDisableEcho(t *testing.T) {
tty, err := newTtyForTest(t)
defer tty.Close()
require.NoError(t, err)
state, err := SetRawTerminal(tty.Fd())
defer RestoreTerminal(tty.Fd(), state)
require.NoError(t, err)
require.NotNil(t, state)
err = DisableEcho(tty.Fd(), state)
require.NoError(t, err)
}