relaytest

Documentation

Index

• func HostFunc(fn func(relay.CallFrame, *relay.Conn) (string, error)) tchannel.RelayHost
• type FluentMockCallStats
  • func (f *FluentMockCallStats) Failed(reason string) *FluentMockCallStats
  • func (f *FluentMockCallStats) Succeeded() *FluentMockCallStats
• type MockCallStats
  • func (m *MockCallStats) End()
  • func (m *MockCallStats) Failed(reason string)
  • func (m *MockCallStats) ReceivedBytes(size uint16)
  • func (m *MockCallStats) SentBytes(size uint16)
  • func (m *MockCallStats) Succeeded()
• type MockStats
  • func NewMockStats() *MockStats
  • func (m *MockStats) Add(caller, callee, procedure string) *FluentMockCallStats
  • func (m *MockStats) AssertEqual(t testing.TB, expected *MockStats)
  • func (m *MockStats) Begin(f relay.CallFrame) *MockCallStats
  • func (m *MockStats) Map() map[string]int
  • func (m *MockStats) WaitForEnd()
• type StubRelayHost
  • func NewStubRelayHost() *StubRelayHost
  • func (rh *StubRelayHost) Add(service, hostPort string)
  • func (rh *StubRelayHost) SetChannel(ch *tchannel.Channel)
  • func (rh *StubRelayHost) SetFrameFn(f func(relay.CallFrame, *relay.Conn))
  • func (rh *StubRelayHost) SetRespFrameFn(f func(relay.RespFrame))
  • func (rh *StubRelayHost) Start(cf relay.CallFrame, conn *relay.Conn) (tchannel.RelayCall, error)
  • func (rh *StubRelayHost) Stats() *MockStats

Functions

func HostFunc

func HostFunc(fn func(relay.CallFrame, *relay.Conn) (string, error)) tchannel.RelayHost

HostFunc wraps a given function to implement tchannel.RelayHost.

Types

type FluentMockCallStats

type FluentMockCallStats struct {
	*MockCallStats
}

FluentMockCallStats wraps the MockCallStats in a fluent API that's convenient for tests.

func (*FluentMockCallStats) Failed

func (f *FluentMockCallStats) Failed(reason string) *FluentMockCallStats

Failed marks the RPC as failed.

func (*FluentMockCallStats) Succeeded

func (f *FluentMockCallStats) Succeeded() *FluentMockCallStats

Succeeded marks the RPC as succeeded.

type MockCallStats

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

MockCallStats is a testing spy for the CallStats interface.

func (*MockCallStats) End

func (m *MockCallStats) End()

End halts timer and metric collection for the RPC.

func (*MockCallStats) Failed

func (m *MockCallStats) Failed(reason string)

Failed marks the RPC as failed for the provided reason.

func (*MockCallStats) ReceivedBytes

func (m *MockCallStats) ReceivedBytes(size uint16)

ReceivedBytes tracks the received bytes.

func (*MockCallStats) SentBytes

func (m *MockCallStats) SentBytes(size uint16)

SentBytes tracks the sent bytes.

func (*MockCallStats) Succeeded

func (m *MockCallStats) Succeeded()

Succeeded marks the RPC as succeeded.

type MockStats

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

MockStats is a testing spy for the Stats interface.

func NewMockStats

func NewMockStats() *MockStats

NewMockStats constructs a MockStats.

func (*MockStats) Add

func (m *MockStats) Add(caller, callee, procedure string) *FluentMockCallStats

Add explicitly adds a new call along an edge of the call graph.

func (*MockStats) AssertEqual

func (m *MockStats) AssertEqual(t testing.TB, expected *MockStats)

AssertEqual asserts that two MockStats describe the same call graph.

func (*MockStats) Begin

func (m *MockStats) Begin(f relay.CallFrame) *MockCallStats

Begin starts collecting metrics for an RPC.

func (*MockStats) Map

func (m *MockStats) Map() map[string]int

Map returns all stats as a map of key to int. It waits for any ongoing calls to end first to avoid races.

func (*MockStats) WaitForEnd

func (m *MockStats) WaitForEnd()

WaitForEnd waits for all calls to End.

type StubRelayHost

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

StubRelayHost is a stub RelayHost for tests that backs peer selection to an underlying channel using isolated subchannels and the default peer selection.

func NewStubRelayHost

func NewStubRelayHost() *StubRelayHost

NewStubRelayHost creates a new stub RelayHost for tests.

func (*StubRelayHost) Add

func (rh *StubRelayHost) Add(service, hostPort string)

Add adds a service instance with the specified host:port.

func (*StubRelayHost) SetChannel

func (rh *StubRelayHost) SetChannel(ch *tchannel.Channel)

SetChannel is called by the channel after creation so we can get a reference to the channels' peers.

func (*StubRelayHost) SetFrameFn

func (rh *StubRelayHost) SetFrameFn(f func(relay.CallFrame, *relay.Conn))

SetFrameFn sets a function to run on every frame.

func (*StubRelayHost) SetRespFrameFn

func (rh *StubRelayHost) SetRespFrameFn(f func(relay.RespFrame))

SetRespFrameFn sets a function to run on every frame.

func (*StubRelayHost) Start

func (rh *StubRelayHost) Start(cf relay.CallFrame, conn *relay.Conn) (tchannel.RelayCall, error)

Start starts a new RelayCall for the given call on a specific connection.

func (*StubRelayHost) Stats

func (rh *StubRelayHost) Stats() *MockStats

Stats returns the *MockStats tracked for this channel.