github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-10 04:27:54 +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
import (
"context"
"encoding/json"
"testing"
"time"
@ -42,7 +43,7 @@ func TestConditionIsolation(t *testing.T) {
fsClient := loopbackClient.FlowcontrolV1().FlowSchemas()
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{})
if err != nil {
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)
return dangleOrig != nil, nil
}, ctx.Done())
})
if err != nil {
t.Error(err)
}
ssaType := flowcontrol.FlowSchemaConditionType("test-ssa")
patchSSA := flowcontrolapply.FlowSchema(fsOrig.Name).