63 lines
2.0 KiB
Go
63 lines
2.0 KiB
Go
package testenv
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
// Offline keeps the package's tests off the network: http.DefaultTransport
|
|
// (and so http.DefaultClient, and every transport cloned from it after the
|
|
// call) dials this computer only — localhost, a name under .localhost, a
|
|
// loopback or unspecified address — and refuses anything else before it is
|
|
// looked up, saying so on stderr once per address. A test that wants a
|
|
// vendor serves it on loopback and points the request there (an
|
|
// httptest.Server, a Transport rewriting the host). Without it the tests
|
|
// of internal/gateway and internal/provider asked the real models.dev (its
|
|
// catalog, fetched as each plugin host started), api.github.com,
|
|
// chatgpt.com and Devin's server.codeium.com with made-up accounts, and a
|
|
// plan key with a vendor's real base URL would have its windows read for
|
|
// routing (#1016). A package calls it from its TestMain, before m.Run.
|
|
func Offline() {
|
|
t, ok := http.DefaultTransport.(*http.Transport)
|
|
if !ok {
|
|
return
|
|
}
|
|
dial := t.DialContext
|
|
if dial == nil {
|
|
dial = (&net.Dialer{}).DialContext
|
|
}
|
|
var mu sync.Mutex
|
|
said := map[string]bool{}
|
|
t.DialContext = func(ctx context.Context, network, addr string) (net.Conn, error) {
|
|
if local(addr) {
|
|
return dial(ctx, network, addr)
|
|
}
|
|
mu.Lock()
|
|
if !said[addr] {
|
|
said[addr] = true
|
|
fmt.Fprintf(os.Stderr, "testenv: a test asked %s, refused: tests stay off the network\n", addr)
|
|
}
|
|
mu.Unlock()
|
|
return nil, fmt.Errorf("testenv: %s is not this computer; tests stay off the network", addr)
|
|
}
|
|
}
|
|
|
|
// local says addr, a host:port, is this computer, without looking it up.
|
|
func local(addr string) bool {
|
|
host, _, err := net.SplitHostPort(addr)
|
|
if err != nil {
|
|
host = addr
|
|
}
|
|
host = strings.ToLower(strings.TrimSuffix(host, "."))
|
|
if host == "localhost" || strings.HasSuffix(host, ".localhost") {
|
|
return true
|
|
}
|
|
ip := net.ParseIP(strings.Trim(host, "[]"))
|
|
return ip != nil && (ip.IsLoopback() || ip.IsUnspecified())
|
|
}
|