testhelper

Documentation

Index

• Constants
• Variables
• func Context(opts ...ContextOpt) (context.Context, func())
• func CopyFile(t testing.TB, src, dst string)
• func CreateGlobalDirectory(t testing.TB, name string) string
• func DiscardTestEntry(tb testing.TB) *log.Entry
• func DiscardTestLogger(tb testing.TB) *log.Logger
• func GenerateCerts(t *testing.T) (string, string)
• func GetLocalhostListener(t testing.TB) (net.Listener, string)
• func GetTemporaryGitalySocketFileName(t testing.TB) string
• func GitLabTestCommit(id string) *gitalypb.GitCommit
• func GitlabTestStoragePath() string
• func MergeIncomingMetadata(ctx context.Context, md ...metadata.MD) context.Context
• func MergeOutgoingMetadata(ctx context.Context, md ...metadata.MD) context.Context
• func ModifyEnvironment(t testing.TB, key string, value string) func()
• func MustClose(t testing.TB, closer io.Closer)
• func MustReadFile(t testing.TB, filename string) []byte
• func MustRunCommand(t testing.TB, stdin io.Reader, name string, args ...string) []byte
• func NewHealthServerWithListener(t testing.TB, listener net.Listener) *health.Server
• func NewServerWithHealth(t testing.TB, socketName string) *health.Server
• func RequireGrpcError(t testing.TB, err error, expectedCode codes.Code)
• func Run(m *testing.M, opts ...RunOption)
• func SetCtxGrpcMethod(ctx context.Context, method string) context.Context
• func SkipWithPraefect(t testing.TB, reason string)
• func TempDir(t testing.TB) string
• func WriteExecutable(t testing.TB, path string, content []byte)
• type Cleanup
• type ContextOpt
  • func ContextWithLogger(logger *log.Entry) ContextOpt
  • func ContextWithTimeout(duration time.Duration) ContextOpt
• type FeatureSet
  • func (f FeatureSet) Desc() string
  • func (f FeatureSet) Disable(ctx context.Context) context.Context
• type FeatureSets
  • func NewFeatureSets(goFeatures []featureflag.FeatureFlag, rubyFeatures ...featureflag.FeatureFlag) FeatureSets
  • func (s FeatureSets) Bench(b *testing.B, test func(b *testing.B, ctx context.Context))
  • func (s FeatureSets) Run(t *testing.T, test func(t *testing.T, ctx context.Context))
• type RunOption
  • func WithDisabledGoroutineChecker() RunOption
  • func WithSetup(setup func() error) RunOption

Constants

const (
	// RepositoryAuthToken is the default token used to authenticate
	// against other Gitaly servers. It is inject as part of the
	// GitalyServers metadata.
	RepositoryAuthToken = "the-secret-token"
	// DefaultStorageName is the default name of the Gitaly storage.
	DefaultStorageName = "default"
)

Variables

var NewTestLogger = DiscardTestLogger

NewTestLogger creates logger that should be used in the tests.

Functions

func Context

func Context(opts ...ContextOpt) (context.Context, func())

Context returns a cancellable context.

func CopyFile

func CopyFile(t testing.TB, src, dst string)

CopyFile copies a file at the path src to a file at the path dst

func CreateGlobalDirectory

func CreateGlobalDirectory(t testing.TB, name string) string

CreateGlobalDirectory creates a directory in the test directory that is shared across all between all tests.

func DiscardTestEntry

func DiscardTestEntry(tb testing.TB) *log.Entry

DiscardTestEntry creates a logrus entry that discards everything.

func DiscardTestLogger

func DiscardTestLogger(tb testing.TB) *log.Logger

DiscardTestLogger created a logrus hook that discards everything.

func GenerateCerts

func GenerateCerts(t *testing.T) (string, string)

GenerateCerts creates a certificate that can be used to establish TLS protected TCP connection. It returns paths to the file with the certificate and its private key.

func GetLocalhostListener

func GetLocalhostListener(t testing.TB) (net.Listener, string)

GetLocalhostListener listens on the next available TCP port and returns the listener and the localhost address (host:port) string.

func GetTemporaryGitalySocketFileName

func GetTemporaryGitalySocketFileName(t testing.TB) string

GetTemporaryGitalySocketFileName will return a unique, useable socket file name

func GitLabTestCommit

func GitLabTestCommit(id string) *gitalypb.GitCommit

GitLabTestCommit provides a key value lookup for commits in the GitLab-Test repository

func GitlabTestStoragePath

func GitlabTestStoragePath() string

GitlabTestStoragePath returns the storage path to the gitlab-test repo.

func MergeIncomingMetadata

func MergeIncomingMetadata(ctx context.Context, md ...metadata.MD) context.Context

MergeIncomingMetadata merges provided metadata-s and returns context with resulting value.

func MergeOutgoingMetadata

func MergeOutgoingMetadata(ctx context.Context, md ...metadata.MD) context.Context

MergeOutgoingMetadata merges provided metadata-s and returns context with resulting value.

func ModifyEnvironment

func ModifyEnvironment(t testing.TB, key string, value string) func()

ModifyEnvironment will change an environment variable and return a func suitable for `defer` to change the value back. If the given value is empty, then the envvar will be unset.

func MustClose

func MustClose(t testing.TB, closer io.Closer)

MustClose calls Close() on the Closer and fails the test in case it returns an error. This function is useful when closing via `defer`, as a simple `defer require.NoError(t, closer.Close())` would cause `closer.Close()` to be executed early already.

func MustReadFile

func MustReadFile(t testing.TB, filename string) []byte

MustReadFile returns the content of a file or fails at once.

func MustRunCommand

func MustRunCommand(t testing.TB, stdin io.Reader, name string, args ...string) []byte

MustRunCommand runs a command with an optional standard input and returns the standard output, or fails.

func NewHealthServerWithListener

func NewHealthServerWithListener(t testing.TB, listener net.Listener) *health.Server

NewHealthServerWithListener creates a new gRPC server with the health server set up. It will listen on the given listener.

func NewServerWithHealth

func NewServerWithHealth(t testing.TB, socketName string) *health.Server

NewServerWithHealth creates a new gRPC server with the health server set up. It will listen on the socket identified by `socketName`.

func RequireGrpcError

func RequireGrpcError(t testing.TB, err error, expectedCode codes.Code)

RequireGrpcError asserts the passed err is of the same code as expectedCode.

func Run

func Run(m *testing.M, opts ...RunOption)

Run sets up required testing state and executes the given test suite. It can optionally receive a variable number of RunOptions.

func SetCtxGrpcMethod

func SetCtxGrpcMethod(ctx context.Context, method string) context.Context

SetCtxGrpcMethod will set the gRPC context value for the proper key responsible for an RPC full method name. This directly corresponds to the gRPC function responsible for extracting the method: https://godoc.org/google.golang.org/grpc#Method

func SkipWithPraefect

func SkipWithPraefect(t testing.TB, reason string)

SkipWithPraefect skips the test if it is being executed with Praefect in front of the Gitaly.

func TempDir

func TempDir(t testing.TB) string

TempDir is a wrapper around os.MkdirTemp that provides a cleanup function.

func WriteExecutable

func WriteExecutable(t testing.TB, path string, content []byte)

WriteExecutable ensures that the parent directory exists, and writes an executable with provided content

Types

type Cleanup

type Cleanup func()

Cleanup functions should be called in a defer statement immediately after they are returned from a test helper

type ContextOpt

type ContextOpt func(context.Context) (context.Context, func())

ContextOpt returns a new context instance with the new additions to it.

func ContextWithLogger

func ContextWithLogger(logger *log.Entry) ContextOpt

ContextWithLogger allows to inject provided logger into the context.

func ContextWithTimeout

func ContextWithTimeout(duration time.Duration) ContextOpt

ContextWithTimeout allows to set provided timeout to the context.

type FeatureSet

type FeatureSet struct {
	// contains filtered or unexported fields
}

FeatureSet is a representation of a set of features that should be disabled. This is useful in situations where a test needs to test any combination of features toggled on and off. It is designed to disable features as all features are enabled by default, please see: testhelper.Context()

func (FeatureSet) Desc

func (f FeatureSet) Desc() string

Desc describes the feature such that it is suitable as a testcase description.

func (FeatureSet) Disable

func (f FeatureSet) Disable(ctx context.Context) context.Context

Disable disables all feature flags in the given FeatureSet in the given context. The context is treated as an outgoing context.

type FeatureSets

type FeatureSets []FeatureSet

FeatureSets is a slice containing many FeatureSets

func NewFeatureSets

func NewFeatureSets(goFeatures []featureflag.FeatureFlag, rubyFeatures ...featureflag.FeatureFlag) FeatureSets

NewFeatureSets takes a slice of go feature flags, and an optional variadic set of ruby feature flags and returns a FeatureSets slice

func (FeatureSets) Bench

func (s FeatureSets) Bench(b *testing.B, test func(b *testing.B, ctx context.Context))

Bench executes the given benchmarking function for each of the FeatureSets. The passed in context has the feature flags set accordingly.

func (FeatureSets) Run

func (s FeatureSets) Run(t *testing.T, test func(t *testing.T, ctx context.Context))

Run executes the given test function for each of the FeatureSets. The passed in context has the feature flags set accordingly.

type RunOption

type RunOption func(*runConfig)

RunOption is an option that can be passed to Run.

func WithDisabledGoroutineChecker

func WithDisabledGoroutineChecker() RunOption

WithDisabledGoroutineChecker disables checking for leaked Goroutines after tests have run. This should ideally only be used as a temporary measure until all Goroutine leaks have been fixed.

Deprecated: This should not be used, but instead you should try to fix all Goroutine leakages.

func WithSetup

func WithSetup(setup func() error) RunOption

WithSetup allows the caller of Run to pass a setup function that will be called after global test state has been configured.