mirror of
https://github.com/juanfont/headscale.git
synced 2026-04-08 05:57:46 +09:00
9b1a6b6c05
Add ConnectToNetwork to the TailscaleClient interface for multi-network test scenarios and implement peer relay ping support. Use these to test that cap/relay grants correctly enable peer-to-peer relay connections between tagged nodes. Updates #2180
68 lines
2.0 KiB
Go
68 lines
2.0 KiB
Go
package integration
|
|
|
|
import (
|
|
"io"
|
|
"net/netip"
|
|
"net/url"
|
|
"time"
|
|
|
|
"github.com/juanfont/headscale/hscontrol/types"
|
|
"github.com/juanfont/headscale/hscontrol/util"
|
|
"github.com/juanfont/headscale/integration/dockertestutil"
|
|
"github.com/juanfont/headscale/integration/tsic"
|
|
"github.com/ory/dockertest/v3"
|
|
"tailscale.com/ipn/ipnstate"
|
|
"tailscale.com/net/netcheck"
|
|
"tailscale.com/types/key"
|
|
"tailscale.com/types/netmap"
|
|
"tailscale.com/wgengine/filter"
|
|
)
|
|
|
|
// nolint
|
|
type TailscaleClient interface {
|
|
Hostname() string
|
|
Shutdown() (string, string, error)
|
|
Version() string
|
|
Execute(
|
|
command []string,
|
|
options ...dockertestutil.ExecuteCommandOption,
|
|
) (string, string, error)
|
|
Login(loginServer, authKey string) error
|
|
LoginWithURL(loginServer string) (*url.URL, error)
|
|
Logout() error
|
|
Restart() error
|
|
Up() error
|
|
Down() error
|
|
IPs() ([]netip.Addr, error)
|
|
MustIPs() []netip.Addr
|
|
IPv4() (netip.Addr, error)
|
|
MustIPv4() netip.Addr
|
|
MustIPv6() netip.Addr
|
|
FQDN() (string, error)
|
|
MustFQDN() string
|
|
Status(...bool) (*ipnstate.Status, error)
|
|
MustStatus() *ipnstate.Status
|
|
Netmap() (*netmap.NetworkMap, error)
|
|
DebugDERPRegion(region string) (*ipnstate.DebugDERPRegionReport, error)
|
|
GetNodePrivateKey() (*key.NodePrivate, error)
|
|
Netcheck() (*netcheck.Report, error)
|
|
WaitForNeedsLogin(timeout time.Duration) error
|
|
WaitForRunning(timeout time.Duration) error
|
|
WaitForPeers(expected int, timeout, retryInterval time.Duration) error
|
|
Ping(hostnameOrIP string, opts ...tsic.PingOption) error
|
|
Curl(url string, opts ...tsic.CurlOption) (string, error)
|
|
CurlFailFast(url string) (string, error)
|
|
Traceroute(netip.Addr) (util.Traceroute, error)
|
|
ContainerID() string
|
|
MustID() types.NodeID
|
|
ReadFile(path string) ([]byte, error)
|
|
PacketFilter() ([]filter.Match, error)
|
|
ConnectToNetwork(network *dockertest.Network) error
|
|
|
|
// FailingPeersAsString returns a formatted-ish multi-line-string of peers in the client
|
|
// and a bool indicating if the clients online count and peer count is equal.
|
|
FailingPeersAsString() (string, bool, error)
|
|
|
|
WriteLogs(stdout, stderr io.Writer) error
|
|
}
|