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

Use PollUntilContextCancel to replace PollUntil in test

Browse Source 
Signed-off-by: bzsuni <bingzhe.sun@daocloud.io>
This commit is contained in:
bzsuni 2024-06-29 19:28:24 +08:00
parent a7832cc169
commit b7dc31760f
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
test/integration/apiserver/flowcontrol/fs_condition_test.go
View File

@ -17,6 +17,7 @@ limitations under the License.
package flowcontrol package flowcontrol
import ( import (
"context"
"encoding/json" "encoding/json"
"testing" "testing"
"time" "time"
@ -42,7 +43,7 @@ func TestConditionIsolation(t *testing.T) {
fsClient := loopbackClient.FlowcontrolV1().FlowSchemas() fsClient := loopbackClient.FlowcontrolV1().FlowSchemas()
var dangleOrig *flowcontrol.FlowSchemaCondition var dangleOrig *flowcontrol.FlowSchemaCondition
wait.PollUntil(time.Second, func() (bool, error) { err := wait.PollUntilContextCancel(ctx, time.Second, false, func(ctx context.Context) (done bool, err error) {
fsGot, err := fsClient.Get(ctx, fsOrig.Name, metav1.GetOptions{}) fsGot, err := fsClient.Get(ctx, fsOrig.Name, metav1.GetOptions{})
if err != nil { if err != nil {
klog.Errorf("Failed to fetch FlowSchema %q: %v", fsOrig.Name, err) klog.Errorf("Failed to fetch FlowSchema %q: %v", fsOrig.Name, err)
@ -50,7 +51,10 @@ func TestConditionIsolation(t *testing.T) {
} }
dangleOrig = getCondition(fsGot.Status.Conditions, flowcontrol.FlowSchemaConditionDangling) dangleOrig = getCondition(fsGot.Status.Conditions, flowcontrol.FlowSchemaConditionDangling)
return dangleOrig != nil, nil return dangleOrig != nil, nil
}, ctx.Done()) })
if err != nil {
t.Error(err)
}
ssaType := flowcontrol.FlowSchemaConditionType("test-ssa") ssaType := flowcontrol.FlowSchemaConditionType("test-ssa")
patchSSA := flowcontrolapply.FlowSchema(fsOrig.Name). patchSSA := flowcontrolapply.FlowSchema(fsOrig.Name).