Sebastian UnterschützandClaude Fable 5.2
4287b7b650
- read netssh.yaml from the user config directory only, not from the working directory - refuse addresses from NetBox or the cache that are neither an IP address nor a hostname (ssh read "-oProxyCommand=…" as an option) - resolver chain stops at an API failure; only "no address" falls through to the next strategy - netssh --help and -h show netssh's help - start at most one background cache refresh per 5 minutes - validate the legacy netbox.url - page through all IP addresses of a host - CI: run format check, vet, and tests on pull requests and main - changelog for v0.0.25 Co-Authored-By: Claude Fable 5.2 <noreply@anthropic.com>
71 lines
2.5 KiB
Go
71 lines
2.5 KiB
Go
package resolver
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"testing"
|
|
|
|
"git.zb-server.de/Sebi/ssh-netbox-wrapper/internal/netbox"
|
|
)
|
|
|
|
// NetBox ignores unknown filters, so the strategy must use the real filter
|
|
// names; otherwise it silently returns the host's first address.
|
|
func TestInterfaceNameStrategyFilters(t *testing.T) {
|
|
for kind, want := range map[string]string{
|
|
"device": "device_id=7&interface=mgmt0",
|
|
"vm": "virtual_machine_id=7&vminterface=mgmt0",
|
|
} {
|
|
var got string
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
q := r.URL.Query()
|
|
q.Del("limit")
|
|
q.Del("offset")
|
|
got = q.Encode()
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.Write([]byte(`{"count":1,"results":[{"address":"10.0.0.7/24"}]}`))
|
|
}))
|
|
ip, err := (&InterfaceNameStrategy{name: "mgmt0"}).Resolve(context.Background(), &netbox.HostEntry{ID: 7, Kind: kind}, netbox.NewClient(srv.URL, "t", 1))
|
|
srv.Close()
|
|
if err != nil || ip != "10.0.0.7" {
|
|
t.Fatalf("%s: %q %v", kind, ip, err)
|
|
}
|
|
if parsed, _ := urlQuery(want); got != parsed {
|
|
t.Errorf("%s: query %q, want %q", kind, got, parsed)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestInterfaceNameStrategyMissingInterfaceFallsThrough(t *testing.T) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
http.Error(w, `{"vminterface":["Select a valid choice."]}`, http.StatusBadRequest)
|
|
}))
|
|
defer srv.Close()
|
|
client := netbox.NewClient(srv.URL, "t", 1)
|
|
chain := &Chain{strategies: []Strategy{&InterfaceNameStrategy{name: "mgmt0"}, &PrimaryIPStrategy{}}}
|
|
ip, err := chain.Resolve(context.Background(), &netbox.HostEntry{ID: 7, Kind: "vm", PrimaryIP4: "10.0.0.9"}, client)
|
|
if err != nil || ip != "10.0.0.9" {
|
|
t.Fatalf("got %q %v", ip, err)
|
|
}
|
|
}
|
|
|
|
func urlQuery(s string) (string, error) {
|
|
q, err := url.ParseQuery(s)
|
|
return q.Encode(), err
|
|
}
|
|
|
|
// Only the HTTP 400 of a missing interface means "no address here"; a server
|
|
// error must stop the chain instead of silently using the next strategy.
|
|
func TestInterfaceNameStrategyServerErrorIsNotNoIP(t *testing.T) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusBadGateway)
|
|
}))
|
|
defer srv.Close()
|
|
_, err := (&InterfaceNameStrategy{name: "mgmt0"}).Resolve(context.Background(), &netbox.HostEntry{ID: 7, Kind: "device"}, netbox.NewClient(srv.URL, "t", 1))
|
|
if err == nil || errors.Is(err, ErrNoIP) {
|
|
t.Fatalf("got %v, want an API error", err)
|
|
}
|
|
}
|