Extend report engine and cluster client coverage

pkg/proxmox/cluster_client_additional_api5_test.go

@@ -0,0 +1,63 @@
+package proxmox
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestClusterClient_GetStorageContentFiltersBackup(t *testing.T) {
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Content-Type", "application/json")
+		switch r.URL.Path {
+		case "/api2/json/nodes":
+			fmt.Fprint(w, `{"data":[{"node":"node1","status":"online"}]}`)
+		case "/api2/json/nodes/node1/storage/local/content":
+			fmt.Fprint(w, `{"data":[{"volid":"local:backup/ct-100.tar","content":"backup"},{"volid":"local:iso/ubuntu.iso","content":"iso"}]}`)
+		default:
+			w.WriteHeader(http.StatusNotFound)
+		}
+	}))
+	defer server.Close()
+
+	cfg := ClientConfig{Host: server.URL, TokenName: "u@p!t", TokenValue: "v"}
+	cc := NewClusterClient("test", cfg, []string{server.URL}, nil)
+
+	content, err := cc.GetStorageContent(context.Background(), "node1", "local")
+	if err != nil {
+		t.Fatalf("GetStorageContent failed: %v", err)
+	}
+	if len(content) != 1 || content[0].Content != "backup" {
+		t.Fatalf("unexpected storage content: %+v", content)
+	}
+}
+
+func TestClusterClient_GetNodePendingUpdates_PermissionDenied(t *testing.T) {
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Content-Type", "application/json")
+		if r.URL.Path == "/api2/json/nodes" {
+			fmt.Fprint(w, `{"data":[{"node":"node1","status":"online"}]}`)
+			return
+		}
+		if r.URL.Path == "/api2/json/nodes/node1/apt/update" {
+			w.WriteHeader(http.StatusForbidden)
+			fmt.Fprint(w, "permission denied")
+			return
+		}
+		w.WriteHeader(http.StatusNotFound)
+	}))
+	defer server.Close()
+
+	cfg := ClientConfig{Host: server.URL, TokenName: "u@p!t", TokenValue: "v"}
+	cc := NewClusterClient("test", cfg, []string{server.URL}, nil)
+
+	updates, err := cc.GetNodePendingUpdates(context.Background(), "node1")
+	if err != nil {
+		t.Fatalf("expected permission error to be swallowed, got %v", err)
+	}
+	if len(updates) != 0 {
+		t.Fatalf("expected empty updates on permission error, got %+v", updates)
+	}
+}

pkg/reporting/engine_additional2_test.go

@@ -0,0 +1,67 @@
+package reporting
+
+import (
+	"strings"
+	"testing"
+	"time"
+)
+
+func TestReportEngineGenerateMultiMissingStore(t *testing.T) {
+	engine := NewReportEngine(EngineConfig{})
+
+	_, _, err := engine.GenerateMulti(MultiReportRequest{
+		Resources: []MetricReportRequest{{ResourceType: "node", ResourceID: "node-1"}},
+		Start:     time.Now().Add(-1 * time.Hour),
+		End:       time.Now(),
+		Format:    FormatCSV,
+	})
+	if err == nil {
+		t.Fatal("expected error when metrics store is nil")
+	}
+	if !strings.Contains(err.Error(), "metrics store not initialized") {
+		t.Fatalf("unexpected error: %v", err)
+	}
+}
+
+func TestReportEngineGenerateMultiUnsupportedFormat(t *testing.T) {
+	store := newTestMetricsStore(t)
+	defer store.Close()
+
+	engine := NewReportEngine(EngineConfig{MetricsStore: store})
+
+	_, _, err := engine.GenerateMulti(MultiReportRequest{
+		Resources: []MetricReportRequest{{ResourceType: "node", ResourceID: "node-1"}},
+		Start:     time.Now().Add(-1 * time.Hour),
+		End:       time.Now(),
+		Format:    ReportFormat("xls"),
+	})
+	if err == nil {
+		t.Fatal("expected unsupported format error")
+	}
+	if !strings.Contains(err.Error(), "unsupported format") {
+		t.Fatalf("unexpected error: %v", err)
+	}
+}
+
+func TestReportEngineGenerateMultiAllResourcesFail(t *testing.T) {
+	store := newTestMetricsStore(t)
+	store.Close()
+
+	engine := NewReportEngine(EngineConfig{MetricsStore: store})
+
+	_, _, err := engine.GenerateMulti(MultiReportRequest{
+		Resources: []MetricReportRequest{
+			{ResourceType: "node", ResourceID: "node-1"},
+			{ResourceType: "node", ResourceID: "node-2"},
+		},
+		Start:  time.Now().Add(-1 * time.Hour),
+		End:    time.Now(),
+		Format: FormatCSV,
+	})
+	if err == nil {
+		t.Fatal("expected error when all resources fail")
+	}
+	if !strings.Contains(err.Error(), "all resources failed") {
+		t.Fatalf("unexpected error: %v", err)
+	}
+}