helpers

Documentation

Overview

Package helpers implements several helpers for testing.

Index

• func CPUInfo() string
• func HTTPHealthCheck(addr string) func() error
• func HealthCheck(addr string, connTimeout time.Duration, rpcTimeout time.Duration, ...) func() error
• func ListKeys(serverAddress string, prefix string) (map[string]*databasev1.Node, error)
• func MemoryInfo() string
• func OSInfo() string
• func PrintDiskUsage(dir string, maxDepth, curDepth int)
• func TimeRange(args Args, shardContext SharedContext) *modelv1.TimeRange
• func UnmarshalYAML(ii []byte, m proto.Message)
• type Args
• type SharedContext

Functions

func CPUInfo

func CPUInfo() string

CPUInfo returns the CPU information of the current machine.

func HTTPHealthCheck

func HTTPHealthCheck(addr string) func() error

HTTPHealthCheck returns a function for ginkgo "Eventually" poll it repeatedly to check whether a HTTP server is ready.

func HealthCheck

func HealthCheck(addr string, connTimeout time.Duration, rpcTimeout time.Duration, opts ...grpc.DialOption) func() error

HealthCheck returns a function for ginkgo "Eventually" poll it repeatedly to check whether a gRPC server is ready.

func ListKeys

func ListKeys(serverAddress string, prefix string) (map[string]*databasev1.Node, error)

ListKeys lists all keys under the given prefix.

func MemoryInfo

func MemoryInfo() string

MemoryInfo returns the memory information of the current machine.

func OSInfo

func OSInfo() string

OSInfo returns the OS information of the current machine.

func PrintDiskUsage

func PrintDiskUsage(dir string, maxDepth, curDepth int)

PrintDiskUsage prints the disk usage of the specified directory and its subdirectories.

func TimeRange

func TimeRange(args Args, shardContext SharedContext) *modelv1.TimeRange

TimeRange returns a modelv1.TimeRange based on Args and SharedContext.

func UnmarshalYAML

func UnmarshalYAML(ii []byte, m proto.Message)

UnmarshalYAML decodes YAML raw bytes to proto.Message.

Types

type Args

type Args struct {
	Begin     *timestamppb.Timestamp
	End       *timestamppb.Timestamp
	Input     string
	Want      string
	Offset    time.Duration
	Duration  time.Duration
	WantEmpty bool
	WantErr   bool
}

Args is a wrapper seals all necessary info for table specs.

type SharedContext

type SharedContext struct {
	Connection *grpclib.ClientConn
	BaseTime   time.Time
}

SharedContext is the context shared between test cases in the integration testing.