mirrors/vault
1
0
Fork 0
mirror of https://github.com/hashicorp/vault.git synced 2026-05-04 12:01:23 +02:00
Code Issues Projects Releases Wiki Activity

command/server: fix one race condition

Browse Source
This commit is contained in:
Mitchell Hashimoto 2015-04-28 19:11:46 -07:00
parent b0d25340da
commit e2aa51c22b
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
command/server/listener_test.go
View File

@ -5,7 +5,6 @@ import (
"io"
"net"
"testing"
"time"
)
type testListenerConnFn func(net.Listener) (net.Conn, error)
@ -30,14 +29,19 @@ func testListenerImpl(t *testing.T, ln net.Listener, connFn testListenerConnFn)
defer server.Close()
var buf bytes.Buffer
go io.Copy(&buf, server)
copyCh := make(chan struct{})
go func() {
io.Copy(&buf, server)
close(copyCh)
}()
if _, err := client.Write([]byte("foo")); err != nil {
t.Fatalf("err: %s", err)
}
time.Sleep(100 * time.Millisecond)
client.Close()
<-copyCh
if buf.String() != "foo" {
t.Fatalf("bad: %v", buf.String())
}