114 lines
4.4 KiB
Go
114 lines
4.4 KiB
Go
/*
|
|
Copyright 2017 The Kubernetes 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 federation
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/pborman/uuid"
|
|
|
|
pkgruntime "k8s.io/apimachinery/pkg/runtime"
|
|
federationapi "k8s.io/kubernetes/federation/apis/federation/v1beta1"
|
|
"k8s.io/kubernetes/federation/pkg/federatedtypes"
|
|
"k8s.io/kubernetes/federation/pkg/federatedtypes/crudtester"
|
|
"k8s.io/kubernetes/test/integration/federation/framework"
|
|
)
|
|
|
|
// TestFederationCRUD validates create/read/update/delete operations for federated resource types.
|
|
func TestFederationCRUD(t *testing.T) {
|
|
fedFixture := framework.FederationFixture{DesiredClusterCount: 2}
|
|
fedFixture.SetUp(t)
|
|
defer fedFixture.TearDown(t)
|
|
|
|
federatedTypes := federatedtypes.FederatedTypes()
|
|
for kind, fedType := range federatedTypes {
|
|
t.Run(kind, func(t *testing.T) {
|
|
fixture, crudTester, obj, _ := initCRUDTest(t, &fedFixture, fedType.AdapterFactory, kind)
|
|
defer fixture.TearDown(t)
|
|
|
|
crudTester.CheckLifecycle(obj)
|
|
})
|
|
}
|
|
|
|
// The following tests target a single type since the underlying logic is common across all types.
|
|
kind := federatedtypes.SecretKind
|
|
adapterFactory := federatedtypes.NewSecretAdapter
|
|
|
|
// Validate deletion handling where orphanDependents is true or nil
|
|
orphanedDependents := true
|
|
testCases := map[string]*bool{
|
|
"Resource should not be deleted from underlying clusters when OrphanDependents is true": &orphanedDependents,
|
|
"Resource should not be deleted from underlying clusters when OrphanDependents is nil": nil,
|
|
}
|
|
for testName, orphanDependents := range testCases {
|
|
t.Run(testName, func(t *testing.T) {
|
|
fixture, crudTester, obj, _ := initCRUDTest(t, &fedFixture, adapterFactory, kind)
|
|
defer fixture.TearDown(t)
|
|
|
|
updatedObj := crudTester.CheckCreate(obj)
|
|
crudTester.CheckDelete(updatedObj, orphanDependents)
|
|
})
|
|
}
|
|
|
|
t.Run("Resource should be propagated to a newly added cluster", func(t *testing.T) {
|
|
fixture, crudTester, obj, _ := initCRUDTest(t, &fedFixture, adapterFactory, kind)
|
|
defer fixture.TearDown(t)
|
|
|
|
updatedObj := crudTester.CheckCreate(obj)
|
|
// Start a new cluster and validate that the resource is propagated to it.
|
|
fedFixture.StartCluster(t)
|
|
// Check propagation to the new cluster by providing the updated set of clients
|
|
objectExpected := true
|
|
crudTester.CheckPropagationForClients(updatedObj, fedFixture.ClusterClients, objectExpected)
|
|
})
|
|
|
|
t.Run("Resource should only be propagated to the cluster with a matching selector", func(t *testing.T) {
|
|
fixture, crudTester, obj, adapter := initCRUDTest(t, &fedFixture, adapterFactory, kind)
|
|
defer fixture.TearDown(t)
|
|
|
|
// Set an annotation to specify that the object is isolated to cluster 1.
|
|
federatedtypes.SetAnnotation(adapter, obj, federationapi.FederationClusterSelectorAnnotation, `[{"key": "cluster", "operator": "==", "values": ["1"]}]`)
|
|
|
|
updatedObj := crudTester.Create(obj)
|
|
|
|
// Check propagation to the first cluster
|
|
objectExpected := true
|
|
crudTester.CheckPropagationForClients(updatedObj, fedFixture.ClusterClients[0:1], objectExpected)
|
|
|
|
// Verify the object is not sent to the second cluster
|
|
objectExpected = false
|
|
crudTester.CheckPropagationForClients(updatedObj, fedFixture.ClusterClients[1:2], objectExpected)
|
|
|
|
})
|
|
}
|
|
|
|
// initCRUDTest initializes common elements of a crud test
|
|
func initCRUDTest(t *testing.T, fedFixture *framework.FederationFixture, adapterFactory federatedtypes.AdapterFactory, kind string) (
|
|
*framework.ControllerFixture, *crudtester.FederatedTypeCRUDTester, pkgruntime.Object, federatedtypes.FederatedTypeAdapter) {
|
|
config := fedFixture.APIFixture.NewConfig()
|
|
fixture := framework.NewControllerFixture(t, kind, adapterFactory, config)
|
|
|
|
client := fedFixture.APIFixture.NewClient(fmt.Sprintf("crud-test-%s", kind))
|
|
adapter := adapterFactory(client, config, nil)
|
|
|
|
crudTester := framework.NewFederatedTypeCRUDTester(t, adapter, fedFixture.ClusterClients)
|
|
|
|
obj := adapter.NewTestObject(uuid.New())
|
|
|
|
return fixture, crudTester, obj, adapter
|
|
}
|