Marcel S. Henselin
944b872db7
Some checks failed
CI Workflow / Check GoReleaser config (pull_request) Successful in 6s
CI Workflow / CI (pull_request) Failing after 16m49s
CI Workflow / Code coverage report (pull_request) Has been skipped
CI Workflow / Test readiness for publishing provider (pull_request) Successful in 18m3s
336 lines
9.5 KiB
Go
336 lines
9.5 KiB
Go
// Copyright (c) STACKIT
|
|
|
|
package postgresflexalpha
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/stackitcloud/stackit-sdk-go/core/oapierror"
|
|
"github.com/stackitcloud/stackit-sdk-go/core/utils"
|
|
|
|
postgresflex "tf-provider.git.onstackit.cloud/stackit-dev-tools/terraform-provider-stackitprivatepreview/pkg_gen/postgresflexalpha"
|
|
)
|
|
|
|
// Used for testing instance operations
|
|
type apiClientInstanceMocked struct {
|
|
instanceId string
|
|
instanceState string
|
|
instanceNetwork postgresflex.InstanceNetwork
|
|
instanceIsForceDeleted bool
|
|
instanceGetFails bool
|
|
usersGetErrorStatus int
|
|
}
|
|
|
|
func (a *apiClientInstanceMocked) GetInstanceRequestExecute(
|
|
_ context.Context,
|
|
_, _, _ string,
|
|
) (*postgresflex.GetInstanceResponse, error) {
|
|
if a.instanceGetFails {
|
|
return nil, &oapierror.GenericOpenAPIError{
|
|
StatusCode: 500,
|
|
}
|
|
}
|
|
|
|
if a.instanceIsForceDeleted {
|
|
return nil, &oapierror.GenericOpenAPIError{
|
|
StatusCode: 404,
|
|
}
|
|
}
|
|
|
|
return &postgresflex.GetInstanceResponse{
|
|
Id: &a.instanceId,
|
|
Status: postgresflex.GetInstanceResponseGetStatusAttributeType(&a.instanceState),
|
|
Network: postgresflex.GetInstanceResponseGetNetworkAttributeType(&a.instanceNetwork),
|
|
}, nil
|
|
}
|
|
|
|
func (a *apiClientInstanceMocked) ListUsersRequestExecute(
|
|
_ context.Context,
|
|
_, _, _ string,
|
|
) (*postgresflex.ListUserResponse, error) {
|
|
if a.usersGetErrorStatus != 0 {
|
|
return nil, &oapierror.GenericOpenAPIError{
|
|
StatusCode: a.usersGetErrorStatus,
|
|
}
|
|
}
|
|
|
|
aux := int64(0)
|
|
return &postgresflex.ListUserResponse{
|
|
Pagination: &postgresflex.Pagination{
|
|
TotalRows: &aux,
|
|
},
|
|
Users: &[]postgresflex.ListUser{},
|
|
}, nil
|
|
}
|
|
|
|
func TestCreateInstanceWaitHandler(t *testing.T) {
|
|
tests := []struct {
|
|
desc string
|
|
instanceGetFails bool
|
|
instanceState string
|
|
instanceNetwork postgresflex.InstanceNetwork
|
|
usersGetErrorStatus int
|
|
wantErr bool
|
|
wantRes *postgresflex.GetInstanceResponse
|
|
}{
|
|
{
|
|
desc: "create_succeeded",
|
|
instanceGetFails: false,
|
|
instanceState: InstanceStateSuccess,
|
|
instanceNetwork: postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
wantErr: false,
|
|
wantRes: &postgresflex.GetInstanceResponse{
|
|
Id: utils.Ptr("foo-bar"),
|
|
Status: postgresflex.GetInstanceResponseGetStatusAttributeType(utils.Ptr(InstanceStateSuccess)),
|
|
Network: &postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
desc: "create_failed",
|
|
instanceGetFails: false,
|
|
instanceState: InstanceStateFailed,
|
|
instanceNetwork: postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
wantErr: true,
|
|
wantRes: nil,
|
|
},
|
|
{
|
|
desc: "create_failed_2",
|
|
instanceGetFails: false,
|
|
instanceState: InstanceStateEmpty,
|
|
instanceNetwork: postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
wantErr: true,
|
|
wantRes: nil,
|
|
},
|
|
{
|
|
desc: "instance_get_fails",
|
|
instanceGetFails: true,
|
|
wantErr: true,
|
|
wantRes: nil,
|
|
},
|
|
{
|
|
desc: "users_get_fails",
|
|
instanceGetFails: false,
|
|
instanceState: InstanceStateSuccess,
|
|
instanceNetwork: postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
usersGetErrorStatus: 500,
|
|
wantErr: true,
|
|
wantRes: nil,
|
|
},
|
|
{
|
|
desc: "users_get_fails_2",
|
|
instanceGetFails: false,
|
|
instanceState: InstanceStateSuccess,
|
|
instanceNetwork: postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
usersGetErrorStatus: 400,
|
|
wantErr: true,
|
|
wantRes: &postgresflex.GetInstanceResponse{
|
|
Id: utils.Ptr("foo-bar"),
|
|
Status: postgresflex.GetInstanceResponseGetStatusAttributeType(utils.Ptr(InstanceStateSuccess)),
|
|
Network: &postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
desc: "fail when response has no instance address",
|
|
instanceGetFails: false,
|
|
instanceState: InstanceStateSuccess,
|
|
instanceNetwork: postgresflex.InstanceNetwork{
|
|
AccessScope: postgresflex.InstanceNetworkGetAccessScopeAttributeType(utils.Ptr("SNA")),
|
|
Acl: nil,
|
|
InstanceAddress: nil,
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
wantErr: true,
|
|
wantRes: nil,
|
|
},
|
|
{
|
|
desc: "timeout",
|
|
instanceGetFails: false,
|
|
instanceState: InstanceStateProgressing,
|
|
instanceNetwork: postgresflex.InstanceNetwork{
|
|
AccessScope: postgresflex.InstanceNetworkGetAccessScopeAttributeType(utils.Ptr("SNA")),
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
wantErr: true,
|
|
wantRes: nil,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(
|
|
tt.desc, func(t *testing.T) {
|
|
instanceId := "foo-bar"
|
|
|
|
apiClient := &apiClientInstanceMocked{
|
|
instanceId: instanceId,
|
|
instanceState: tt.instanceState,
|
|
instanceNetwork: tt.instanceNetwork,
|
|
instanceGetFails: tt.instanceGetFails,
|
|
usersGetErrorStatus: tt.usersGetErrorStatus,
|
|
}
|
|
|
|
handler := CreateInstanceWaitHandler(context.Background(), apiClient, "", "", instanceId)
|
|
|
|
gotRes, err := handler.SetTimeout(10 * time.Millisecond).SetSleepBeforeWait(1 * time.Millisecond).WaitWithContext(context.Background())
|
|
if (err != nil) != tt.wantErr {
|
|
t.Fatalf("handler error = %v, wantErr %v", err, tt.wantErr)
|
|
}
|
|
|
|
if !cmp.Equal(gotRes, tt.wantRes) {
|
|
t.Fatalf("handler gotRes = %v, want %v", gotRes, tt.wantRes)
|
|
}
|
|
},
|
|
)
|
|
}
|
|
}
|
|
|
|
func TestUpdateInstanceWaitHandler(t *testing.T) {
|
|
tests := []struct {
|
|
desc string
|
|
instanceGetFails bool
|
|
instanceState string
|
|
instanceNetwork postgresflex.InstanceNetwork
|
|
wantErr bool
|
|
wantRes *postgresflex.GetInstanceResponse
|
|
}{
|
|
{
|
|
desc: "update_succeeded",
|
|
instanceGetFails: false,
|
|
instanceState: InstanceStateSuccess,
|
|
instanceNetwork: postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
wantErr: false,
|
|
wantRes: &postgresflex.GetInstanceResponse{
|
|
Id: utils.Ptr("foo-bar"),
|
|
Status: postgresflex.GetInstanceResponseGetStatusAttributeType(utils.Ptr(InstanceStateSuccess)),
|
|
Network: &postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
desc: "update_failed",
|
|
instanceGetFails: false,
|
|
instanceState: InstanceStateFailed,
|
|
instanceNetwork: postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
wantErr: true,
|
|
wantRes: &postgresflex.GetInstanceResponse{
|
|
Id: utils.Ptr("foo-bar"),
|
|
Status: postgresflex.GetInstanceResponseGetStatusAttributeType(utils.Ptr(InstanceStateFailed)),
|
|
Network: &postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
desc: "update_failed_2",
|
|
instanceGetFails: false,
|
|
instanceState: InstanceStateEmpty,
|
|
instanceNetwork: postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
wantErr: true,
|
|
wantRes: nil,
|
|
},
|
|
{
|
|
desc: "get_fails",
|
|
instanceGetFails: true,
|
|
wantErr: true,
|
|
wantRes: nil,
|
|
},
|
|
{
|
|
desc: "timeout",
|
|
instanceGetFails: false,
|
|
instanceState: InstanceStateProgressing,
|
|
instanceNetwork: postgresflex.InstanceNetwork{
|
|
AccessScope: nil,
|
|
Acl: nil,
|
|
InstanceAddress: utils.Ptr("10.0.0.1"),
|
|
RouterAddress: utils.Ptr("10.0.0.1"),
|
|
},
|
|
wantErr: true,
|
|
wantRes: nil,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(
|
|
tt.desc, func(t *testing.T) {
|
|
instanceId := "foo-bar"
|
|
|
|
apiClient := &apiClientInstanceMocked{
|
|
instanceId: instanceId,
|
|
instanceState: tt.instanceState,
|
|
instanceNetwork: tt.instanceNetwork,
|
|
instanceGetFails: tt.instanceGetFails,
|
|
}
|
|
|
|
handler := PartialUpdateInstanceWaitHandler(context.Background(), apiClient, "", "", instanceId)
|
|
|
|
gotRes, err := handler.SetTimeout(10 * time.Millisecond).WaitWithContext(context.Background())
|
|
if (err != nil) != tt.wantErr {
|
|
t.Fatalf("handler error = %v, wantErr %v", err, tt.wantErr)
|
|
}
|
|
|
|
if !cmp.Equal(gotRes, tt.wantRes) {
|
|
t.Fatalf("handler gotRes = %v, want %v", gotRes, tt.wantRes)
|
|
}
|
|
},
|
|
)
|
|
}
|
|
}
|