Skip to content

Commit d0579b6

Browse files
authored
Merge pull request kubernetes#125683 from likakuli/fix-benchmarkupdatesnapshot
clean: add nodeinfo to cache
2 parents beb48b7 + ea6ca27 commit d0579b6

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

pkg/scheduler/internal/cache/cache_test.go

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2050,6 +2050,7 @@ func setupCacheOf1kNodes30kPods(b *testing.B) Cache {
20502050
cache := newCache(ctx, time.Second, time.Second)
20512051
for i := 0; i < 1000; i++ {
20522052
nodeName := fmt.Sprintf("node-%d", i)
2053+
cache.AddNode(logger, st.MakeNode().Name(nodeName).Obj())
20532054
for j := 0; j < 30; j++ {
20542055
objName := fmt.Sprintf("%s-pod-%d", nodeName, j)
20552056
pod := makeBasePod(b, nodeName, objName, "0", "0", "", nil)
@@ -2065,10 +2066,15 @@ func setupCacheOf1kNodes30kPods(b *testing.B) Cache {
20652066
func setupCacheWithAssumedPods(b *testing.B, podNum int, assumedTime time.Time) *cacheImpl {
20662067
logger, ctx := ktesting.NewTestContext(b)
20672068
ctx, cancel := context.WithCancel(ctx)
2069+
addedNodes := make(map[string]struct{})
20682070
defer cancel()
20692071
cache := newCache(ctx, time.Second, time.Second)
20702072
for i := 0; i < podNum; i++ {
20712073
nodeName := fmt.Sprintf("node-%d", i/10)
2074+
if _, ok := addedNodes[nodeName]; !ok {
2075+
cache.AddNode(logger, st.MakeNode().Name(nodeName).Obj())
2076+
addedNodes[nodeName] = struct{}{}
2077+
}
20722078
objName := fmt.Sprintf("%s-pod-%d", nodeName, i%10)
20732079
pod := makeBasePod(b, nodeName, objName, "0", "0", "", nil)
20742080

0 commit comments

Comments
 (0)