refactor(scaletest): add runner for creating users (#19811)

Closes https://github.com/coder/internal/issues/985

Simple refactor of the user creation logic into it's own test runner. This lets us create users independently of workspaces, for use in a bunch of load generators, including the Coder Connect load generator.

This PR creates the new runner, and has the existing `createworkspaces` runner use it.

scaletest/createusers/config.go

@@ -0,0 +1,23 @@
+package createusers
+
+import (
+	"github.com/google/uuid"
+	"golang.org/x/xerrors"
+)
+
+type Config struct {
+	// OrganizationID is the ID of the organization to add the user to.
+	OrganizationID uuid.UUID `json:"organization_id"`
+	// Username is the username of the new user. Generated if empty.
+	Username string `json:"username"`
+	// Email is the email of the new user. Generated if empty.
+	Email string `json:"email"`
+}
+
+func (c Config) Validate() error {
+	if c.OrganizationID == uuid.Nil {
+		return xerrors.New("organization_id must not be a nil UUID")
+	}
+
+	return nil
+}

scaletest/createusers/run.go

@@ -0,0 +1,106 @@
+package createusers
+
+import (
+	"context"
+	"fmt"
+	"io"
+
+	"github.com/google/uuid"
+	"golang.org/x/xerrors"
+
+	"cdr.dev/slog"
+	"cdr.dev/slog/sloggers/sloghuman"
+
+	"github.com/coder/coder/v2/coderd/tracing"
+	"github.com/coder/coder/v2/codersdk"
+	"github.com/coder/coder/v2/cryptorand"
+	"github.com/coder/coder/v2/scaletest/loadtestutil"
+)
+
+type Runner struct {
+	client *codersdk.Client
+	cfg    Config
+
+	user codersdk.User
+}
+
+type User struct {
+	codersdk.User
+	SessionToken string
+}
+
+func NewRunner(client *codersdk.Client, cfg Config) *Runner {
+	return &Runner{
+		client: client,
+		cfg:    cfg,
+	}
+}
+
+func (r *Runner) RunReturningUser(ctx context.Context, id string, logs io.Writer) (User, error) {
+	ctx, span := tracing.StartSpan(ctx)
+	defer span.End()
+
+	logs = loadtestutil.NewSyncWriter(logs)
+	logger := slog.Make(sloghuman.Sink(logs)).Leveled(slog.LevelDebug)
+	r.client.SetLogger(logger)
+	r.client.SetLogBodies(true)
+
+	if r.cfg.Username == "" || r.cfg.Email == "" {
+		genUsername, genEmail, err := loadtestutil.GenerateUserIdentifier(id)
+		if err != nil {
+			return User{}, xerrors.Errorf("generate user identifier: %w", err)
+		}
+		if r.cfg.Username == "" {
+			r.cfg.Username = genUsername
+		}
+		if r.cfg.Email == "" {
+			r.cfg.Email = genEmail
+		}
+	}
+
+	_, _ = fmt.Fprintln(logs, "Generating user password...")
+	password, err := cryptorand.String(16)
+	if err != nil {
+		return User{}, xerrors.Errorf("generate random password for user: %w", err)
+	}
+
+	_, _ = fmt.Fprintln(logs, "Creating user:")
+	user, err := r.client.CreateUserWithOrgs(ctx, codersdk.CreateUserRequestWithOrgs{
+		OrganizationIDs: []uuid.UUID{r.cfg.OrganizationID},
+		Username:        r.cfg.Username,
+		Email:           r.cfg.Email,
+		Password:        password,
+	})
+	if err != nil {
+		return User{}, xerrors.Errorf("create user: %w", err)
+	}
+	r.user = user
+
+	_, _ = fmt.Fprintln(logs, "\nLogging in as new user...")
+	client := codersdk.New(r.client.URL)
+	loginRes, err := client.LoginWithPassword(ctx, codersdk.LoginWithPasswordRequest{
+		Email:    r.cfg.Email,
+		Password: password,
+	})
+	if err != nil {
+		return User{}, xerrors.Errorf("login as new user: %w", err)
+	}
+
+	_, _ = fmt.Fprintf(logs, "\tOrg ID:   %s\n", r.cfg.OrganizationID.String())
+	_, _ = fmt.Fprintf(logs, "\tUsername: %s\n", user.Username)
+	_, _ = fmt.Fprintf(logs, "\tEmail:    %s\n", user.Email)
+	_, _ = fmt.Fprintf(logs, "\tPassword: ****************\n")
+
+	return User{User: user, SessionToken: loginRes.SessionToken}, nil
+}
+
+func (r *Runner) Cleanup(ctx context.Context, _ string, logs io.Writer) error {
+	if r.user.ID != uuid.Nil {
+		err := r.client.DeleteUser(ctx, r.user.ID)
+		if err != nil {
+			_, _ = fmt.Fprintf(logs, "failed to delete user %q: %v\n", r.user.ID.String(), err)
+			return xerrors.Errorf("delete user: %w", err)
+		}
+	}
+	return nil
+}