iframe-proxy | Sunbelt Computer Software

75 lines (62 loc) · 1.3 KB

package progress
	"io/ioutil"
func TestOutputOnPrematureClose(t *testing.T) {
	content := []byte("TESTING")
	reader := ioutil.NopCloser(bytes.NewReader(content))
	progressChan := make(chan Progress, 10)
	pr := NewProgressReader(reader, ChanOutput(progressChan), int64(len(content)), "Test", "Read")
	part := make([]byte, 4, 4)
	_, err := io.ReadFull(pr, part)
	if err != nil {
		pr.Close()
		t.Fatal(err)
		case <-progressChan:
			break drainLoop
	pr.Close()
	case <-progressChan:
		t.Fatalf("Expected some output when closing prematurely")
func TestCompleteSilently(t *testing.T) {
	content := []byte("TESTING")
	reader := ioutil.NopCloser(bytes.NewReader(content))
	progressChan := make(chan Progress, 10)
	pr := NewProgressReader(reader, ChanOutput(progressChan), int64(len(content)), "Test", "Read")
	out, err := ioutil.ReadAll(pr)
	if err != nil {
		pr.Close()
		t.Fatal(err)
	if string(out) != "TESTING" {
		pr.Close()
		t.Fatalf("Unexpected output %q from reader", string(out))
		case <-progressChan:
			break drainLoop
	pr.Close()
	case <-progressChan:
		t.Fatalf("Should have closed silently when read is complete")

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

progressreader_test.go

Latest commit

History

progressreader_test.go

File metadata and controls