poll_fallback_test.go

package uv

import (
	"bytes"
	"fmt"
	"io"
	"sync"
	"testing"
)

type blockingReader struct {
	sync.Mutex
	read      bool
	unblockCh chan bool
	startedCh chan bool
}

func (r *blockingReader) Read([]byte) (int, error) {
	defer func() {
		r.Lock()
		defer r.Unlock()
		r.read = true
	}()
	r.startedCh <- true
	<-r.unblockCh
	return 0, fmt.Errorf("this error should be ignored")
}

func TestFallbackReaderConcurrentCancel(t *testing.T) {
	doneCh := make(chan bool, 1)
	startedCh := make(chan bool, 1)
	unblockCh := make(chan bool, 1)
	r := blockingReader{
		startedCh: startedCh,
		unblockCh: unblockCh,
	}
	pr, err := newFallbackReader(&r)
	if err != nil {
		t.Errorf("expected no error, but got %s", err)
	}

	go func() {
		defer func() { doneCh <- true }()
		if _, err := io.ReadAll(pr); err != ErrCanceled {
			t.Errorf("expected canceled error, got %v", err)
		}
	}()

	// make sure the read started before canceling the reader
	<-startedCh
	pr.Cancel()
	unblockCh <- true

	// wait for the read to end to ensure its assertions were made
	<-doneCh

	// make sure that it waited for the reader
	if !r.read {
		t.Error("seems like the reader was canceled before the read, this shouldn't happen")
	}
}

func TestFallbackReader(t *testing.T) {
	var r bytes.Buffer
	pr, err := newFallbackReader(&r)
	if err != nil {
		t.Errorf("expected no error, but got %s", err)
	}

	txt := "first"
	_, _ = r.WriteString(txt)
	first, err := io.ReadAll(pr)
	if err != nil {
		t.Errorf("expected no error, but got %s", err)
	}
	if string(first) != txt {
		t.Errorf("expected output to be %q, got %q", txt, string(first))
	}

	pr.Cancel()
	second, err := io.ReadAll(pr)
	if err != ErrCanceled {
		t.Errorf("expected ErrCanceled, got %v", err)
	}
	if len(second) > 0 {
		t.Errorf("expected an empty read, got %q", string(second))
	}
}