mirror of
https://github.com/rcourtman/Pulse.git
synced 2026-02-18 00:17:39 +01:00
Enforce host config token binding
This commit is contained in:
46
internal/api/host_agent_config_binding_router_test.go
Normal file
46
internal/api/host_agent_config_binding_router_test.go
Normal file
@@ -0,0 +1,46 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/rcourtman/pulse-go-rewrite/internal/config"
|
||||
"github.com/rcourtman/pulse-go-rewrite/internal/models"
|
||||
"github.com/rcourtman/pulse-go-rewrite/internal/monitoring"
|
||||
)
|
||||
|
||||
func TestHostAgentConfigUsesTokenBindingInRouter(t *testing.T) {
|
||||
rawToken := "host-config-binding-token-123.12345678"
|
||||
record := newTokenRecord(t, rawToken, []string{config.ScopeHostConfigRead}, nil)
|
||||
record.ID = "token-1"
|
||||
|
||||
cfg := newTestConfigWithTokens(t, record)
|
||||
|
||||
state := models.NewState()
|
||||
state.UpsertHost(models.Host{ID: "host-1", TokenID: "token-1"})
|
||||
monitor := &monitoring.Monitor{}
|
||||
setUnexportedField(t, monitor, "state", state)
|
||||
|
||||
router := NewRouter(cfg, monitor, nil, nil, nil, "1.0.0")
|
||||
|
||||
req := httptest.NewRequest(http.MethodGet, "/api/agents/host/host-2/config", nil)
|
||||
req.Header.Set("X-API-Token", rawToken)
|
||||
rec := httptest.NewRecorder()
|
||||
router.Handler().ServeHTTP(rec, req)
|
||||
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("expected 200 for bound host config, got %d", rec.Code)
|
||||
}
|
||||
|
||||
var resp struct {
|
||||
HostID string `json:"hostId"`
|
||||
}
|
||||
if err := json.NewDecoder(rec.Body).Decode(&resp); err != nil {
|
||||
t.Fatalf("decode response: %v", err)
|
||||
}
|
||||
if resp.HostID != "host-1" {
|
||||
t.Fatalf("expected host id %q, got %q", "host-1", resp.HostID)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user