term_test.go

//go:build !windows
// +build !windows

package term

import (
	"os"
	"reflect"
	"testing"

	cpty "github.com/creack/pty"
)

func newTTYForTest(t *testing.T) *os.File {
	t.Helper()
	pty, tty, err := cpty.Open()
	if err != nil {
		t.Fatalf("error creating pty: %v", err)
	} else {
		t.Cleanup(func() {
			_ = pty.Close()
			_ = tty.Close()
		})
	}

	return pty
}

func newTempFile(t *testing.T) *os.File {
	t.Helper()
	tmpFile, err := os.CreateTemp(t.TempDir(), "temp")
	if err != nil {
		t.Fatalf("error creating tempfile: %v", err)
	} else {
		t.Cleanup(func() { _ = tmpFile.Close() })
	}
	return tmpFile
}

func TestGetWinsize(t *testing.T) {
	tty := newTTYForTest(t)
	winSize, err := GetWinsize(tty.Fd())
	if err != nil {
		t.Error(err)
	}
	if winSize == nil {
		t.Fatal("winSize is nil")
	}

	newSize := Winsize{Width: 200, Height: 200, x: winSize.x, y: winSize.y}
	err = SetWinsize(tty.Fd(), &newSize)
	if err != nil {
		t.Fatal(err)
	}
	winSize, err = GetWinsize(tty.Fd())
	if err != nil {
		t.Fatal(err)
	}
	if !reflect.DeepEqual(*winSize, newSize) {
		t.Fatalf("expected: %+v, got: %+v", newSize, *winSize)
	}
}

func TestSetWinsize(t *testing.T) {
	tty := newTTYForTest(t)
	winSize, err := GetWinsize(tty.Fd())
	if err != nil {
		t.Fatal(err)
	}
	if winSize == nil {
		t.Fatal("winSize is nil")
	}
	newSize := Winsize{Width: 200, Height: 200, x: winSize.x, y: winSize.y}
	err = SetWinsize(tty.Fd(), &newSize)
	if err != nil {
		t.Fatal(err)
	}
	winSize, err = GetWinsize(tty.Fd())
	if err != nil {
		t.Fatal(err)
	}
	if !reflect.DeepEqual(*winSize, newSize) {
		t.Fatalf("expected: %+v, got: %+v", newSize, *winSize)
	}
}

func TestGetFdInfo(t *testing.T) {
	tty := newTTYForTest(t)
	inFd, isTerm := GetFdInfo(tty)
	if inFd != tty.Fd() {
		t.Errorf("expected: %d, got: %d", tty.Fd(), inFd)
	}
	if !isTerm {
		t.Error("expected file-descriptor to be a terminal")
	}
	tmpFile := newTempFile(t)
	inFd, isTerm = GetFdInfo(tmpFile)
	if inFd != tmpFile.Fd() {
		t.Errorf("expected: %d, got: %d", tty.Fd(), inFd)
	}
	if isTerm {
		t.Error("expected file-descriptor to not be a terminal")
	}
}

func TestIsTerminal(t *testing.T) {
	tty := newTTYForTest(t)
	isTerm := IsTerminal(tty.Fd())
	if !isTerm {
		t.Error("expected file-descriptor to be a terminal")
	}
	tmpFile := newTempFile(t)
	isTerm = IsTerminal(tmpFile.Fd())
	if isTerm {
		t.Error("expected file-descriptor to not be a terminal")
	}
}

func TestSaveState(t *testing.T) {
	tty := newTTYForTest(t)
	state, err := SaveState(tty.Fd())
	if err != nil {
		t.Error(err)
	}
	if state == nil {
		t.Fatal("state is nil")
	}
	tty = newTTYForTest(t)
	err = RestoreTerminal(tty.Fd(), state)
	if err != nil {
		t.Fatal(err)
	}
}

func TestDisableEcho(t *testing.T) {
	tty := newTTYForTest(t)
	state, err := SetRawTerminal(tty.Fd())
	defer func() {
		if err := RestoreTerminal(tty.Fd(), state); err != nil {
			t.Error(err)
		}
	}()
	if err != nil {
		t.Error(err)
	}
	if state == nil {
		t.Fatal("state is nil")
	}
	err = DisableEcho(tty.Fd(), state)
	if err != nil {
		t.Fatal(err)
	}
}