feat(e2e): Common API/CLI functions for targets

pull/6010/head
Hugo 10 months ago committed by irenarindos
parent 288f505482
commit e8a5c5c80d

@ -16,6 +16,44 @@ import (
"github.com/hashicorp/go-secure-stdlib/base62"
)
// ReadTargetApi uses the Go API to read a target given its id.
func ReadTargetApi(t testing.TB, ctx context.Context, client *api.Client, targetId string) (*targets.Target, error) {
tClient := targets.NewClient(client)
trr, err := tClient.Read(ctx, targetId)
if err != nil {
return nil, err
}
if trr == nil {
return nil, fmt.Errorf("target read response was empty")
}
if trr.GetItem() == nil {
return nil, fmt.Errorf("target read response item was empty")
}
return trr.GetItem(), nil
}
// ListTargetsApi uses the Go API to list targets in a project scope.
func ListTargetsApi(t testing.TB, ctx context.Context, client *api.Client, projectId string) ([]*targets.Target, error) {
tClient := targets.NewClient(client)
tlr, err := tClient.List(ctx, projectId)
if err != nil {
return nil, err
}
if tlr == nil {
return nil, fmt.Errorf("target list response was empty")
}
return tlr.GetItems(), nil
}
// DeleteTargetApi uses the Go API to delete a target with the given id.
func DeleteTargetApi(t testing.TB, ctx context.Context, client *api.Client, targetId string) error {
tClient := targets.NewClient(client)
_, err := tClient.Delete(ctx, targetId)
return err
}
// CreateTargetApi uses the Go API to create a new target of the provided type
// in Boundary. Automatically sets a random target name.
// Returns the id of the new target.
@ -37,6 +75,15 @@ func CreateTargetApi(t testing.TB, ctx context.Context, client *api.Client, proj
return targetId, nil
}
// UpdateTargetApi uses the Go API to update a Boundary target given its id.
func UpdateTargetApi(t testing.TB, ctx context.Context, client *api.Client, targetId string, opts ...targets.Option) error {
tClient := targets.NewClient(client)
opts = append(opts, targets.WithAutomaticVersioning(true))
_, err := tClient.Update(ctx, targetId, 0, opts...)
return err
}
// AddHostSourceToTargetApi uses the Go api to add a host source (host set or host) to a target
func AddHostSourceToTargetApi(t testing.TB, ctx context.Context, client *api.Client, targetId string, hostSourceId string) error {
tClient := targets.NewClient(client)
@ -47,6 +94,17 @@ func AddHostSourceToTargetApi(t testing.TB, ctx context.Context, client *api.Cli
return err
}
// AddBrokeredCredentialSourceToTargetApi uses the Go API to add a brokered
// credential source to a target.
func AddBrokeredCredentialSourceToTargetApi(t testing.TB, ctx context.Context, client *api.Client, targetId, credentialSourceId string) error {
tClient := targets.NewClient(client)
_, err := tClient.AddCredentialSources(ctx, targetId, 0,
targets.WithAutomaticVersioning(true),
targets.WithBrokeredCredentialSourceIds([]string{credentialSourceId}),
)
return err
}
// CreateTargetCli uses the cli to create a new target in boundary
// Returns the id of the new target.
func CreateTargetCli(t testing.TB, ctx context.Context, projectId string, defaultPort string, opt ...target.Option) (string, error) {
@ -77,6 +135,9 @@ func CreateTargetCli(t testing.TB, ctx context.Context, projectId string, defaul
} else {
args = append(args, "-name", fmt.Sprintf("e2e Target %s", name))
}
if opts.WithDescription != "" {
args = append(args, "-description", opts.WithDescription)
}
if opts.WithAddress != "" {
args = append(args, "-address", opts.WithAddress)
}
@ -121,6 +182,122 @@ func CreateTargetCli(t testing.TB, ctx context.Context, projectId string, defaul
return targetId, nil
}
// UpdateTargetCli uses the CLI to update a Boundary target.
func UpdateTargetCli(t testing.TB, ctx context.Context, targetId string, opt ...target.Option) error {
opts := target.GetOpts(opt...)
var args []string
// Set target type. Default to tcp if not specified
if opts.WithType != "" {
args = append(args, string(opts.WithType))
} else {
args = append(args, "tcp")
}
args = append(args, "-format", "json")
args = append(args, "-id", targetId)
if opts.WithName != "" {
args = append(args, "-name", opts.WithName)
}
if opts.WithDescription != "" {
args = append(args, "-description", opts.WithDescription)
}
if opts.WithAddress != "" {
args = append(args, "-address", opts.WithAddress)
}
if opts.WithDefaultPort != 0 {
args = append(args, "-default-port", fmt.Sprintf("%d", opts.WithDefaultPort))
}
if opts.WithDefaultClientPort != 0 {
args = append(args, "-default-client-port", fmt.Sprintf("%d", opts.WithDefaultClientPort))
}
if opts.WithEnableSessionRecording != false {
args = append(args, "-enable-session-recording", fmt.Sprintf("%v", opts.WithEnableSessionRecording))
}
if opts.WithStorageBucketId != "" {
args = append(args, "-storage-bucket-id", opts.WithStorageBucketId)
}
if opts.WithIngressWorkerFilter != "" {
args = append(args, "-ingress-worker-filter", opts.WithIngressWorkerFilter)
}
if opts.WithEgressWorkerFilter != "" {
args = append(args, "-egress-worker-filter", opts.WithEgressWorkerFilter)
}
if opts.WithSessionConnectionLimit != 0 {
args = append(args, "-session-connection-limit", fmt.Sprintf("%d", opts.WithSessionConnectionLimit))
}
if opts.WithSessionMaxSeconds != 0 {
args = append(args, "-session-max-seconds", fmt.Sprintf("%d", opts.WithSessionMaxSeconds))
}
output := e2e.RunCommand(ctx, "boundary",
e2e.WithArgs("targets", "update"),
e2e.WithArgs(args...),
)
if output.Err != nil {
return fmt.Errorf("%w: %s", output.Err, string(output.Stderr))
}
return nil
}
// ReadTargetCli uses the CLI to read a Boundary target with the given id.
func ReadTargetCli(t testing.TB, ctx context.Context, targetId string) (*targets.Target, error) {
output := e2e.RunCommand(ctx, "boundary",
e2e.WithArgs("targets", "read"),
e2e.WithArgs("-id", targetId),
e2e.WithArgs("-format", "json"),
)
if output.Err != nil {
return nil, fmt.Errorf("%w: %s", output.Err, string(output.Stderr))
}
var readTargetResult targets.TargetReadResult
err := json.Unmarshal(output.Stdout, &readTargetResult)
if err != nil {
return nil, err
}
if readTargetResult.GetItem() == nil {
return nil, fmt.Errorf("target read response item was empty")
}
return readTargetResult.GetItem(), nil
}
// ListTargetsCli uses the CLI to list Boundary targets in a given project scope.
func ListTargetsCli(t testing.TB, ctx context.Context, projectId string) ([]*targets.Target, error) {
output := e2e.RunCommand(ctx, "boundary",
e2e.WithArgs("targets", "list"),
e2e.WithArgs("-scope-id", projectId),
e2e.WithArgs("-format", "json"),
)
if output.Err != nil {
return nil, fmt.Errorf("%w: %s", output.Err, string(output.Stderr))
}
var listTargetResult targets.TargetListResult
err := json.Unmarshal(output.Stdout, &listTargetResult)
if err != nil {
return nil, err
}
return listTargetResult.GetItems(), nil
}
// DeleteTargetCli uses the CLI to delete a Boundary target with the given id.
func DeleteTargetCli(t testing.TB, ctx context.Context, targetId string) error {
output := e2e.RunCommand(ctx, "boundary",
e2e.WithArgs("targets", "delete"),
e2e.WithArgs("-id", targetId),
e2e.WithArgs("-format", "json"),
)
if output.Err != nil {
return fmt.Errorf("%w: %s", output.Err, string(output.Stderr))
}
return nil
}
// AddHostSourceToTargetCli uses the cli to add a host source (host set or host)
// to a target.
// Boundary's `add-host-sources` functionality appends a new host source to the

Loading…
Cancel
Save