fix(rdb): add test

Author: Monitob

internal/namespaces/rdb/v1/custom_instance_test.go

@@ -180,4 +180,19 @@ func Test_Connect(t *testing.T) {
 		OverrideExec: core.OverrideExecSimple("psql --host {{ .Instance.Endpoint.IP }} --port {{ .Instance.Endpoint.Port }} --username {{ .username }} --dbname rdb", 0),
 		AfterFunc:    deleteInstance(),
 	}))
+	t.Run("psql", core.Test(&core.TestConfig{
+		Commands: GetCommands(),
+		BeforeFunc: core.BeforeFuncCombine(
+			core.BeforeFuncStoreInMeta("username", user),
+			createPN(),
+			createInstanceWithPrivateNetwork("PostgreSQL-14"),
+		),
+		Cmd: "scw rdb instance connect {{ .Instance.ID }} username={{ .username }}",
+		Check: core.TestCheckCombine(
+			core.TestCheckGolden(),
+			core.TestCheckExitCode(0),
+		),
+		OverrideExec: core.OverrideExecSimple("psql --host {{ .Instance.Endpoint.IP }} --port {{ .Instance.Endpoint.Port }} --username {{ .username }} --dbname rdb", 0),
+		AfterFunc:    deleteInstance(),
+	}))
 }

internal/namespaces/rdb/v1/helper_test.go

@@ -4,6 +4,8 @@ import (
 	"fmt"
 
 	"github.com/scaleway/scaleway-cli/v2/internal/core"
+	"github.com/scaleway/scaleway-sdk-go/api/vpc/v1"
+	"github.com/scaleway/scaleway-sdk-go/scw"
 )
 
 const (
@@ -20,6 +22,40 @@ func createInstance(engine string) core.BeforeFunc {
 	)
 }
 
+func createInstanceWithPrivateNetwork(engine string) core.BeforeFunc {
+	return core.ExecStoreBeforeCmd(
+		"Instance",
+		fmt.Sprintf("scw rdb instance create node-type=DB-DEV-S is-ha-cluster=false name=%s engine=%s user-name=%s password=%s init-endpoints.0.private-network.private-network-id={{ .PN.ID }} init-endpoints.0.private-network.service-ip={{ .IPNet }} --wait", name, engine, user, password),
+	)
+}
+
+func createPN() core.BeforeFunc {
+	return func(ctx *core.BeforeFuncCtx) error {
+		var err error
+		api := vpc.NewAPI(ctx.Client)
+		pn, _ := api.CreatePrivateNetwork(&vpc.CreatePrivateNetworkRequest{})
+		ctx.Meta["PN"] = pn
+		ctx.Meta["IPNet"], err = getIPSubnet(pn.Subnets[0])
+		if err != nil {
+			return err
+		}
+		return nil
+	}
+}
+
+func getIPSubnet(ipNet scw.IPNet) (*string, error) {
+	addr := ipNet.IP.To4()
+	if addr == nil {
+		return nil, fmt.Errorf("could get ip 4 bytes")
+	}
+	addr = addr.Mask(addr.DefaultMask())
+	addr[3] = +3
+
+	sz, _ := ipNet.Mask.Size()
+	ipNetStr := fmt.Sprintf("%s/%d", addr.String(), sz)
+	return &ipNetStr, nil
+}
+
 func deleteInstance() core.AfterFunc {
 	return core.ExecAfterCmd("scw rdb instance delete {{ .Instance.ID }}")
 }