added DeadlineHolder

photostorm · Oct 28, 2023 · e9c1f00 · e9c1f00
1 parent 7f5c495
commit e9c1f00
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/doc.go b/doc.go
@@ -19,6 +19,10 @@ func Open() (Pty, Tty, error) {
 // FdHolder surfaces the Fd() method of the underlying handle.
 type FdHolder interface {
 	Fd() uintptr
+}
+
+// DeadlineHolder surfaces the SetDeadline() method to sets the read and write deadlines.
+type DeadlineHolder interface {
 	SetDeadline(t time.Time) error
 }
 

diff --git a/io_test.go b/io_test.go
@@ -27,12 +27,14 @@ var mu sync.Mutex
 func TestReadDeadline(t *testing.T) {
 	ptmx, success := prepare(t)
 
-	err := ptmx.SetDeadline(time.Now().Add(timeout / 10))
-	if err != nil {
-		if errors.Is(err, os.ErrNoDeadline) {
-			t.Skipf("deadline is not supported on %s/%s", runtime.GOOS, runtime.GOARCH)
-		} else {
-			t.Fatalf("error: set deadline: %v\n", err)
+	if ptmxd, ok := ptmx.(DeadlineHolder); ok {
+		err := ptmxd.SetDeadline(time.Now().Add(timeout / 10))
+		if err != nil {
+			if errors.Is(err, os.ErrNoDeadline) {
+				t.Skipf("deadline is not supported on %s/%s", runtime.GOOS, runtime.GOARCH)
+			} else {
+				t.Fatalf("error: set deadline: %v\n", err)
+			}
 		}
 	}