From 9dc74d60f7a758646c5ec74d836ffc460ad340fb Mon Sep 17 00:00:00 2001
From: Maru Newby <marun@redhat.com>
Date: Thu, 13 Apr 2017 14:09:10 -0700
Subject: [PATCH] fed: Add simple upgrade test

---
 hack/.linted_packages                  |  1 -
 test/e2e_federation/upgrade.go         |  2 +-
 test/e2e_federation/upgrades/BUILD     | 13 ++++-
 test/e2e_federation/upgrades/simple.go | 72 ++++++++++++++++++++++++++
 4 files changed, 84 insertions(+), 4 deletions(-)
 create mode 100644 test/e2e_federation/upgrades/simple.go

diff --git a/hack/.linted_packages b/hack/.linted_packages
index bd85c8e10f84..6e29dc246e56 100644
--- a/hack/.linted_packages
+++ b/hack/.linted_packages
@@ -397,7 +397,6 @@ staging/src/k8s.io/kube-aggregator/pkg/controllers
 staging/src/k8s.io/sample-apiserver
 staging/src/k8s.io/sample-apiserver/pkg/apis/wardle/install
 test/e2e/perftype
-test/e2e_federation/upgrades
 test/e2e_node/runner/local
 test/images/clusterapi-tester
 test/images/entrypoint-tester
diff --git a/test/e2e_federation/upgrade.go b/test/e2e_federation/upgrade.go
index 6b893067e1a4..677f837afec2 100644
--- a/test/e2e_federation/upgrade.go
+++ b/test/e2e_federation/upgrade.go
@@ -25,7 +25,7 @@ import (
 	. "github.com/onsi/ginkgo"
 )
 
-var upgradeTests = []upgrades.Test{}
+var upgradeTests = upgrades.SimpleUpgradeTests()
 
 var _ = framework.KubeDescribe("Upgrade [Feature:Upgrade]", func() {
 	f := fedframework.NewDefaultFederatedFramework("federation-upgrade")
diff --git a/test/e2e_federation/upgrades/BUILD b/test/e2e_federation/upgrades/BUILD
index 4296f48f5f4b..e187b3607a85 100644
--- a/test/e2e_federation/upgrades/BUILD
+++ b/test/e2e_federation/upgrades/BUILD
@@ -9,9 +9,18 @@ load(
 
 go_library(
     name = "go_default_library",
-    srcs = ["upgrade.go"],
+    srcs = [
+        "simple.go",
+        "upgrade.go",
+    ],
     tags = ["automanaged"],
-    deps = ["//test/e2e_federation/framework:go_default_library"],
+    deps = [
+        "//federation/pkg/federatedtypes:go_default_library",
+        "//federation/pkg/federatedtypes/crudtester:go_default_library",
+        "//test/e2e_federation/framework:go_default_library",
+        "//vendor/github.com/onsi/ginkgo:go_default_library",
+        "//vendor/k8s.io/apimachinery/pkg/runtime:go_default_library",
+    ],
 )
 
 filegroup(
diff --git a/test/e2e_federation/upgrades/simple.go b/test/e2e_federation/upgrades/simple.go
new file mode 100644
index 000000000000..8d51c59abe88
--- /dev/null
+++ b/test/e2e_federation/upgrades/simple.go
@@ -0,0 +1,72 @@
+/*
+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 upgrades
+
+import (
+	"fmt"
+
+	pkgruntime "k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/kubernetes/federation/pkg/federatedtypes"
+	crudtester "k8s.io/kubernetes/federation/pkg/federatedtypes/crudtester"
+	fedframework "k8s.io/kubernetes/test/e2e_federation/framework"
+
+	. "github.com/onsi/ginkgo"
+)
+
+// SimpleUpgradeTest validates that a federated resource remains
+// propagated before and after a control plane upgrade
+type SimpleUpgradeTest struct {
+	kind           string
+	adapterFactory federatedtypes.AdapterFactory
+	crudTester     *crudtester.FederatedTypeCRUDTester
+	obj            pkgruntime.Object
+}
+
+// Setup creates a resource and validates its propagation to member clusters
+func (ut *SimpleUpgradeTest) Setup(f *fedframework.Framework) {
+	adapter := ut.adapterFactory(f.FederationClientset)
+	clients := f.GetClusterClients()
+	ut.crudTester = fedframework.NewFederatedTypeCRUDTester(adapter, clients)
+
+	By(fmt.Sprintf("Creating a resource of kind %q and validating propagation to member clusters", ut.kind))
+	obj := adapter.NewTestObject(f.Namespace.Name)
+	ut.obj = ut.crudTester.CheckCreate(obj)
+}
+
+// Test validates that a resource remains propagated post-upgrade
+func (ut *SimpleUpgradeTest) Test(f *fedframework.Framework, done <-chan struct{}, upgrade FederationUpgradeType) {
+	<-done
+	By(fmt.Sprintf("Validating that a resource of kind %q remains propagated to member clusters after upgrade", ut.kind))
+	ut.crudTester.CheckPropagation(ut.obj)
+}
+
+// Teardown cleans up remaining resources
+func (ut *SimpleUpgradeTest) Teardown(f *fedframework.Framework) {
+	// Rely on the namespace deletion to clean up everything
+}
+
+// SimpleUpgradeTests collects simple upgrade tests for registered federated types
+func SimpleUpgradeTests() []Test {
+	tests := []Test{}
+	for kind, fedType := range federatedtypes.FederatedTypes() {
+		tests = append(tests, &SimpleUpgradeTest{
+			kind:           kind,
+			adapterFactory: fedType.AdapterFactory,
+		})
+	}
+	return tests
+}