mirror of
https://github.com/open-cluster-management-io/ocm.git
synced 2026-02-14 18:09:57 +00:00
e810520961
* Refactor code to fix lint warning Signed-off-by: Jian Qiu <jqiu@redhat.com> * enable lint for testing files Signed-off-by: Jian Qiu <jqiu@redhat.com> --------- Signed-off-by: Jian Qiu <jqiu@redhat.com>
95 lines
3.6 KiB
Go
95 lines
3.6 KiB
Go
package lease
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/openshift/library-go/pkg/operator/events/eventstesting"
|
|
coordinationv1 "k8s.io/api/coordination/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
kubefake "k8s.io/client-go/kubernetes/fake"
|
|
clienttesting "k8s.io/client-go/testing"
|
|
|
|
clusterfake "open-cluster-management.io/api/client/cluster/clientset/versioned/fake"
|
|
clusterinformers "open-cluster-management.io/api/client/cluster/informers/externalversions"
|
|
|
|
testingcommon "open-cluster-management.io/ocm/pkg/common/testing"
|
|
testinghelpers "open-cluster-management.io/ocm/pkg/registration/helpers/testing"
|
|
)
|
|
|
|
func TestLeaseUpdate(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
clusters []runtime.Object
|
|
validateActions func(t *testing.T, actions []clienttesting.Action)
|
|
needToStartUpdateBefore bool
|
|
expectedErr string
|
|
}{
|
|
{
|
|
name: "start lease update routine",
|
|
clusters: []runtime.Object{testinghelpers.NewAcceptedManagedCluster()},
|
|
validateActions: func(t *testing.T, actions []clienttesting.Action) {
|
|
testingcommon.AssertUpdateActions(t, actions)
|
|
leaseObj := actions[1].(clienttesting.UpdateActionImpl).Object
|
|
lastLeaseObj := actions[len(actions)-1].(clienttesting.UpdateActionImpl).Object
|
|
testinghelpers.AssertLeaseUpdated(t, leaseObj.(*coordinationv1.Lease), lastLeaseObj.(*coordinationv1.Lease))
|
|
},
|
|
},
|
|
{
|
|
name: "delete a managed cluster after lease update routine is started",
|
|
clusters: []runtime.Object{},
|
|
needToStartUpdateBefore: true,
|
|
validateActions: testingcommon.AssertNoMoreUpdates,
|
|
expectedErr: "unable to get managed cluster \"testmanagedcluster\" from hub: " +
|
|
"managedcluster.cluster.open-cluster-management.io \"testmanagedcluster\" not found",
|
|
},
|
|
{
|
|
name: "unaccept a managed cluster after lease update routine is started",
|
|
clusters: []runtime.Object{testinghelpers.NewManagedCluster()},
|
|
needToStartUpdateBefore: true,
|
|
validateActions: testingcommon.AssertNoMoreUpdates,
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
clusterClient := clusterfake.NewSimpleClientset(c.clusters...)
|
|
clusterInformerFactory := clusterinformers.NewSharedInformerFactory(clusterClient, time.Minute*10)
|
|
clusterStore := clusterInformerFactory.Cluster().V1().ManagedClusters().Informer().GetStore()
|
|
for _, cluster := range c.clusters {
|
|
if err := clusterStore.Add(cluster); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
hubClient := kubefake.NewSimpleClientset(testinghelpers.NewManagedClusterLease("managed-cluster-lease", time.Now()))
|
|
|
|
leaseUpdater := &leaseUpdater{
|
|
hubClient: hubClient,
|
|
clusterName: testinghelpers.TestManagedClusterName,
|
|
leaseName: "managed-cluster-lease",
|
|
recorder: eventstesting.NewTestingEventRecorder(t),
|
|
}
|
|
|
|
if c.needToStartUpdateBefore {
|
|
leaseUpdater.start(context.TODO(), time.Duration(testinghelpers.TestLeaseDurationSeconds)*time.Second)
|
|
// wait a few milliseconds to start the lease update routine
|
|
time.Sleep(500 * time.Millisecond)
|
|
}
|
|
|
|
ctrl := &managedClusterLeaseController{
|
|
clusterName: testinghelpers.TestManagedClusterName,
|
|
hubClusterLister: clusterInformerFactory.Cluster().V1().ManagedClusters().Lister(),
|
|
leaseUpdater: leaseUpdater,
|
|
}
|
|
syncErr := ctrl.sync(context.TODO(), testingcommon.NewFakeSyncContext(t, ""))
|
|
testingcommon.AssertError(t, syncErr, c.expectedErr)
|
|
|
|
// wait one cycle (1 ~ 1.25s)
|
|
time.Sleep(2000 * time.Millisecond)
|
|
c.validateActions(t, hubClient.Actions())
|
|
})
|
|
}
|
|
}
|