package proxmox

import "testing"

func TestPveClusterStatusGetClusterModeNumeric(t *testing.T) {
	tests := []struct {
		name   string
		status PveClusterStatus
		want   float64
	}{
		{
			name:   "cluster mode",
			status: PveClusterStatus{Type: "cluster"},
			want:   1,
		},
		{
			name:   "standalone node",
			status: PveClusterStatus{},
			want:   0,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := tt.status.GetClusterModeNumeric(); got != tt.want {
				t.Fatalf("GetClusterModeNumeric() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestPveSubscriptionGetStatusNumeric(t *testing.T) {
	tests := []struct {
		status string
		want   float64
	}{
		{status: "notfound", want: 0},
		{status: "active", want: 1},
		{status: "expired", want: 2},
		{status: "invalid", want: 2},
		{status: "suspended", want: 2},
		{status: "new", want: 2},
		{status: "unknown", want: 2},
	}

	for _, tt := range tests {
		t.Run(tt.status, func(t *testing.T) {
			subscription := PveSubscription{Status: tt.status}
			if got := subscription.GetStatusNumeric(); got != tt.want {
				t.Fatalf("GetStatusNumeric() = %v, want %v", got, tt.want)
			}
		})
	}
}