navi/internal/apphost/handler_test.go

package apphost

import (
	"bytes"
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"testing"

	"vpnclient/internal/config"
	"vpnclient/internal/core"
	"vpnclient/internal/logbuf"
)

func TestHTTPBridgeGetState(t *testing.T) {
	cfg := &config.Config{
		Active: "demo",
		Profiles: []config.Profile{
			{Name: "demo", Protocol: config.ProtocolNaive, Proxy: "https://u:p@example.com"},
		},
	}
	mgr, err := core.NewManager("", cfg, logbuf.New(0))
	if err != nil {
		t.Fatal(err)
	}
	a := New(mgr, "", logbuf.New(0))
	a.APIToken = "test-token"

	body, _ := json.Marshal(map[string]any{"args": []any{}})
	req := httptest.NewRequest(http.MethodPost, "/api/getState", bytes.NewReader(body))
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("X-Navis-Token", "test-token")
	rr := httptest.NewRecorder()
	a.Handler().ServeHTTP(rr, req)
	if rr.Code != http.StatusOK {
		t.Fatalf("status %d body %s", rr.Code, rr.Body.String())
	}
	var resp struct {
		Result UIState `json:"result"`
		Error  string  `json:"error"`
	}
	if err := json.Unmarshal(rr.Body.Bytes(), &resp); err != nil {
		t.Fatal(err)
	}
	if resp.Error != "" {
		t.Fatalf("api error: %s", resp.Error)
	}
	if resp.Result.ActiveProfile != "demo" {
		t.Fatalf("active_profile=%q", resp.Result.ActiveProfile)
	}
}

func TestHTTPBridgeRejectsBadToken(t *testing.T) {
	cfg := &config.Config{}
	mgr, err := core.NewManager("", cfg, logbuf.New(0))
	if err != nil {
		t.Fatal(err)
	}
	a := New(mgr, "", logbuf.New(0))
	a.APIToken = "secret"
	req := httptest.NewRequest(http.MethodPost, "/api/getState", bytes.NewReader([]byte(`{"args":[]}`)))
	req.Header.Set("X-Navis-Token", "wrong")
	rr := httptest.NewRecorder()
	a.Handler().ServeHTTP(rr, req)
	if rr.Code != http.StatusUnauthorized {
		t.Fatalf("want 401, got %d", rr.Code)
	}
}