testutil

Documentation

Overview

Package testutil contains helper functions for writing tests.

Index

• func Diff(x, y interface{}, opts ...cmp.Option) string
• func Equal(x, y interface{}, opts ...cmp.Option) bool
• func JWTConfig() (*jwt.Config, error)
• func PageBounds(pageSize int, pageToken string, length int) (from, to int, nextPageToken string, err error)
• func ProjID() string
• func TokenSource(ctx context.Context, scopes ...string) oauth2.TokenSource
• func TokenSourceEnv(ctx context.Context, envVar string, scopes ...string) oauth2.TokenSource
• type Server
  • func NewServer(opts ...grpc.ServerOption) (*Server, error)
  • func (s *Server) Close()
  • func (s *Server) Start()
• type TestExporter
  • func NewTestExporter() *TestExporter
  • func (te *TestExporter) ExportSpan(s *trace.SpanData)
  • func (te *TestExporter) ExportView(vd *view.Data)
  • func (te *TestExporter) Unregister()
• type UIDSpace
  • func NewUIDSpace(prefix string) *UIDSpace
  • func NewUIDSpaceSep(prefix string, sep rune) *UIDSpace
  • func (s *UIDSpace) New() string
  • func (s *UIDSpace) Older(uid string, d time.Duration) bool
  • func (s *UIDSpace) Timestamp(uid string) (time.Time, bool)

Functions

func Diff

func Diff(x, y interface{}, opts ...cmp.Option) string

Diff reports the differences between two values. Diff(x, y) == "" iff Equal(x, y).

func Equal

func Equal(x, y interface{}, opts ...cmp.Option) bool

Equal tests two values for equality.

func JWTConfig

func JWTConfig() (*jwt.Config, error)

JWTConfig reads the JSON private key file whose name is in the default environment variable, and returns the jwt.Config it contains. It ignores scopes. If the environment variable is empty, it returns (nil, nil).

func PageBounds

func PageBounds(pageSize int, pageToken string, length int) (from, to int, nextPageToken string, err error)

PageBounds converts an incoming page size and token from an RPC request into slice bounds and the outgoing next-page token.

PageBounds assumes that the complete, unpaginated list of items exists as a single slice. In addition to the page size and token, PageBounds needs the length of that slice.

PageBounds's first two return values should be used to construct a sub-slice of the complete, unpaginated slice. E.g. if the complete slice is s, then s[from:to] is the desired page. Its third return value should be set as the NextPageToken field of the RPC response.

func ProjID

func ProjID() string

ProjID returns the project ID to use in integration tests, or the empty string if none is configured.

func TokenSource

func TokenSource(ctx context.Context, scopes ...string) oauth2.TokenSource

TokenSource returns the OAuth2 token source to use in integration tests, or nil if none is configured. It uses the standard environment variable for tests in this repo.

func TokenSourceEnv

func TokenSourceEnv(ctx context.Context, envVar string, scopes ...string) oauth2.TokenSource

TokenSourceEnv returns the OAuth2 token source to use in integration tests. or nil if none is configured. It tries to get credentials from the filename in the environment variable envVar. If the environment variable is unset, TokenSourceEnv will try to find 'Application Default Credentials'. Else, TokenSourceEnv will return nil. TokenSourceEnv will log.Fatal if the token source is specified but missing or invalid.

Types

type Server

type Server struct {
	Addr string

	Gsrv *grpc.Server
	// contains filtered or unexported fields
}

A Server is an in-process gRPC server, listening on a system-chosen port on the local loopback interface. Servers are for testing only and are not intended to be used in production code.

To create a server, make a new Server, register your handlers, then call Start:

srv, err := NewServer()
...
mypb.RegisterMyServiceServer(srv.Gsrv, &myHandler)
....
srv.Start()

Clients should connect to the server with no security:

conn, err := grpc.Dial(srv.Addr, grpc.WithInsecure())
...

func NewServer

func NewServer(opts ...grpc.ServerOption) (*Server, error)

NewServer creates a new Server. The Server will be listening for gRPC connections at the address named by the Addr field, without TLS.

func (*Server) Close

func (s *Server) Close()

Close shuts down the server.

func (*Server) Start

func (s *Server) Start()

Start causes the server to start accepting incoming connections. Call Start after registering handlers.

type TestExporter

type TestExporter struct {
	Spans []*trace.SpanData
	Stats chan *view.Data
}

func NewTestExporter

func NewTestExporter() *TestExporter

func (*TestExporter) ExportSpan

func (te *TestExporter) ExportSpan(s *trace.SpanData)

func (*TestExporter) ExportView

func (te *TestExporter) ExportView(vd *view.Data)

func (*TestExporter) Unregister

func (te *TestExporter) Unregister()

type UIDSpace

type UIDSpace struct {
	Prefix string
	Sep    rune
	// contains filtered or unexported fields
}

A UIDSpace manages a set of unique IDs distinguished by a prefix.

func NewUIDSpace

func NewUIDSpace(prefix string) *UIDSpace

func NewUIDSpaceSep

func NewUIDSpaceSep(prefix string, sep rune) *UIDSpace

func (*UIDSpace) New

func (s *UIDSpace) New() string

New generates a new unique ID . The ID consists of the UIDSpace's prefix, a timestamp, and a counter value. All unique IDs generated in the same test execution will have the same timestamp.

Aside from the characters in the prefix, IDs contain only letters, numbers and sep.

func (*UIDSpace) Older

func (s *UIDSpace) Older(uid string, d time.Duration) bool

Older reports whether uid was created by m and has a timestamp older than the current time by at least d.

func (*UIDSpace) Timestamp

func (s *UIDSpace) Timestamp(uid string) (time.Time, bool)

Timestamp extracts the timestamp of uid, which must have been generated by s. The second return value is true on success, false if there was a problem.