github.com/demonoid81/moby@v0.0.0-20200517203328-62dd8e17c460/pkg/progress/progressreader_test.go (about) 1 package progress // import "github.com/demonoid81/moby/pkg/progress" 2 3 import ( 4 "bytes" 5 "io" 6 "io/ioutil" 7 "testing" 8 ) 9 10 func TestOutputOnPrematureClose(t *testing.T) { 11 content := []byte("TESTING") 12 reader := ioutil.NopCloser(bytes.NewReader(content)) 13 progressChan := make(chan Progress, 10) 14 15 pr := NewProgressReader(reader, ChanOutput(progressChan), int64(len(content)), "Test", "Read") 16 17 part := make([]byte, 4) 18 _, err := io.ReadFull(pr, part) 19 if err != nil { 20 pr.Close() 21 t.Fatal(err) 22 } 23 24 drainLoop: 25 for { 26 select { 27 case <-progressChan: 28 default: 29 break drainLoop 30 } 31 } 32 33 pr.Close() 34 35 select { 36 case <-progressChan: 37 default: 38 t.Fatalf("Expected some output when closing prematurely") 39 } 40 } 41 42 func TestCompleteSilently(t *testing.T) { 43 content := []byte("TESTING") 44 reader := ioutil.NopCloser(bytes.NewReader(content)) 45 progressChan := make(chan Progress, 10) 46 47 pr := NewProgressReader(reader, ChanOutput(progressChan), int64(len(content)), "Test", "Read") 48 49 out, err := ioutil.ReadAll(pr) 50 if err != nil { 51 pr.Close() 52 t.Fatal(err) 53 } 54 if string(out) != "TESTING" { 55 pr.Close() 56 t.Fatalf("Unexpected output %q from reader", string(out)) 57 } 58 59 drainLoop: 60 for { 61 select { 62 case <-progressChan: 63 default: 64 break drainLoop 65 } 66 } 67 68 pr.Close() 69 70 select { 71 case <-progressChan: 72 t.Fatalf("Should have closed silently when read is complete") 73 default: 74 } 75 }