From 423e5b24b0ff62d5a2711268d253634ede66825d Mon Sep 17 00:00:00 2001
From: j4ckstraw <j4ckstraw@foxmail.com>
Date: Mon, 5 Feb 2024 17:56:51 +0800
Subject: [PATCH] fix koordlet config test

Signed-off-by: j4ckstraw <j4ckstraw@foxmail.com>
---
 pkg/koordlet/qosmanager/framework/config_test.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/pkg/koordlet/qosmanager/framework/config_test.go b/pkg/koordlet/qosmanager/framework/config_test.go
index 5ebb97836..122bf2f04 100644
--- a/pkg/koordlet/qosmanager/framework/config_test.go
+++ b/pkg/koordlet/qosmanager/framework/config_test.go
@@ -31,6 +31,7 @@ func Test_NewDefaultConfig(t *testing.T) {
 		MemoryEvictIntervalSeconds: 1,
 		MemoryEvictCoolTimeSeconds: 4,
 		CPUEvictCoolTimeSeconds:    20,
+		OnlyEvictByAPI:             false,
 		QOSExtensionCfg:            &QOSExtensionConfig{FeatureGates: map[string]bool{}},
 	}
 	defaultConfig := NewDefaultConfig()
@@ -47,6 +48,7 @@ func Test_InitFlags(t *testing.T) {
 		"--memory-evict-cool-time-seconds=8",
 		"--cpu-evict-cool-time-seconds=40",
 		"--qos-extension-plugins=test-plugin=true",
+		"--only-evict-by-api=false",
 	}
 	fs := flag.NewFlagSet(cmdArgs[0], flag.ExitOnError)
 
@@ -57,6 +59,7 @@ func Test_InitFlags(t *testing.T) {
 		MemoryEvictIntervalSeconds int
 		MemoryEvictCoolTimeSeconds int
 		CPUEvictCoolTimeSeconds    int
+		OnlyEvictByAPI             bool
 		QOSExtensionCfg            *QOSExtensionConfig
 	}
 	type args struct {
@@ -76,6 +79,7 @@ func Test_InitFlags(t *testing.T) {
 				MemoryEvictIntervalSeconds: 2,
 				MemoryEvictCoolTimeSeconds: 8,
 				CPUEvictCoolTimeSeconds:    40,
+				OnlyEvictByAPI:             false,
 				QOSExtensionCfg:            &QOSExtensionConfig{FeatureGates: map[string]bool{"test-plugin": true}},
 			},
 			args: args{fs: fs},
@@ -90,6 +94,7 @@ func Test_InitFlags(t *testing.T) {
 				MemoryEvictIntervalSeconds: tt.fields.MemoryEvictIntervalSeconds,
 				MemoryEvictCoolTimeSeconds: tt.fields.MemoryEvictCoolTimeSeconds,
 				CPUEvictCoolTimeSeconds:    tt.fields.CPUEvictCoolTimeSeconds,
+				OnlyEvictByAPI:             tt.fields.OnlyEvictByAPI,
 				QOSExtensionCfg:            tt.fields.QOSExtensionCfg,
 			}
 			c := NewDefaultConfig()