Jeff Mitchell
a9e372930d
Post-merge some issues were discovered with the ways typing is used for proto reflection. This will be reworked without moving the subtypes around.
81 lines
2.3 KiB
Go
81 lines
2.3 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package tcp
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/boundary/internal/db"
|
|
"github.com/hashicorp/boundary/internal/target"
|
|
"github.com/hashicorp/go-uuid"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// TestTarget is used to create a Target that can be used by tests in other packages.
|
|
func TestTarget(ctx context.Context, t testing.TB, conn *db.DB, projectId, name string, opt ...target.Option) target.Target {
|
|
t.Helper()
|
|
opt = append(opt, target.WithName(name))
|
|
opts := target.GetOpts(opt...)
|
|
require := require.New(t)
|
|
rw := db.New(conn)
|
|
tar, err := target.New(ctx, Subtype, projectId, opt...)
|
|
require.NoError(err)
|
|
id, err := db.NewPublicId(ctx, TargetPrefix)
|
|
require.NoError(err)
|
|
tar.SetPublicId(ctx, id)
|
|
err = rw.Create(ctx, tar)
|
|
require.NoError(err)
|
|
|
|
if opts.WithAddress != "" {
|
|
address, err := target.NewAddress(ctx, tar.GetPublicId(), opts.WithAddress)
|
|
require.NoError(err)
|
|
require.NotNil(address)
|
|
err = rw.Create(context.Background(), address)
|
|
require.NoError(err)
|
|
}
|
|
if len(opts.WithHostSources) > 0 {
|
|
newHostSets := make([]any, 0, len(opts.WithHostSources))
|
|
for _, s := range opts.WithHostSources {
|
|
hostSet, err := target.NewTargetHostSet(ctx, tar.GetPublicId(), s)
|
|
require.NoError(err)
|
|
newHostSets = append(newHostSets, hostSet)
|
|
}
|
|
err := rw.CreateItems(ctx, newHostSets)
|
|
require.NoError(err)
|
|
}
|
|
if len(opts.WithCredentialLibraries) > 0 {
|
|
newCredLibs := make([]any, 0, len(opts.WithCredentialLibraries))
|
|
for _, cl := range opts.WithCredentialLibraries {
|
|
cl.TargetId = tar.GetPublicId()
|
|
newCredLibs = append(newCredLibs, cl)
|
|
}
|
|
err := rw.CreateItems(ctx, newCredLibs)
|
|
require.NoError(err)
|
|
}
|
|
if len(opts.WithStaticCredentials) > 0 {
|
|
newCreds := make([]any, 0, len(opts.WithStaticCredentials))
|
|
for _, c := range opts.WithStaticCredentials {
|
|
c.TargetId = tar.GetPublicId()
|
|
newCreds = append(newCreds, c)
|
|
}
|
|
err := rw.CreateItems(ctx, newCreds)
|
|
require.NoError(err)
|
|
}
|
|
return tar
|
|
}
|
|
|
|
func testTargetName(t testing.TB, projectId string) string {
|
|
t.Helper()
|
|
return fmt.Sprintf("%s-%s", projectId, testId(t))
|
|
}
|
|
|
|
func testId(t testing.TB) string {
|
|
t.Helper()
|
|
id, err := uuid.GenerateUUID()
|
|
require.NoError(t, err)
|
|
return fmt.Sprintf("%s_%s", TargetPrefix, id)
|
|
}
|