deploy/test/e2e/fixture/account/consequences.go

package project

import (
	"context"
	"errors"
	"time"

	"github.com/stretchr/testify/require"

	"github.com/argoproj/argo-cd/v3/pkg/apiclient/session"

	"github.com/argoproj/argo-cd/v3/pkg/apiclient/account"
	"github.com/argoproj/argo-cd/v3/test/e2e/fixture"
	utilio "github.com/argoproj/argo-cd/v3/util/io"
)

// this implements the "then" part of given/when/then
type Consequences struct {
	context *Context
	actions *Actions
}

func (c *Consequences) And(block func(account *account.Account, err error)) *Consequences {
	c.context.t.Helper()
	block(c.get())
	return c
}

func (c *Consequences) AndCLIOutput(block func(output string, err error)) *Consequences {
	c.context.t.Helper()
	block(c.actions.lastOutput, c.actions.lastError)
	return c
}

func (c *Consequences) CurrentUser(block func(user *session.GetUserInfoResponse, err error)) *Consequences {
	c.context.t.Helper()
	block(c.getCurrentUser())
	return c
}

func (c *Consequences) get() (*account.Account, error) {
	_, accountClient, _ := fixture.ArgoCDClientset.NewAccountClient()
	accList, err := accountClient.ListAccounts(context.Background(), &account.ListAccountRequest{})
	if err != nil {
		return nil, err
	}
	for _, acc := range accList.Items {
		if acc.Name == c.context.name {
			return acc, nil
		}
	}
	return nil, errors.New("account not found")
}

func (c *Consequences) getCurrentUser() (*session.GetUserInfoResponse, error) {
	c.context.t.Helper()
	closer, client, err := fixture.ArgoCDClientset.NewSessionClient()
	require.NoError(c.context.t, err)
	defer utilio.Close(closer)
	return client.GetUserInfo(context.Background(), &session.GetUserInfoRequest{})
}

func (c *Consequences) Given() *Context {
	return c.context
}

func (c *Consequences) When() *Actions {
	time.Sleep(fixture.WhenThenSleepInterval)
	return c.actions
}