Function TestPollMod

poll_test.go:54–114

(t *testing.T)

}

func TestPollMod(t *testing.T) {
	var rn, wn, hn int32
	read := func(p Poll) error {
		atomic.AddInt32(&rn, 1)
		return nil
	}
	write := func(p Poll) error {
		atomic.AddInt32(&wn, 1)
		return nil
	}
	hup := func(p Poll) error {
		atomic.AddInt32(&hn, 1)
		return nil
	}
	stop := make(chan error)
	p, err := openDefaultPoll()
	MustNil(t, err)
	go func() {
		stop <- p.Wait()
	}()

	rfd, wfd := GetSysFdPairs()
	rop := &FDOperator{FD: rfd, OnRead: read, OnWrite: write, OnHup: hup, poll: p}
	wop := &FDOperator{FD: wfd, OnRead: read, OnWrite: write, OnHup: hup, poll: p}
	var r, w, h int32
	r, w, h = atomic.LoadInt32(&rn), atomic.LoadInt32(&wn), atomic.LoadInt32(&hn)
	Assert(t, r == 0 && w == 0 && h == 0, r, w, h)
	err = p.Control(rop, PollReadable)
	MustNil(t, err)
	r, w, h = atomic.LoadInt32(&rn), atomic.LoadInt32(&wn), atomic.LoadInt32(&hn)
	Assert(t, r == 0 && w == 0 && h == 0, r, w, h)

	err = p.Control(wop, PollWritable) // trigger one shot
	MustNil(t, err)
	for atomic.LoadInt32(&wn) == 0 {
		runtime.Gosched()
	}
	r, w, h = atomic.LoadInt32(&rn), atomic.LoadInt32(&wn), atomic.LoadInt32(&hn)
	Assert(t, r == 0 && w >= 1 && h == 0, r, w, h)

	err = p.Control(rop, PollR2RW) // trigger write
	MustNil(t, err)
	for atomic.LoadInt32(&wn) <= 1 {
		runtime.Gosched()
	}
	r, w, h = atomic.LoadInt32(&rn), atomic.LoadInt32(&wn), atomic.LoadInt32(&hn)
	Assert(t, r == 0 && w >= 2 && h == 0, r, w, h)

	// close wfd, then trigger hup rfd
	err = syscall.Close(wfd) // trigger hup
	MustNil(t, err)
	for atomic.LoadInt32(&hn) == 0 {
		runtime.Gosched()
	}
	w, h = atomic.LoadInt32(&wn), atomic.LoadInt32(&hn)
	Assert(t, w >= 2 && h >= 1, r, w, h)

	p.Close()