Fixed the data race found in kubelet_test.go

This commit is contained in:
Dawn Chen 2014-07-01 15:15:40 -07:00
parent de06869d30
commit 17bad028ce

View File

@ -887,7 +887,12 @@ func TestWatchEtcd(t *testing.T) {
data, err := json.Marshal(manifest)
expectNoError(t, err)
go kubelet.WatchEtcd(watchChannel, updateChannel)
var wg sync.WaitGroup
wg.Add(1)
go func() {
kubelet.WatchEtcd(watchChannel, updateChannel)
wg.Done()
}()
watchChannel <- &etcd.Response{
Node: &etcd.Node{
@ -895,6 +900,7 @@ func TestWatchEtcd(t *testing.T) {
},
}
close(watchChannel)
wg.Wait()
close(updateChannel)
read := reader.GetList()