Marcel S. Henselin
e01ae1a920
## Description
<!-- **Please link some issue here describing what you are trying to achieve.**
In case there is no issue present for your PR, please consider creating one.
At least please give us some description what you are trying to achieve and why your change is needed. -->
relates to #1234
## Checklist
- [ ] Issue was linked above
- [ ] Code format was applied: `make fmt`
- [ ] Examples were added / adjusted (see `examples/` directory)
- [x] Docs are up-to-date: `make generate-docs` (will be checked by CI)
- [ ] Unit tests got implemented or updated
- [ ] Acceptance tests got implemented or updated (see e.g. [here](f5f99d1709/stackit/internal/services/dns/dns_acc_test.go))
- [x] Unit tests are passing: `make test` (will be checked by CI)
- [x] No linter issues: `make lint` (will be checked by CI)
Reviewed-on: #58
Co-authored-by: Marcel S. Henselin <marcel.henselin@stackit.cloud>
Co-committed-by: Marcel S. Henselin <marcel.henselin@stackit.cloud>
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)
|
|
}
|
|
},
|
|
)
|
|
}
|
|
}
|