зеркало из https://github.com/Azure/ARO-RP.git
111 строки
3.0 KiB
Go
111 строки
3.0 KiB
Go
package frontend
|
|
|
|
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the Apache License 2.0.
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"go.uber.org/mock/gomock"
|
|
|
|
"github.com/Azure/ARO-RP/pkg/api"
|
|
"github.com/Azure/ARO-RP/pkg/env"
|
|
"github.com/Azure/ARO-RP/pkg/frontend/adminactions"
|
|
"github.com/Azure/ARO-RP/pkg/metrics/noop"
|
|
mock_adminactions "github.com/Azure/ARO-RP/pkg/util/mocks/adminactions"
|
|
testdatabase "github.com/Azure/ARO-RP/test/database"
|
|
)
|
|
|
|
func TestAdminDrainNode(t *testing.T) {
|
|
mockSubID := "00000000-0000-0000-0000-000000000000"
|
|
mockTenantID := "00000000-0000-0000-0000-000000000000"
|
|
|
|
ctx := context.Background()
|
|
|
|
type test struct {
|
|
name string
|
|
resourceID string
|
|
fixture func(*testdatabase.Fixture)
|
|
vmName string
|
|
mocks func(*test, *mock_adminactions.MockKubeActions)
|
|
wantStatusCode int
|
|
wantResponse []byte
|
|
wantError string
|
|
}
|
|
|
|
for _, tt := range []*test{
|
|
{
|
|
name: "basic coverage",
|
|
vmName: "aro-worker-australiasoutheast-7tcq7",
|
|
resourceID: testdatabase.GetResourcePath(mockSubID, "resourceName"),
|
|
fixture: func(f *testdatabase.Fixture) {
|
|
f.AddOpenShiftClusterDocuments(&api.OpenShiftClusterDocument{
|
|
Key: strings.ToLower(testdatabase.GetResourcePath(mockSubID, "resourceName")),
|
|
OpenShiftCluster: &api.OpenShiftCluster{
|
|
ID: testdatabase.GetResourcePath(mockSubID, "resourceName"),
|
|
Properties: api.OpenShiftClusterProperties{
|
|
ClusterProfile: api.ClusterProfile{
|
|
ResourceGroupID: fmt.Sprintf("/subscriptions/%s/resourceGroups/test-cluster", mockSubID),
|
|
},
|
|
},
|
|
},
|
|
})
|
|
|
|
f.AddSubscriptionDocuments(&api.SubscriptionDocument{
|
|
ID: mockSubID,
|
|
Subscription: &api.Subscription{
|
|
State: api.SubscriptionStateRegistered,
|
|
Properties: &api.SubscriptionProperties{
|
|
TenantID: mockTenantID,
|
|
},
|
|
},
|
|
})
|
|
},
|
|
mocks: func(tt *test, k *mock_adminactions.MockKubeActions) {
|
|
k.EXPECT().DrainNode(gomock.Any(), tt.vmName).Return(nil)
|
|
},
|
|
wantStatusCode: http.StatusOK,
|
|
},
|
|
} {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
ti := newTestInfra(t).WithOpenShiftClusters().WithSubscriptions()
|
|
defer ti.done()
|
|
|
|
k := mock_adminactions.NewMockKubeActions(ti.controller)
|
|
tt.mocks(tt, k)
|
|
|
|
err := ti.buildFixtures(tt.fixture)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
f, err := NewFrontend(ctx, ti.audit, ti.log, ti.env, ti.dbGroup, api.APIs, &noop.Noop{}, &noop.Noop{}, nil, nil, func(*logrus.Entry, env.Interface, *api.OpenShiftCluster) (adminactions.KubeActions, error) {
|
|
return k, nil
|
|
}, nil, nil, nil)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
go f.Run(ctx, nil, nil)
|
|
|
|
resp, b, err := ti.request(http.MethodPost,
|
|
fmt.Sprintf("https://server/admin%s/drainnode?vmName=%s", tt.resourceID, tt.vmName),
|
|
nil, nil)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
err = validateResponse(resp, b, tt.wantStatusCode, tt.wantError, tt.wantResponse)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
})
|
|
}
|
|
}
|