44 lines
833 B
Go
44 lines
833 B
Go
package nullio
|
|
|
|
import (
|
|
"context"
|
|
"github.com/stretchr/testify/assert"
|
|
"io"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
const shortTimeout = 100 * time.Millisecond
|
|
|
|
func TestReaderDoesNotReturnImmediately(t *testing.T) {
|
|
readingContext, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
reader := &Reader{readingContext}
|
|
readerReturned := make(chan bool)
|
|
go func() {
|
|
p := make([]byte, 0, 5)
|
|
_, err := reader.Read(p)
|
|
assert.ErrorIs(t, io.EOF, err)
|
|
close(readerReturned)
|
|
}()
|
|
|
|
var received bool
|
|
select {
|
|
case <-readerReturned:
|
|
received = true
|
|
case <-time.After(shortTimeout):
|
|
received = false
|
|
}
|
|
|
|
assert.False(t, received)
|
|
}
|
|
|
|
func TestReadWriterWritesEverything(t *testing.T) {
|
|
readWriter := &ReadWriter{}
|
|
p := []byte{1, 2, 3}
|
|
n, err := readWriter.Write(p)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, len(p), n)
|
|
}
|