fix: postgres_fixes (#54)

## 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: #54
Co-authored-by: Marcel S. Henselin <marcel.henselin@stackit.cloud>
Co-committed-by: Marcel S. Henselin <marcel.henselin@stackit.cloud>

stackit/internal/wait/postgresflexalpha/wait.go

@@ -2,7 +2,10 @@ package postgresflexalpha
 
 import (
 	"context"
+	"errors"
 	"fmt"
+	"math"
+	"net/http"
 	"time"
 
 	"github.com/hashicorp/terraform-plugin-log/tflog"
@@ -40,10 +43,18 @@ type APIClientInstanceInterface interface {
 
 // APIClientUserInterface Interface needed for tests
 type APIClientUserInterface interface {
-	GetUserRequestExecute(ctx context.Context, projectId, region, instanceId string, userId int64) (
+	GetUserRequestExecute(ctx context.Context, projectId, region, instanceId string, userId int32) (
 		*postgresflex.GetUserResponse,
 		error,
 	)
+
+	GetDatabaseRequestExecute(
+		ctx context.Context,
+		projectId string,
+		region string,
+		instanceId string,
+		databaseId int32,
+	) (*postgresflex.GetDatabaseResponse, error)
 }
 
 // CreateInstanceWaitHandler will wait for instance creation
@@ -202,3 +213,62 @@ func PartialUpdateInstanceWaitHandler(
 	handler.SetTimeout(45 * time.Minute).SetSleepBeforeWait(30 * time.Second)
 	return handler
 }
+
+// GetUserByIdWaitHandler will wait for instance creation
+func GetUserByIdWaitHandler(
+	ctx context.Context,
+	a APIClientUserInterface,
+	projectId, instanceId, region string,
+	userId int64,
+) *wait.AsyncActionHandler[postgresflex.GetUserResponse] {
+	handler := wait.New(
+		func() (waitFinished bool, response *postgresflex.GetUserResponse, err error) {
+			if userId > math.MaxInt32 {
+				return false, nil, fmt.Errorf("userId value is too big for int32")
+			}
+			userId32 := int32(userId)
+			s, err := a.GetUserRequestExecute(ctx, projectId, region, instanceId, userId32)
+			if err != nil {
+				var oapiErr *oapierror.GenericOpenAPIError
+				ok := errors.As(err, &oapiErr)
+				if !ok {
+					return false, nil, fmt.Errorf("could not convert error to oapierror.GenericOpenAPIError")
+				}
+				if oapiErr.StatusCode != http.StatusNotFound {
+					return false, nil, err
+				}
+				return false, nil, nil
+			}
+			return true, s, nil
+		},
+	)
+	return handler
+}
+
+// GetDatabaseByIdWaitHandler will wait for instance creation
+func GetDatabaseByIdWaitHandler(
+	ctx context.Context,
+	a APIClientUserInterface,
+	projectId, instanceId, region string,
+	databaseId int64,
+) *wait.AsyncActionHandler[postgresflex.GetDatabaseResponse] {
+	handler := wait.New(
+		func() (waitFinished bool, response *postgresflex.GetDatabaseResponse, err error) {
+			dbId32 := int32(databaseId)
+			s, err := a.GetDatabaseRequestExecute(ctx, projectId, region, instanceId, dbId32)
+			if err != nil {
+				var oapiErr *oapierror.GenericOpenAPIError
+				ok := errors.As(err, &oapiErr)
+				if !ok {
+					return false, nil, fmt.Errorf("could not convert error to oapierror.GenericOpenAPIError")
+				}
+				if oapiErr.StatusCode != http.StatusNotFound {
+					return false, nil, err
+				}
+				return false, nil, nil
+			}
+			return true, s, nil
+		},
+	)
+	return handler
+}