Function adminHandlerFixture

modules/caddyhttp/reverseproxy/admin_test.go:30–56

adminHandlerFixture sets up the global host state for an admin endpoint test and returns a cleanup function that must be deferred by the caller. staticAddrs are inserted into the UsagePool (as a static upstream would be). dynamicAddrs are inserted into the dynamicHosts map (as a dynamic upstream wo

(t *testing.T, staticAddrs, dynamicAddrs []string)

// staticAddrs are inserted into the UsagePool (as a static upstream would be).
// dynamicAddrs are inserted into the dynamicHosts map (as a dynamic upstream would be).
func adminHandlerFixture(t *testing.T, staticAddrs, dynamicAddrs []string) func() {
	t.Helper()

	for _, addr := range staticAddrs {
		u := &Upstream{Dial: addr}
		u.fillHost()
	}

	dynamicHostsMu.Lock()
	for _, addr := range dynamicAddrs {
		dynamicHosts[addr] = dynamicHostEntry{host: new(Host), lastSeen: time.Now()}
	}
	dynamicHostsMu.Unlock()

	return func() {
		// Remove static entries from the UsagePool.
		for _, addr := range staticAddrs {
			_, _ = hosts.Delete(addr)
		}
		// Remove dynamic entries.
		dynamicHostsMu.Lock()
		for _, addr := range dynamicAddrs {
			delete(dynamicHosts, addr)
		}
		dynamicHostsMu.Unlock()
	}
}

// callAdminUpstreams fires a GET against handleUpstreams and returns the
// decoded response body.