Skip to content

Commit

Permalink
Use v1beta1 API in E2E tests
Browse files Browse the repository at this point in the history
  • Loading branch information
xmudrii committed Jun 8, 2020
1 parent 3b39f37 commit 1de04c2
Show file tree
Hide file tree
Showing 5 changed files with 65 additions and 43 deletions.
49 changes: 49 additions & 0 deletions pkg/apis/kubeone/v1beta1/helpers.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
/*
Copyright 2020 The KubeOne 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 v1beta1

import (
"github.com/pkg/errors"
)

// SetCloudProvider parses the string representation of the provider
// name and sets the appropriate CloudProviderSpec field.
func SetCloudProvider(cp *CloudProviderSpec, name string) error {
switch name {
case "aws":
cp.AWS = &AWSSpec{}
case "azure":
cp.Azure = &AzureSpec{}
case "digitalocean":
cp.DigitalOcean = &DigitalOceanSpec{}
case "gce":
cp.GCE = &GCESpec{}
case "hetzner":
cp.Hetzner = &HetznerSpec{}
case "openstack":
cp.Openstack = &OpenstackSpec{}
case "packet":
cp.Packet = &PacketSpec{}
case "vsphere":
cp.Vsphere = &VsphereSpec{}
case "none":
cp.None = &NoneSpec{}
default:
return errors.Errorf("provider %q is not supported", name)
}
return nil
}
29 changes: 1 addition & 28 deletions pkg/terraform/v1beta1/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ func (c *Config) Apply(cluster *kubeonev1beta1.KubeOneCluster) error {
cp := c.KubeOneHosts.Value.ControlPlane

if cp.CloudProvider != nil {
if err := setCloudProvider(&cluster.CloudProvider, *cp.CloudProvider); err != nil {
if err := kubeonev1beta1.SetCloudProvider(&cluster.CloudProvider, *cp.CloudProvider); err != nil {
return errors.Wrap(err, "failed to set cloud provider")
}
}
Expand Down Expand Up @@ -459,33 +459,6 @@ func (c *Config) updateVSphereWorkerset(existingWorkerSet *kubeonev1beta1.Dynami
return nil
}

func setCloudProvider(cp *kubeonev1beta1.CloudProviderSpec, name string) error {
switch name {
case "aws":
cp.AWS = &kubeonev1beta1.AWSSpec{}
case "azure":
cp.Azure = &kubeonev1beta1.AzureSpec{}
case "digitalocean":
cp.DigitalOcean = &kubeonev1beta1.DigitalOceanSpec{}
case "gce":
cp.GCE = &kubeonev1beta1.GCESpec{}
case "hetzner":
cp.Hetzner = &kubeonev1beta1.HetznerSpec{}
case "openstack":
cp.Openstack = &kubeonev1beta1.OpenstackSpec{}
case "packet":
cp.Packet = &kubeonev1beta1.PacketSpec{}
case "vsphere":
cp.Vsphere = &kubeonev1beta1.VsphereSpec{}
case "none":
cp.None = &kubeonev1beta1.NoneSpec{}
default:
return errors.Errorf("provider %q is not supported", name)
}

return nil
}

func setWorkersetFlag(w *kubeonev1beta1.DynamicWorkerConfig, name string, value interface{}) error {
// ignore empty values (i.e. not set in terraform output)
switch s := value.(type) {
Expand Down
11 changes: 5 additions & 6 deletions test/e2e/conformance_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,6 @@ import (
"testing"
"time"

k1api "github.com/kubermatic/kubeone/pkg/apis/kubeone/v1alpha1"
"github.com/kubermatic/kubeone/test/e2e/provisioner"
"github.com/kubermatic/kubeone/test/e2e/testutil"

Expand All @@ -40,7 +39,7 @@ const (
func TestClusterConformance(t *testing.T) {
testcases := []struct {
name string
provider k1api.CloudProviderName
provider string
providerExternal bool
scenario string
configFilePath string
Expand Down Expand Up @@ -102,7 +101,7 @@ func TestClusterConformance(t *testing.T) {
t.Run(tc.name, func(t *testing.T) {
// Only run selected test suite.
// Test options are controlled using flags.
if testProvider != string(tc.provider) {
if testProvider != tc.provider {
t.SkipNow()
}

Expand All @@ -129,7 +128,7 @@ func TestClusterConformance(t *testing.T) {
// Create provisioner
testPath := fmt.Sprintf("../../_build/%s", testRunIdentifier)

pr, err := provisioner.CreateProvisioner(testPath, testRunIdentifier, string(tc.provider))
pr, err := provisioner.CreateProvisioner(testPath, testRunIdentifier, tc.provider)
if err != nil {
t.Fatalf("failed to create provisioner: %v", err)
}
Expand Down Expand Up @@ -172,7 +171,7 @@ func TestClusterConformance(t *testing.T) {
args := []string{}

if osControlPlane != OperatingSystemDefault {
tfFlags, errFlags := ControlPlaneImageFlags(string(tc.provider), osControlPlane)
tfFlags, errFlags := ControlPlaneImageFlags(tc.provider, osControlPlane)
if errFlags != nil {
t.Fatalf("failed to discover control plane os image: %v", errFlags)
}
Expand Down Expand Up @@ -224,7 +223,7 @@ func TestClusterConformance(t *testing.T) {
args = []string{}

if osControlPlane != OperatingSystemDefault {
tfFlags, errFlags := ControlPlaneImageFlags(string(tc.provider), osControlPlane)
tfFlags, errFlags := ControlPlaneImageFlags(tc.provider, osControlPlane)
if errFlags != nil {
t.Fatalf("failed to discover control plane os image: %v", errFlags)
}
Expand Down
12 changes: 7 additions & 5 deletions test/e2e/kubeone.go
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ import (
"github.com/pkg/errors"
"gopkg.in/yaml.v2"

k1api "github.com/kubermatic/kubeone/pkg/apis/kubeone/v1alpha1"
k1api "github.com/kubermatic/kubeone/pkg/apis/kubeone/v1beta1"
"github.com/kubermatic/kubeone/test/e2e/testutil"

metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
Expand All @@ -51,25 +51,27 @@ func NewKubeone(kubeoneDir, configurationFilePath string) *Kubeone {
// CreateConfig creates a KubeOneCluster manifest
func (k1 *Kubeone) CreateConfig(
kubernetesVersion string,
providerName k1api.CloudProviderName,
providerName string,
providerExternal bool,
clusterNetworkPod string,
clusterNetworkService string,
credentialsFile string,
) error {
k1Cluster := k1api.KubeOneCluster{
TypeMeta: metav1.TypeMeta{
APIVersion: "kubeone.io/v1alpha1",
Kind: "KubeOneCluster",
APIVersion: "kubeone.io/v1beta1",
Kind: k1api.SchemeGroupVersion.String(),
},
}

k1api.SetObjectDefaults_KubeOneCluster(&k1Cluster)

k1Cluster.CloudProvider = k1api.CloudProviderSpec{
Name: providerName,
External: providerExternal,
}
if err := k1api.SetCloudProvider(&k1Cluster.CloudProvider, providerName); err != nil {
return errors.Wrap(err, "failed to set cloud provider")
}

k1Cluster.Versions = k1api.VersionConfig{
Kubernetes: kubernetesVersion,
Expand Down
7 changes: 3 additions & 4 deletions test/e2e/upgrade_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,6 @@ import (
"github.com/Masterminds/semver"
"github.com/pkg/errors"

k1api "github.com/kubermatic/kubeone/pkg/apis/kubeone/v1alpha1"
"github.com/kubermatic/kubeone/test/e2e/provisioner"
"github.com/kubermatic/kubeone/test/e2e/testutil"

Expand All @@ -46,7 +45,7 @@ const (
func TestClusterUpgrade(t *testing.T) {
testcases := []struct {
name string
provider k1api.CloudProviderName
provider string
providerExternal bool
initialConfigPath string
targetConfigPath string
Expand Down Expand Up @@ -120,7 +119,7 @@ func TestClusterUpgrade(t *testing.T) {
t.Fatal("-target-version must be set")
}

if testProvider != string(tc.provider) {
if testProvider != tc.provider {
t.SkipNow()
}

Expand All @@ -129,7 +128,7 @@ func TestClusterUpgrade(t *testing.T) {
// Create provisioner
testPath := fmt.Sprintf("../../_build/%s", testRunIdentifier)

pr, err := provisioner.CreateProvisioner(testPath, testRunIdentifier, string(tc.provider))
pr, err := provisioner.CreateProvisioner(testPath, testRunIdentifier, tc.provider)
if err != nil {
t.Fatalf("failed to create provisioner: %v", err)
}
Expand Down

0 comments on commit 1de04c2

Please sign in to comment.