vimto/file_test.go at main · lmb/vimto · 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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
package main

import (
	"errors"
	"os"
	"testing"

	"github.com/creack/pty/v2"
	"github.com/go-quicktest/qt"
	"golang.org/x/sys/unix"
)

func TestFileControl(t *testing.T) {
	f, err := os.CreateTemp("", "")
	qt.Assert(t, qt.IsNil(err))
	defer f.Close()
	defer os.Remove(f.Name())

	sentinel := errors.New("sentinel")

	_, err = fileControl(f, func(fd uintptr) (struct{}, error) {
		return struct{}{}, sentinel
	})

	qt.Assert(t, qt.ErrorIs(err, sentinel))
}

func TestFlock(t *testing.T) {
	tmp := t.TempDir()
	f1, err := os.Open(tmp)
	qt.Assert(t, qt.IsNil(err))
	defer f1.Close()

	f2, err := os.Open(tmp)
	qt.Assert(t, qt.IsNil(err))
	defer f2.Close()

	qt.Assert(t, qt.IsNil(flock(f1, unix.LOCK_SH)))
	qt.Assert(t, qt.ErrorIs(flock(f2, unix.LOCK_EX|unix.LOCK_NB), unix.EWOULDBLOCK))

	f1.Close()
	qt.Assert(t, qt.IsNil(flock(f2, unix.LOCK_EX|unix.LOCK_NB)))
}

func TestFileIsTTY(t *testing.T) {
	pty, tty, err := pty.Open()
	qt.Assert(t, qt.IsNil(err))
	defer pty.Close()
	defer tty.Close()

	ok, err := fileIsTTY(pty)
	qt.Assert(t, qt.IsNil(err))
	qt.Assert(t, qt.Equals(ok, true))

	ok, err = fileIsTTY(tty)
	qt.Assert(t, qt.IsNil(err))
	qt.Assert(t, qt.Equals(ok, true))

	dir, err := os.Open(t.TempDir())
	qt.Assert(t, qt.IsNil(err))
	defer dir.Close()

	ok, err = fileIsTTY(dir)
	qt.Assert(t, qt.IsNil(err))
	qt.Assert(t, qt.Equals(ok, false))
}