diff --git a/test/e2e/framework/BUILD b/test/e2e/framework/BUILD
index a3fa78a6ad7..51d8cb4943b 100644
--- a/test/e2e/framework/BUILD
+++ b/test/e2e/framework/BUILD
@@ -32,6 +32,7 @@ go_library(
         "service_util.go",
         "statefulset_utils.go",
         "test_context.go",
+        "upgrade_util.go",
         "util.go",
     ],
     tags = ["automanaged"],
diff --git a/test/e2e_federation/BUILD b/test/e2e_federation/BUILD
index f403d60b718..2fecaefc4a8 100644
--- a/test/e2e_federation/BUILD
+++ b/test/e2e_federation/BUILD
@@ -20,6 +20,7 @@ go_library(
         "replicaset.go",
         "secret.go",
         "service.go",
+        "upgrade.go",
         "util.go",
     ],
     tags = ["automanaged"],
@@ -34,9 +35,11 @@ go_library(
         "//pkg/api/v1:go_default_library",
         "//pkg/apis/extensions/v1beta1:go_default_library",
         "//pkg/client/clientset_generated/clientset:go_default_library",
+        "//test/e2e/chaosmonkey:go_default_library",
         "//test/e2e/common:go_default_library",
         "//test/e2e/framework:go_default_library",
         "//test/e2e_federation/framework:go_default_library",
+        "//test/e2e_federation/upgrades:go_default_library",
         "//vendor:github.com/onsi/ginkgo",
         "//vendor:github.com/onsi/gomega",
         "//vendor:k8s.io/apimachinery/pkg/api/errors",
@@ -63,6 +66,7 @@ filegroup(
     srcs = [
         ":package-srcs",
         "//test/e2e_federation/framework:all-srcs",
+        "//test/e2e_federation/upgrades:all-srcs",
     ],
     tags = ["automanaged"],
 )
diff --git a/test/e2e_federation/upgrades/BUILD b/test/e2e_federation/upgrades/BUILD
new file mode 100644
index 00000000000..4296f48f5f4
--- /dev/null
+++ b/test/e2e_federation/upgrades/BUILD
@@ -0,0 +1,28 @@
+package(default_visibility = ["//visibility:public"])
+
+licenses(["notice"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["upgrade.go"],
+    tags = ["automanaged"],
+    deps = ["//test/e2e_federation/framework:go_default_library"],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+)