terraform/vendor/github.com/mitchellh/prefixedio/reader_test.go

97 lines
1.5 KiB
Go

package prefixedio
import (
"bytes"
"io"
"strings"
"testing"
)
func TestReader(t *testing.T) {
var fooBuf, barBuf, defBuf bytes.Buffer
original := bytes.NewReader([]byte(strings.TrimSpace(testInput)))
r, err := NewReader(original)
if err != nil {
t.Fatalf("err: %s", err)
}
pFoo, err := r.Prefix("foo: ")
if err != nil {
t.Fatalf("err: %s", err)
}
_, err = r.Prefix("foo: ")
if err == nil {
t.Fatalf("expected prefix already registered error")
}
pBar, err := r.Prefix("bar: ")
if err != nil {
t.Fatalf("err: %s", err)
}
pDefault, err := r.Prefix("")
if err != nil {
t.Fatalf("err: %s", err)
}
doneCh := make(chan struct{}, 3)
go func() {
if _, err := io.Copy(&fooBuf, pFoo); err != nil {
t.Fatalf("err: %s", err)
}
doneCh <- struct{}{}
}()
go func() {
if _, err := io.Copy(&barBuf, pBar); err != nil {
t.Fatalf("err: %s", err)
}
doneCh <- struct{}{}
}()
go func() {
if _, err := io.Copy(&defBuf, pDefault); err != nil {
t.Fatalf("err: %s", err)
}
doneCh <- struct{}{}
}()
// Wait for all the reads to be done
<-doneCh
<-doneCh
<-doneCh
if fooBuf.String() != testInputFoo {
t.Fatalf("bad: %s", fooBuf.String())
}
if barBuf.String() != testInputBar {
t.Fatalf("bad: %s", barBuf.String())
}
if defBuf.String() != testInputDef {
t.Fatalf("bad: %s", defBuf.String())
}
}
const testInput = `
what
hello
foo: 1
bar: 2
bar: 3
42
foo: 4
foo: 5
bar: 6
`
const testInputFoo = `1
4
5
`
const testInputBar = `2
3
42
6`
const testInputDef = `what
hello
`