Pass context to tryAcquireOrRenew

Kubernetes-commit: c049f30ef2a83172f46a587ddaf2104b39df8301
This commit is contained in:
Ted Yu
2020-02-10 12:31:24 -08:00
committed by Kubernetes Publisher
parent 200280e336
commit d6bfb32c17
8 changed files with 48 additions and 55 deletions

View File

@@ -17,6 +17,7 @@ limitations under the License.
package leaderelection
import (
"context"
"encoding/json"
"fmt"
"sync"
@@ -296,7 +297,7 @@ func testTryAcquireOrRenew(t *testing.T, objectType string) {
observedTime: test.observedTime,
clock: clock.RealClock{},
}
if test.expectSuccess != le.tryAcquireOrRenew() {
if test.expectSuccess != le.tryAcquireOrRenew(context.Background()) {
t.Errorf("unexpected result of tryAcquireOrRenew: [succeeded=%v]", !test.expectSuccess)
}
@@ -879,7 +880,7 @@ func testTryAcquireOrRenewMultiLock(t *testing.T, objectType string) {
observedTime: test.observedTime,
clock: clock.RealClock{},
}
if test.expectSuccess != le.tryAcquireOrRenew() {
if test.expectSuccess != le.tryAcquireOrRenew(context.Background()) {
t.Errorf("unexpected result of tryAcquireOrRenew: [succeeded=%v]", !test.expectSuccess)
}