482 lines
13 KiB
Go
482 lines
13 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 upgrade
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
versionutil "k8s.io/kubernetes/pkg/util/version"
|
|
)
|
|
|
|
type fakeVersionGetter struct {
|
|
clusterVersion, kubeadmVersion, stableVersion, latestVersion, latestDevBranchVersion, stablePatchVersion, kubeletVersion string
|
|
}
|
|
|
|
var _ VersionGetter = &fakeVersionGetter{}
|
|
|
|
// ClusterVersion gets a fake API server version
|
|
func (f *fakeVersionGetter) ClusterVersion() (string, *versionutil.Version, error) {
|
|
return f.clusterVersion, versionutil.MustParseSemantic(f.clusterVersion), nil
|
|
}
|
|
|
|
// KubeadmVersion gets a fake kubeadm version
|
|
func (f *fakeVersionGetter) KubeadmVersion() (string, *versionutil.Version, error) {
|
|
return f.kubeadmVersion, versionutil.MustParseSemantic(f.kubeadmVersion), nil
|
|
}
|
|
|
|
// VersionFromCILabel gets fake latest versions from CI
|
|
func (f *fakeVersionGetter) VersionFromCILabel(ciVersionLabel, _ string) (string, *versionutil.Version, error) {
|
|
if ciVersionLabel == "stable" {
|
|
return f.stableVersion, versionutil.MustParseSemantic(f.stableVersion), nil
|
|
}
|
|
if ciVersionLabel == "latest" {
|
|
return f.latestVersion, versionutil.MustParseSemantic(f.latestVersion), nil
|
|
}
|
|
if ciVersionLabel == "latest-1.8" {
|
|
return f.latestDevBranchVersion, versionutil.MustParseSemantic(f.latestDevBranchVersion), nil
|
|
}
|
|
return f.stablePatchVersion, versionutil.MustParseSemantic(f.stablePatchVersion), nil
|
|
}
|
|
|
|
// KubeletVersions gets the versions of the kubelets in the cluster
|
|
func (f *fakeVersionGetter) KubeletVersions() (map[string]uint16, error) {
|
|
return map[string]uint16{
|
|
f.kubeletVersion: 1,
|
|
}, nil
|
|
}
|
|
|
|
func TestGetAvailableUpgrades(t *testing.T) {
|
|
tests := []struct {
|
|
vg *fakeVersionGetter
|
|
expectedUpgrades []Upgrade
|
|
allowExperimental, allowRCs bool
|
|
errExpected bool
|
|
}{
|
|
{ // no action needed, already up-to-date
|
|
vg: &fakeVersionGetter{
|
|
clusterVersion: "v1.7.3",
|
|
kubeletVersion: "v1.7.3",
|
|
kubeadmVersion: "v1.7.3",
|
|
|
|
stablePatchVersion: "v1.7.3",
|
|
stableVersion: "v1.7.3",
|
|
},
|
|
expectedUpgrades: []Upgrade{},
|
|
allowExperimental: false,
|
|
errExpected: false,
|
|
},
|
|
{ // simple patch version upgrade
|
|
vg: &fakeVersionGetter{
|
|
clusterVersion: "v1.7.1",
|
|
kubeletVersion: "v1.7.1", // the kubelet are on the same version as the control plane
|
|
kubeadmVersion: "v1.7.2",
|
|
|
|
stablePatchVersion: "v1.7.3",
|
|
stableVersion: "v1.7.3",
|
|
},
|
|
expectedUpgrades: []Upgrade{
|
|
{
|
|
Description: "version in the v1.7 series",
|
|
Before: ClusterState{
|
|
KubeVersion: "v1.7.1",
|
|
KubeletVersions: map[string]uint16{
|
|
"v1.7.1": 1,
|
|
},
|
|
KubeadmVersion: "v1.7.2",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
After: ClusterState{
|
|
KubeVersion: "v1.7.3",
|
|
KubeadmVersion: "v1.7.3",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
},
|
|
},
|
|
allowExperimental: false,
|
|
errExpected: false,
|
|
},
|
|
{ // minor version upgrade only
|
|
vg: &fakeVersionGetter{
|
|
clusterVersion: "v1.7.3",
|
|
kubeletVersion: "v1.7.3", // the kubelet are on the same version as the control plane
|
|
kubeadmVersion: "v1.8.0",
|
|
|
|
stablePatchVersion: "v1.7.3",
|
|
stableVersion: "v1.8.0",
|
|
},
|
|
expectedUpgrades: []Upgrade{
|
|
{
|
|
Description: "stable version",
|
|
Before: ClusterState{
|
|
KubeVersion: "v1.7.3",
|
|
KubeletVersions: map[string]uint16{
|
|
"v1.7.3": 1,
|
|
},
|
|
KubeadmVersion: "v1.8.0",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
After: ClusterState{
|
|
KubeVersion: "v1.8.0",
|
|
KubeadmVersion: "v1.8.0",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
},
|
|
},
|
|
allowExperimental: false,
|
|
errExpected: false,
|
|
},
|
|
{ // both minor version upgrade and patch version upgrade available
|
|
vg: &fakeVersionGetter{
|
|
clusterVersion: "v1.7.3",
|
|
kubeletVersion: "v1.7.3", // the kubelet are on the same version as the control plane
|
|
kubeadmVersion: "v1.8.1",
|
|
|
|
stablePatchVersion: "v1.7.5",
|
|
stableVersion: "v1.8.2",
|
|
},
|
|
expectedUpgrades: []Upgrade{
|
|
{
|
|
Description: "version in the v1.7 series",
|
|
Before: ClusterState{
|
|
KubeVersion: "v1.7.3",
|
|
KubeletVersions: map[string]uint16{
|
|
"v1.7.3": 1,
|
|
},
|
|
KubeadmVersion: "v1.8.1",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
After: ClusterState{
|
|
KubeVersion: "v1.7.5",
|
|
KubeadmVersion: "v1.8.1", // Note: The kubeadm version mustn't be "downgraded" here
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
},
|
|
{
|
|
Description: "stable version",
|
|
Before: ClusterState{
|
|
KubeVersion: "v1.7.3",
|
|
KubeletVersions: map[string]uint16{
|
|
"v1.7.3": 1,
|
|
},
|
|
KubeadmVersion: "v1.8.1",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
After: ClusterState{
|
|
KubeVersion: "v1.8.2",
|
|
KubeadmVersion: "v1.8.2",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
},
|
|
},
|
|
allowExperimental: false,
|
|
errExpected: false,
|
|
},
|
|
{ // allow experimental upgrades, but no upgrade available
|
|
vg: &fakeVersionGetter{
|
|
clusterVersion: "v1.8.0-alpha.2",
|
|
kubeletVersion: "v1.7.5",
|
|
kubeadmVersion: "v1.7.5",
|
|
|
|
stablePatchVersion: "v1.7.5",
|
|
stableVersion: "v1.7.5",
|
|
latestVersion: "v1.8.0-alpha.2",
|
|
},
|
|
expectedUpgrades: []Upgrade{},
|
|
allowExperimental: true,
|
|
errExpected: false,
|
|
},
|
|
{ // upgrade to an unstable version should be supported
|
|
vg: &fakeVersionGetter{
|
|
clusterVersion: "v1.7.5",
|
|
kubeletVersion: "v1.7.5",
|
|
kubeadmVersion: "v1.7.5",
|
|
|
|
stablePatchVersion: "v1.7.5",
|
|
stableVersion: "v1.7.5",
|
|
latestVersion: "v1.8.0-alpha.2",
|
|
},
|
|
expectedUpgrades: []Upgrade{
|
|
{
|
|
Description: "experimental version",
|
|
Before: ClusterState{
|
|
KubeVersion: "v1.7.5",
|
|
KubeletVersions: map[string]uint16{
|
|
"v1.7.5": 1,
|
|
},
|
|
KubeadmVersion: "v1.7.5",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
After: ClusterState{
|
|
KubeVersion: "v1.8.0-alpha.2",
|
|
KubeadmVersion: "v1.8.0-alpha.2",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
},
|
|
},
|
|
allowExperimental: true,
|
|
errExpected: false,
|
|
},
|
|
{ // upgrade from an unstable version to an unstable version should be supported
|
|
vg: &fakeVersionGetter{
|
|
clusterVersion: "v1.8.0-alpha.1",
|
|
kubeletVersion: "v1.7.5",
|
|
kubeadmVersion: "v1.7.5",
|
|
|
|
stablePatchVersion: "v1.7.5",
|
|
stableVersion: "v1.7.5",
|
|
latestVersion: "v1.8.0-alpha.2",
|
|
},
|
|
expectedUpgrades: []Upgrade{
|
|
{
|
|
Description: "experimental version",
|
|
Before: ClusterState{
|
|
KubeVersion: "v1.8.0-alpha.1",
|
|
KubeletVersions: map[string]uint16{
|
|
"v1.7.5": 1,
|
|
},
|
|
KubeadmVersion: "v1.7.5",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
After: ClusterState{
|
|
KubeVersion: "v1.8.0-alpha.2",
|
|
KubeadmVersion: "v1.8.0-alpha.2",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
},
|
|
},
|
|
allowExperimental: true,
|
|
errExpected: false,
|
|
},
|
|
{ // v1.X.0-alpha.0 should be ignored
|
|
vg: &fakeVersionGetter{
|
|
clusterVersion: "v1.7.5",
|
|
kubeletVersion: "v1.7.5",
|
|
kubeadmVersion: "v1.7.5",
|
|
|
|
stablePatchVersion: "v1.7.5",
|
|
stableVersion: "v1.7.5",
|
|
latestDevBranchVersion: "v1.8.0-beta.1",
|
|
latestVersion: "v1.9.0-alpha.0",
|
|
},
|
|
expectedUpgrades: []Upgrade{
|
|
{
|
|
Description: "experimental version",
|
|
Before: ClusterState{
|
|
KubeVersion: "v1.7.5",
|
|
KubeletVersions: map[string]uint16{
|
|
"v1.7.5": 1,
|
|
},
|
|
KubeadmVersion: "v1.7.5",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
After: ClusterState{
|
|
KubeVersion: "v1.8.0-beta.1",
|
|
KubeadmVersion: "v1.8.0-beta.1",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
},
|
|
},
|
|
allowExperimental: true,
|
|
errExpected: false,
|
|
},
|
|
{ // upgrade to an RC version should be supported
|
|
vg: &fakeVersionGetter{
|
|
clusterVersion: "v1.7.5",
|
|
kubeletVersion: "v1.7.5",
|
|
kubeadmVersion: "v1.7.5",
|
|
|
|
stablePatchVersion: "v1.7.5",
|
|
stableVersion: "v1.7.5",
|
|
latestDevBranchVersion: "v1.8.0-rc.1",
|
|
latestVersion: "v1.9.0-alpha.1",
|
|
},
|
|
expectedUpgrades: []Upgrade{
|
|
{
|
|
Description: "release candidate version",
|
|
Before: ClusterState{
|
|
KubeVersion: "v1.7.5",
|
|
KubeletVersions: map[string]uint16{
|
|
"v1.7.5": 1,
|
|
},
|
|
KubeadmVersion: "v1.7.5",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
After: ClusterState{
|
|
KubeVersion: "v1.8.0-rc.1",
|
|
KubeadmVersion: "v1.8.0-rc.1",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
},
|
|
},
|
|
allowRCs: true,
|
|
errExpected: false,
|
|
},
|
|
{ // it is possible (but very uncommon) that the latest version from the previous branch is an rc and the current latest version is alpha.0. In that case, show the RC
|
|
vg: &fakeVersionGetter{
|
|
clusterVersion: "v1.7.5",
|
|
kubeletVersion: "v1.7.5",
|
|
kubeadmVersion: "v1.7.5",
|
|
|
|
stablePatchVersion: "v1.7.5",
|
|
stableVersion: "v1.7.5",
|
|
latestDevBranchVersion: "v1.8.0-rc.1",
|
|
latestVersion: "v1.9.0-alpha.0",
|
|
},
|
|
expectedUpgrades: []Upgrade{
|
|
{
|
|
Description: "experimental version", // Note that this is considered an experimental version in this uncommon scenario
|
|
Before: ClusterState{
|
|
KubeVersion: "v1.7.5",
|
|
KubeletVersions: map[string]uint16{
|
|
"v1.7.5": 1,
|
|
},
|
|
KubeadmVersion: "v1.7.5",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
After: ClusterState{
|
|
KubeVersion: "v1.8.0-rc.1",
|
|
KubeadmVersion: "v1.8.0-rc.1",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
},
|
|
},
|
|
allowExperimental: true,
|
|
errExpected: false,
|
|
},
|
|
{ // upgrade to an RC version should be supported. There may also be an even newer unstable version.
|
|
vg: &fakeVersionGetter{
|
|
clusterVersion: "v1.7.5",
|
|
kubeletVersion: "v1.7.5",
|
|
kubeadmVersion: "v1.7.5",
|
|
|
|
stablePatchVersion: "v1.7.5",
|
|
stableVersion: "v1.7.5",
|
|
latestDevBranchVersion: "v1.8.0-rc.1",
|
|
latestVersion: "v1.9.0-alpha.1",
|
|
},
|
|
expectedUpgrades: []Upgrade{
|
|
{
|
|
Description: "release candidate version",
|
|
Before: ClusterState{
|
|
KubeVersion: "v1.7.5",
|
|
KubeletVersions: map[string]uint16{
|
|
"v1.7.5": 1,
|
|
},
|
|
KubeadmVersion: "v1.7.5",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
After: ClusterState{
|
|
KubeVersion: "v1.8.0-rc.1",
|
|
KubeadmVersion: "v1.8.0-rc.1",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
},
|
|
{
|
|
Description: "experimental version",
|
|
Before: ClusterState{
|
|
KubeVersion: "v1.7.5",
|
|
KubeletVersions: map[string]uint16{
|
|
"v1.7.5": 1,
|
|
},
|
|
KubeadmVersion: "v1.7.5",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
After: ClusterState{
|
|
KubeVersion: "v1.9.0-alpha.1",
|
|
KubeadmVersion: "v1.9.0-alpha.1",
|
|
DNSVersion: "1.14.4",
|
|
},
|
|
},
|
|
},
|
|
allowRCs: true,
|
|
allowExperimental: true,
|
|
errExpected: false,
|
|
},
|
|
}
|
|
|
|
for _, rt := range tests {
|
|
|
|
actualUpgrades, actualErr := GetAvailableUpgrades(rt.vg, rt.allowExperimental, rt.allowRCs)
|
|
if !reflect.DeepEqual(actualUpgrades, rt.expectedUpgrades) {
|
|
t.Errorf("failed TestGetAvailableUpgrades\n\texpected upgrades: %v\n\tgot: %v", rt.expectedUpgrades, actualUpgrades)
|
|
}
|
|
if (actualErr != nil) != rt.errExpected {
|
|
t.Errorf("failed TestGetAvailableUpgrades\n\texpected error: %t\n\tgot error: %t", rt.errExpected, (actualErr != nil))
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestKubeletUpgrade(t *testing.T) {
|
|
tests := []struct {
|
|
before map[string]uint16
|
|
after string
|
|
expected bool
|
|
}{
|
|
{ // upgrade available
|
|
before: map[string]uint16{
|
|
"v1.7.1": 1,
|
|
},
|
|
after: "v1.7.3",
|
|
expected: true,
|
|
},
|
|
{ // upgrade available
|
|
before: map[string]uint16{
|
|
"v1.7.1": 1,
|
|
"v1.7.3": 100,
|
|
},
|
|
after: "v1.7.3",
|
|
expected: true,
|
|
},
|
|
{ // upgrade not available
|
|
before: map[string]uint16{
|
|
"v1.7.3": 1,
|
|
},
|
|
after: "v1.7.3",
|
|
expected: false,
|
|
},
|
|
{ // upgrade not available
|
|
before: map[string]uint16{
|
|
"v1.7.3": 100,
|
|
},
|
|
after: "v1.7.3",
|
|
expected: false,
|
|
},
|
|
{ // upgrade not available if we don't know anything about the earlier state
|
|
before: map[string]uint16{},
|
|
after: "v1.7.3",
|
|
expected: false,
|
|
},
|
|
}
|
|
|
|
for _, rt := range tests {
|
|
|
|
upgrade := Upgrade{
|
|
Before: ClusterState{
|
|
KubeletVersions: rt.before,
|
|
},
|
|
After: ClusterState{
|
|
KubeVersion: rt.after,
|
|
},
|
|
}
|
|
actual := upgrade.CanUpgradeKubelets()
|
|
if actual != rt.expected {
|
|
t.Errorf("failed TestKubeletUpgrade\n\texpected: %t\n\tgot: %t\n\ttest object: %v", rt.expected, actual, upgrade)
|
|
}
|
|
}
|
|
}
|