mirror of https://github.com/k3s-io/k3s
Merge pull request #70135 from marc-sensenich/gh-70126/azure-is-node-unmanaged-fix
Correct regexp check in IsNodeUnmanagedByProviderpull/58/head
commit
3905fb6696
|
@ -27,7 +27,6 @@ import (
|
||||||
"github.com/Azure/azure-sdk-for-go/services/network/mgmt/2017-09-01/network"
|
"github.com/Azure/azure-sdk-for-go/services/network/mgmt/2017-09-01/network"
|
||||||
"github.com/Azure/go-autorest/autorest"
|
"github.com/Azure/go-autorest/autorest"
|
||||||
"github.com/golang/glog"
|
"github.com/golang/glog"
|
||||||
|
|
||||||
"k8s.io/apimachinery/pkg/types"
|
"k8s.io/apimachinery/pkg/types"
|
||||||
cloudprovider "k8s.io/cloud-provider"
|
cloudprovider "k8s.io/cloud-provider"
|
||||||
)
|
)
|
||||||
|
@ -302,5 +301,5 @@ func (az *Cloud) IsNodeUnmanaged(nodeName string) (bool, error) {
|
||||||
// IsNodeUnmanagedByProviderID returns true if the node is not managed by Azure cloud provider.
|
// IsNodeUnmanagedByProviderID returns true if the node is not managed by Azure cloud provider.
|
||||||
// All managed node's providerIDs are in format 'azure:///subscriptions/<id>/resourceGroups/<rg>/providers/Microsoft.Compute/.*'
|
// All managed node's providerIDs are in format 'azure:///subscriptions/<id>/resourceGroups/<rg>/providers/Microsoft.Compute/.*'
|
||||||
func (az *Cloud) IsNodeUnmanagedByProviderID(providerID string) bool {
|
func (az *Cloud) IsNodeUnmanagedByProviderID(providerID string) bool {
|
||||||
return azureNodeProviderIDRE.Match([]byte(providerID))
|
return !azureNodeProviderIDRE.Match([]byte(providerID))
|
||||||
}
|
}
|
||||||
|
|
|
@ -107,3 +107,38 @@ func TestIsNodeUnmanaged(t *testing.T) {
|
||||||
assert.Equal(t, test.expected, real, test.name)
|
assert.Equal(t, test.expected, real, test.name)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestIsNodeUnmanagedByProviderID(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
providerID string
|
||||||
|
expected bool
|
||||||
|
name string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
providerID: CloudProviderName + ":///subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/myResourceGroupName/providers/Microsoft.Compute/virtualMachines/k8s-agent-AAAAAAAA-0",
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
providerID: CloudProviderName + "://",
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
providerID: ":///subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/myResourceGroupName/providers/Microsoft.Compute/virtualMachines/k8s-agent-AAAAAAAA-0",
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
providerID: "aws:///subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/myResourceGroupName/providers/Microsoft.Compute/virtualMachines/k8s-agent-AAAAAAAA-0",
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
providerID: "k8s-agent-AAAAAAAA-0",
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
az := getTestCloud()
|
||||||
|
for _, test := range tests {
|
||||||
|
isUnmanagedNode := az.IsNodeUnmanagedByProviderID(test.providerID)
|
||||||
|
assert.Equal(t, test.expected, isUnmanagedNode, test.providerID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue