github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-29 14:37:00 +00:00
Code Issues Projects Releases Wiki Activity

fix scheduling queue unit test

Browse Source 
This change makes sure the Pop() test finish completely.
This commit is contained in:
Cao Shufeng 2017-11-28 17:40:35 +08:00
parent 24042ce6e7
commit 33f6625a84
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
plugin/pkg/scheduler/core/scheduling_queue_test.go
View File

@ -18,6 +18,7 @@ package core
import ( import (
"reflect" "reflect"
"sync"
"testing" "testing"
"k8s.io/api/core/v1" "k8s.io/api/core/v1"
@ -87,12 +88,16 @@ func TestPriorityQueue_Add(t *testing.T) {
func TestPriorityQueue_Pop(t *testing.T) { func TestPriorityQueue_Pop(t *testing.T) {
q := NewPriorityQueue() q := NewPriorityQueue()
wg := sync.WaitGroup{}
wg.Add(1)
go func() { go func() {
defer wg.Done()
if p, err := q.Pop(); err != nil || p != &highPriorityPod { if p, err := q.Pop(); err != nil || p != &highPriorityPod {
t.Errorf("Expected: %v after Pop, but got: %v", highPriorityPod.Name, p.Name) t.Errorf("Expected: %v after Pop, but got: %v", highPriorityPod.Name, p.Name)
} }
}() }()
q.Add(&highPriorityPod) q.Add(&highPriorityPod)
wg.Wait()
} }
func TestPriorityQueue_Update(t *testing.T) { func TestPriorityQueue_Update(t *testing.T) {