go/src/github.com/monopole/mdrip/util/buff_scanner_test.go

package util

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

type stalledReader struct {
	bytes.Buffer
}

func (stalledReader) Read(p []byte) (n int, err error) {
	time.Sleep(5 * time.Second)
	return 0, nil
}
func (stalledReader) Close() error { return nil }

func TestStalledReader(t *testing.T) {
	foo := stalledReader{}
	chOut := BuffScanner(1*time.Second, "heythere", foo, true)

	line, ok := <-chOut
	if !ok {
		t.Fail()
	}
	want := MsgTimeout
	if line != want {
		t.Errorf("got \n\t%v\nwant\n\t%v", line, want)
	}

	line, ok = <-chOut
	if ok {
		t.Fail()
	}
}

type bustedReader struct {
	bytes.Buffer
}

func (bustedReader) Read(p []byte) (n int, err error) {
	return 0, nil
}

func (bustedReader) Close() error { return nil }

func TestBustedReader(t *testing.T) {
	foo := bustedReader{}
	chOut := BuffScanner(1*time.Second, "heythere", foo, true)

	line, ok := <-chOut
	if !ok {
		t.Fail()
	}
	want := MsgError + " : multiple Read calls return no data or error"
	if line != want {
		t.Errorf("got \n\t%v\nwant\n\t%v", line, want)
	}

	line, ok = <-chOut
	if ok {
		t.Fail()
	}
}

type simpleReader struct {
	io.Reader
}

func (simpleReader) Close() error { return nil }

func TestSimpleReader(t *testing.T) {
	foo1 := simpleReader{bytes.NewBufferString("beans and\nrice")}
	chOut := BuffScanner(1*time.Second, "heythere", foo1, true)

	line, ok := <-chOut
	if !ok {
		t.Fail()
	}
	want := "beans and"
	if line != want {
		t.Errorf("got \n\t%v\nwant\n\t%v", line, want)
	}

	line, ok = <-chOut
	if !ok {
		t.Fail()
	}
	want = "rice"
	if line != want {
		t.Errorf("got \n\t%v\nwant\n\t%v", line, want)
	}

	line, ok = <-chOut
	if ok {
		t.Fail()
	}
}

// An example main.
func main() {
	{
		foo := simpleReader{bytes.NewBufferString("beans and\nrice")}
		chOut := BuffScanner(1*time.Second, "heythere", foo, true)
		for line := range chOut {
			fmt.Println(line)
		}
		fmt.Println("-----------------------")
	}
	{
		foo := stalledReader{}
		chOut := BuffScanner(1*time.Second, "heythere", foo, true)
		for line := range chOut {
			fmt.Println(line)
		}
		fmt.Println("-----------------------")
	}
	{
		foo := bustedReader{}
		chOut := BuffScanner(1*time.Second, "heythere", foo, true)
		for line := range chOut {
			fmt.Println(line)
		}
		fmt.Println("-----------------------")
	}
}