feat: Prefer embedded-cluster over k0s when detecting distro (#1544)

* feat: Prefer embedded-cluster over k0s when detecting distro

* Implement check for embedded cluster detection
This commit is contained in:
Evans Mungai
2024-05-15 09:26:03 +01:00
committed by GitHub
parent 224d55b076
commit abfb95ea83
2 changed files with 52 additions and 0 deletions
+5
View File
@@ -191,6 +191,11 @@ func ParseNodesForProviders(nodes []corev1.Node) (providers, string) {
}
}
// If k0s and embedded-cluster are both found, prefer embedded-cluster
if foundProviders.k0s && foundProviders.embeddedCluster {
stringProvider = "embedded-cluster"
}
return foundProviders, stringProvider
}
+47
View File
@@ -5,6 +5,8 @@ import (
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
corev1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)
func Test_compareDistributionConditionalToActual(t *testing.T) {
@@ -167,3 +169,48 @@ func Test_mustNormalizeDistributionName(t *testing.T) {
})
}
}
func TestParseNodesForProviders(t *testing.T) {
tests := []struct {
name string
nodes []corev1.Node
wantProviders providers
wantProviderString string
}{
{
name: "embedded-cluster",
nodes: []corev1.Node{
{
ObjectMeta: metav1.ObjectMeta{
Name: "embedded-cluster",
Labels: map[string]string{
"beta.kubernetes.io/arch": "amd64",
"beta.kubernetes.io/os": "linux",
"kots.io/embedded-cluster-role": "total-1",
"kots.io/embedded-cluster-role-0": "management",
"kubernetes.io/arch": "amd64",
"kubernetes.io/hostname": "evans-vm1",
"kubernetes.io/os": "linux",
"management": "true",
"node-role.kubernetes.io/control-plane": "true",
"node.k0sproject.io/role": "control-plane",
},
},
},
},
wantProviders: providers{embeddedCluster: true, k0s: true},
wantProviderString: "embedded-cluster",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
providers, stringProvider := ParseNodesForProviders(tt.nodes)
assert.Equalf(t, tt.wantProviders, providers,
"ParseNodesForProviders() gotProviders = %v, providers %v", providers, tt.wantProviders,
)
assert.Equalf(t, tt.wantProviderString, stringProvider,
"ParseNodesForProviders() gotStringProvider = %v, stringProvider %v", stringProvider, tt.wantProviderString,
)
})
}
}