From 3e06e0610d56499e676eb8cc3806751561abdac1 Mon Sep 17 00:00:00 2001
From: Kenji Kaneda <kenji@cloudnatix.com>
Date: Tue, 20 Apr 2021 09:09:58 -0700
Subject: [PATCH] Add Azure image to alpha/stable channel

Remove the default image hard-coded in populate_instancegroup_spec.go.
---
 pkg/apis/kops/channel.go                           | 3 ++-
 upup/pkg/fi/cloudup/populate_instancegroup_spec.go | 7 ++-----
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/pkg/apis/kops/channel.go b/pkg/apis/kops/channel.go
index fa4846e4eb66b..f08db1813b853 100644
--- a/pkg/apis/kops/channel.go
+++ b/pkg/apis/kops/channel.go
@@ -335,5 +335,6 @@ func (c *Channel) HasUpstreamImagePrefix(image string) bool {
 	return strings.HasPrefix(image, "kope.io/k8s-") ||
 		strings.HasPrefix(image, "099720109477/ubuntu/images/hvm-ssd/ubuntu-focal-20.04-") ||
 		strings.HasPrefix(image, "cos-cloud/cos-stable-") ||
-		strings.HasPrefix(image, "ubuntu-os-cloud/ubuntu-")
+		strings.HasPrefix(image, "ubuntu-os-cloud/ubuntu-") ||
+		strings.HasPrefix(image, "Canonical:UbuntuServer:")
 }
diff --git a/upup/pkg/fi/cloudup/populate_instancegroup_spec.go b/upup/pkg/fi/cloudup/populate_instancegroup_spec.go
index f9985d1b7fe7b..ec9d9a4ce295c 100644
--- a/upup/pkg/fi/cloudup/populate_instancegroup_spec.go
+++ b/upup/pkg/fi/cloudup/populate_instancegroup_spec.go
@@ -48,9 +48,8 @@ const (
 	defaultMasterMachineTypeALI   = "ecs.n2.medium"
 	defaultMasterMachineTypeAzure = "Standard_B2ms"
 
-	defaultDONodeImage    = "ubuntu-20-04-x64"
-	defaultALINodeImage   = "centos_7_04_64_20G_alibase_201701015.vhd"
-	defaultAzureNodeImage = "Canonical:UbuntuServer:20.04-LTS:latest"
+	defaultDONodeImage  = "ubuntu-20-04-x64"
+	defaultALINodeImage = "centos_7_04_64_20G_alibase_201701015.vhd"
 )
 
 // TODO: this hardcoded list can be replaced with DescribeInstanceTypes' DedicatedHostsSupported field
@@ -254,8 +253,6 @@ func defaultImage(cluster *kops.Cluster, channel *kops.Channel, architecture arc
 		return defaultDONodeImage
 	case kops.CloudProviderALI:
 		return defaultALINodeImage
-	case kops.CloudProviderAzure:
-		return defaultAzureNodeImage
 	}
 	klog.Infof("Cannot set default Image for CloudProvider=%q", cluster.Spec.CloudProvider)
 	return ""