File: eager_reader_test.go

package info (click to toggle)
golang-github-htcat-htcat 1.0.2-1.1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye, forky, sid, trixie
  • size: 100 kB
  • sloc: makefile: 5
file content (33 lines) | stat: -rw-r--r-- 468 bytes parent folder | download | duplicates (2) 
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
 
package htcat

import (
	"bytes"
	"io"
	"io/ioutil"
	"testing"
)

func TestEagerReaderRead(t *testing.T) {
	reader, writer := io.Pipe()
	payload := []byte("hello")

	er := newEagerReader(ioutil.NopCloser(reader), int64(len(payload)))
	done := make(chan struct{})

	go func() {
		writer.Write(payload)
		writer.Close()
	}()

	var buf bytes.Buffer
	go func() {
		er.WriteTo(&buf)
		done <- struct{}{}
	}()

	<-done

	if string(buf.Bytes()) != "hello" {
		t.Fatal()
	}
}