mirror of https://github.com/knative/pkg.git
91 lines
2.4 KiB
Go
91 lines
2.4 KiB
Go
/*
|
|
Copyright 2020 The Knative Authors
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package reconciler
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"k8s.io/apimachinery/pkg/types"
|
|
)
|
|
|
|
func TestLeaderAwareFuncs(t *testing.T) {
|
|
laf := LeaderAwareFuncs{}
|
|
wantBkt := UniversalBucket()
|
|
wantKey := types.NamespacedName{
|
|
Namespace: "foo",
|
|
Name: "bar",
|
|
}
|
|
called := false
|
|
wantFunc := func(gotBkt Bucket, gotKey types.NamespacedName) {
|
|
called = true
|
|
if !cmp.Equal(gotKey, wantKey) {
|
|
t.Error("key (-want, +got) =", cmp.Diff(wantKey, gotKey))
|
|
}
|
|
if !cmp.Equal(gotBkt, wantBkt) {
|
|
t.Error("bucket (-want, +got) =", cmp.Diff(wantBkt, gotBkt))
|
|
}
|
|
}
|
|
|
|
laf.PromoteFunc = func(bkt Bucket, gotFunc func(Bucket, types.NamespacedName)) error {
|
|
if gotFunc == nil {
|
|
t.Error("PromoteFunc should not be called with nil gotFunc")
|
|
return nil
|
|
}
|
|
|
|
gotFunc(bkt, wantKey)
|
|
if !called {
|
|
t.Error("gotFunc didn't call wantFunc!")
|
|
}
|
|
|
|
// IsLeaderFor takes the bucket's lock, so make sure that the callback
|
|
// we provide is not called while the lock is still held by calling a
|
|
// function that we know takes the lock.
|
|
if !laf.IsLeaderFor(wantKey) {
|
|
t.Error("IsLeaderFor() = false, wanted true")
|
|
}
|
|
return nil
|
|
}
|
|
laf.DemoteFunc = func(bkt Bucket) {
|
|
// Check that we're not called while the lock is held,
|
|
// and that we are no longer leader.
|
|
if laf.IsLeaderFor(wantKey) {
|
|
t.Error("IsLeaderFor() = true, wanted false")
|
|
}
|
|
}
|
|
|
|
// We don't start as leader.
|
|
if laf.IsLeaderFor(wantKey) {
|
|
t.Error("IsLeaderFor() = true, wanted false")
|
|
}
|
|
|
|
// After Promote we are leader.
|
|
laf.Promote(wantBkt, wantFunc)
|
|
if !laf.IsLeaderFor(wantKey) {
|
|
t.Error("IsLeaderFor() = false, wanted true")
|
|
}
|
|
|
|
// After Demote we are not leader.
|
|
laf.Demote(wantBkt)
|
|
if laf.IsLeaderFor(wantKey) {
|
|
t.Error("IsLeaderFor() = true, wanted false")
|
|
}
|
|
|
|
// We should elide calls with a nil enq
|
|
laf.Promote(wantBkt, nil)
|
|
}
|