mirror of
https://github.com/rcourtman/Pulse.git
synced 2026-04-29 12:00:13 +00:00
65852e8b4a
- Allow HEAD requests in addition to GET for all download handlers (install scripts, binaries, checksums) to prevent 405 errors - Add /uninstall-host-agent.sh to special routes in ServeHTTP - Add test coverage for HEAD request handling - Resolves 'method not allowed' errors during agent installation
119 lines
3.4 KiB
Go
119 lines
3.4 KiB
Go
package api
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func setupTempPulseBin(t *testing.T) string {
|
|
t.Helper()
|
|
dir := t.TempDir()
|
|
t.Setenv("PULSE_BIN_DIR", dir)
|
|
return dir
|
|
}
|
|
|
|
func TestHandleDownloadHostAgentServesWindowsExe(t *testing.T) {
|
|
binDir := setupTempPulseBin(t)
|
|
filePath := filepath.Join(binDir, "pulse-host-agent-windows-unit-test.exe")
|
|
if err := os.WriteFile(filePath, []byte("exe-binary"), 0o755); err != nil {
|
|
t.Fatalf("failed to write test binary: %v", err)
|
|
}
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/download/pulse-host-agent?platform=windows&arch=unit-test", nil)
|
|
rr := httptest.NewRecorder()
|
|
|
|
router := &Router{}
|
|
router.handleDownloadHostAgent(rr, req)
|
|
|
|
if rr.Code != http.StatusOK {
|
|
t.Fatalf("expected 200 OK, got %d", rr.Code)
|
|
}
|
|
|
|
if got := rr.Body.String(); got != "exe-binary" {
|
|
t.Fatalf("unexpected response body: %q", got)
|
|
}
|
|
}
|
|
|
|
func TestHandleDownloadHostAgentServesLinuxArm64(t *testing.T) {
|
|
binDir := setupTempPulseBin(t)
|
|
filePath := filepath.Join(binDir, "pulse-host-agent-linux-arm64")
|
|
payload := []byte("arm64-binary")
|
|
if err := os.WriteFile(filePath, payload, 0o755); err != nil {
|
|
t.Fatalf("failed to write test binary: %v", err)
|
|
}
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/download/pulse-host-agent?platform=linux&arch=arm64", nil)
|
|
rr := httptest.NewRecorder()
|
|
|
|
router := &Router{}
|
|
router.handleDownloadHostAgent(rr, req)
|
|
|
|
if rr.Code != http.StatusOK {
|
|
t.Fatalf("expected 200 OK, got %d", rr.Code)
|
|
}
|
|
|
|
if got := rr.Body.String(); got != string(payload) {
|
|
t.Fatalf("unexpected response body: %q", got)
|
|
}
|
|
}
|
|
|
|
func TestHandleDownloadHostAgentServesChecksumForWindowsExe(t *testing.T) {
|
|
const (
|
|
arch = "unit-sha"
|
|
filename = "pulse-host-agent-windows-" + arch + ".exe"
|
|
)
|
|
binDir := setupTempPulseBin(t)
|
|
filePath := filepath.Join(binDir, filename)
|
|
|
|
payload := []byte("checksum-data")
|
|
if err := os.WriteFile(filePath, payload, 0o755); err != nil {
|
|
t.Fatalf("failed to write test binary: %v", err)
|
|
}
|
|
|
|
req := httptest.NewRequest(http.MethodGet, fmt.Sprintf("/download/pulse-host-agent.sha256?platform=windows&arch=%s", arch), nil)
|
|
rr := httptest.NewRecorder()
|
|
|
|
router := &Router{}
|
|
router.handleDownloadHostAgent(rr, req)
|
|
|
|
if rr.Code != http.StatusOK {
|
|
t.Fatalf("expected 200 OK, got %d", rr.Code)
|
|
}
|
|
|
|
expected := fmt.Sprintf("%x", sha256.Sum256(payload))
|
|
if got := strings.TrimSpace(rr.Body.String()); got != expected {
|
|
t.Fatalf("unexpected checksum body: got %q want %q", got, expected)
|
|
}
|
|
}
|
|
|
|
func TestHandleDownloadHostAgentAllowsHEAD(t *testing.T) {
|
|
binDir := setupTempPulseBin(t)
|
|
filePath := filepath.Join(binDir, "pulse-host-agent-linux-amd64")
|
|
if err := os.WriteFile(filePath, []byte("binary"), 0o755); err != nil {
|
|
t.Fatalf("failed to write test binary: %v", err)
|
|
}
|
|
|
|
req := httptest.NewRequest(http.MethodHead, "/download/pulse-host-agent?platform=linux&arch=amd64", nil)
|
|
rr := httptest.NewRecorder()
|
|
|
|
router := &Router{}
|
|
router.handleDownloadHostAgent(rr, req)
|
|
|
|
if rr.Code != http.StatusOK {
|
|
t.Fatalf("expected 200 OK for HEAD, got %d", rr.Code)
|
|
}
|
|
|
|
// HEAD response should have Content-Length but no body
|
|
// Note: httptest.ResponseRecorder might capture body even for HEAD if handler writes it,
|
|
// but standard http.Server suppresses it.
|
|
// However, our handler uses http.ServeContent which respects HEAD.
|
|
if rr.Body.Len() > 0 {
|
|
t.Fatalf("expected empty body for HEAD, got %d bytes", rr.Body.Len())
|
|
}
|
|
}
|