97 lines
1.8 KiB
Go
97 lines
1.8 KiB
Go
|
package testutil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
type WrappedServer struct {
|
||
|
|
s *TestServer
|
||
|
|
t *testing.T
|
||
|
|
}
|
||
|
|
|
||
|
|
// Wrap wraps the test server in a `testing.t` for convenience.
|
||
|
|
//
|
||
|
|
// For example, the following code snippets are equivalent.
|
||
|
|
//
|
||
|
|
// server.JoinLAN(t, "1.2.3.4")
|
||
|
|
// server.Wrap(t).JoinLAN("1.2.3.4")
|
||
|
|
//
|
||
|
|
// This is useful when you are calling multiple functions and save the wrapped
|
||
|
|
// value as another variable to reduce the inclusion of "t".
|
||
|
|
func (s *TestServer) Wrap(t *testing.T) *WrappedServer {
|
||
|
|
return &WrappedServer{
|
||
|
|
s: s,
|
||
|
|
t: t,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// See Also
|
||
|
|
//
|
||
|
|
// TestServer.JoinLAN()
|
||
|
|
func (w *WrappedServer) JoinLAN(addr string) {
|
||
|
|
w.s.JoinLAN(w.t, addr)
|
||
|
|
}
|
||
|
|
|
||
|
|
// See Also
|
||
|
|
//
|
||
|
|
// TestServer.JoinWAN()
|
||
|
|
func (w *WrappedServer) JoinWAN(addr string) {
|
||
|
|
w.s.JoinWAN(w.t, addr)
|
||
|
|
}
|
||
|
|
|
||
|
|
// See Also
|
||
|
|
//
|
||
|
|
// TestServer.SetKV()
|
||
|
|
func (w *WrappedServer) SetKV(key string, val []byte) {
|
||
|
|
w.s.SetKV(w.t, key, val)
|
||
|
|
}
|
||
|
|
|
||
|
|
// See Also
|
||
|
|
//
|
||
|
|
// TestServer.SetKVString()
|
||
|
|
func (w *WrappedServer) SetKVString(key string, val string) {
|
||
|
|
w.s.SetKVString(w.t, key, val)
|
||
|
|
}
|
||
|
|
|
||
|
|
// See Also
|
||
|
|
//
|
||
|
|
// TestServer.GetKV()
|
||
|
|
func (w *WrappedServer) GetKV(key string) []byte {
|
||
|
|
return w.s.GetKV(w.t, key)
|
||
|
|
}
|
||
|
|
|
||
|
|
// See Also
|
||
|
|
//
|
||
|
|
// TestServer.GetKVString()
|
||
|
|
func (w *WrappedServer) GetKVString(key string) string {
|
||
|
|
return w.s.GetKVString(w.t, key)
|
||
|
|
}
|
||
|
|
|
||
|
|
// See Also
|
||
|
|
//
|
||
|
|
// TestServer.PopulateKV()
|
||
|
|
func (w *WrappedServer) PopulateKV(data map[string][]byte) {
|
||
|
|
w.s.PopulateKV(w.t, data)
|
||
|
|
}
|
||
|
|
|
||
|
|
// See Also
|
||
|
|
//
|
||
|
|
// TestServer.ListKV()
|
||
|
|
func (w *WrappedServer) ListKV(prefix string) []string {
|
||
|
|
return w.s.ListKV(w.t, prefix)
|
||
|
|
}
|
||
|
|
|
||
|
|
// See Also
|
||
|
|
//
|
||
|
|
// TestServer.AddService()
|
||
|
|
func (w *WrappedServer) AddService(name, status string, tags []string) {
|
||
|
|
w.s.AddService(w.t, name, status, tags)
|
||
|
|
}
|
||
|
|
|
||
|
|
// See Also
|
||
|
|
//
|
||
|
|
// TestServer.AddCheck()
|
||
|
|
func (w *WrappedServer) AddCheck(name, serviceID, status string) {
|
||
|
|
w.s.AddCheck(w.t, name, serviceID, status)
|
||
|
|
}
|