mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-12-28 00:49:26 -05:00
88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package azure
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/edgelesssys/constellation/v2/internal/kubernetes"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
k8s "k8s.io/api/core/v1"
|
|
meta "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
)
|
|
|
|
func TestAutoscalerSecrets(t *testing.T) {
|
|
testCases := map[string]struct {
|
|
providerID string
|
|
cloudServiceAccountURI string
|
|
wantSecrets kubernetes.Secrets
|
|
wantErr bool
|
|
}{
|
|
"Secrets works": {
|
|
providerID: "azure:///subscriptions/subscription-id/resourceGroups/resource-group/providers/Microsoft.Compute/virtualMachineScaleSets/scaleset/virtualMachines/instance-name",
|
|
cloudServiceAccountURI: "serviceaccount://azure?tenant_id=tenant-id&client_id=client-id&client_secret=client-secret",
|
|
wantSecrets: kubernetes.Secrets{
|
|
&k8s.Secret{
|
|
TypeMeta: meta.TypeMeta{
|
|
Kind: "Secret",
|
|
APIVersion: "v1",
|
|
},
|
|
ObjectMeta: meta.ObjectMeta{
|
|
Name: "cluster-autoscaler-azure",
|
|
Namespace: "kube-system",
|
|
},
|
|
Data: map[string][]byte{
|
|
"ClientID": []byte("client-id"),
|
|
"ClientSecret": []byte("client-secret"),
|
|
"ResourceGroup": []byte("resource-group"),
|
|
"SubscriptionID": []byte("subscription-id"),
|
|
"TenantID": []byte("tenant-id"),
|
|
"VMType": []byte("vmss"),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
"invalid providerID fails": {
|
|
providerID: "invalid",
|
|
wantErr: true,
|
|
},
|
|
"invalid cloudServiceAccountURI fails": {
|
|
providerID: "azure:///subscriptions/subscription-id/resourceGroups/resource-group/providers/Microsoft.Compute/virtualMachines/instance-name",
|
|
cloudServiceAccountURI: "invalid",
|
|
wantErr: true,
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
require := require.New(t)
|
|
|
|
autoscaler := Autoscaler{}
|
|
secrets, err := autoscaler.Secrets(tc.providerID, tc.cloudServiceAccountURI)
|
|
if tc.wantErr {
|
|
assert.Error(err)
|
|
return
|
|
}
|
|
require.NoError(err)
|
|
assert.Equal(tc.wantSecrets, secrets)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestTrivialAutoscalerFunctions(t *testing.T) {
|
|
assert := assert.New(t)
|
|
autoscaler := Autoscaler{}
|
|
|
|
assert.NotEmpty(autoscaler.Name())
|
|
assert.Empty(autoscaler.Volumes())
|
|
assert.Empty(autoscaler.VolumeMounts())
|
|
assert.NotEmpty(autoscaler.Env())
|
|
assert.True(autoscaler.Supported())
|
|
}
|