mirror of
https://github.com/moby/moby.git
synced 2026-01-11 18:51:37 +00:00
225b7ca6b7
These comments were added to enforce using the correct import path for
our packages ("github.com/docker/docker", not "github.com/moby/moby").
However, when working in go module mode (not GOPATH / vendor), they have
no effect, so their impact is limited.
Remove these imports in preparation of migrating our code to become an
actual go module.
Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
75 lines
1.3 KiB
Go
75 lines
1.3 KiB
Go
package progress
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"testing"
|
|
)
|
|
|
|
func TestOutputOnPrematureClose(t *testing.T) {
|
|
content := []byte("TESTING")
|
|
reader := io.NopCloser(bytes.NewReader(content))
|
|
progressChan := make(chan Progress, 10)
|
|
|
|
pr := NewProgressReader(reader, ChanOutput(progressChan), int64(len(content)), "Test", "Read")
|
|
|
|
part := make([]byte, 4)
|
|
_, err := io.ReadFull(pr, part)
|
|
if err != nil {
|
|
pr.Close()
|
|
t.Fatal(err)
|
|
}
|
|
|
|
drainLoop:
|
|
for {
|
|
select {
|
|
case <-progressChan:
|
|
default:
|
|
break drainLoop
|
|
}
|
|
}
|
|
|
|
pr.Close()
|
|
|
|
select {
|
|
case <-progressChan:
|
|
default:
|
|
t.Fatalf("Expected some output when closing prematurely")
|
|
}
|
|
}
|
|
|
|
func TestCompleteSilently(t *testing.T) {
|
|
content := []byte("TESTING")
|
|
reader := io.NopCloser(bytes.NewReader(content))
|
|
progressChan := make(chan Progress, 10)
|
|
|
|
pr := NewProgressReader(reader, ChanOutput(progressChan), int64(len(content)), "Test", "Read")
|
|
|
|
out, err := io.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))
|
|
}
|
|
|
|
drainLoop:
|
|
for {
|
|
select {
|
|
case <-progressChan:
|
|
default:
|
|
break drainLoop
|
|
}
|
|
}
|
|
|
|
pr.Close()
|
|
|
|
select {
|
|
case <-progressChan:
|
|
t.Fatalf("Should have closed silently when read is complete")
|
|
default:
|
|
}
|
|
}
|