55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
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)
|
|
}
|
|
})
|
|
}
|
|
}
|