61 lines
No EOL
1.1 KiB
Go
61 lines
No EOL
1.1 KiB
Go
package models
|
|
|
|
import "testing"
|
|
|
|
func TestHostStateStr(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
state int
|
|
want string
|
|
}{
|
|
{
|
|
name: "zero value returns UNKNOWN when no worker has reported",
|
|
state: 0,
|
|
want: "UNKNOWN",
|
|
},
|
|
{
|
|
name: "UNAVAIL flag",
|
|
state: HOSTSTAT["UNAVAIL"],
|
|
want: "UNAVAIL",
|
|
},
|
|
{
|
|
name: "CLOSED flag",
|
|
state: HOSTSTAT["CLOSED"],
|
|
want: "CLOSED",
|
|
},
|
|
{
|
|
name: "BUSY flag",
|
|
state: HOSTSTAT["BUSY"],
|
|
want: "BUSY",
|
|
},
|
|
{
|
|
name: "LOCKEDU flag",
|
|
state: HOSTSTAT["LOCKEDU"],
|
|
want: "LOCKU",
|
|
},
|
|
{
|
|
name: "LOCKEDW flag",
|
|
state: HOSTSTAT["LOCKEDW"],
|
|
want: "LOCKW",
|
|
},
|
|
{
|
|
name: "no flags set returns OK for real host status",
|
|
state: 0x00000001, // some non-zero, non-flag value
|
|
want: "OK",
|
|
},
|
|
{
|
|
name: "combined UNAVAIL+CLOSED flags",
|
|
state: HOSTSTAT["UNAVAIL"] | HOSTSTAT["CLOSED"],
|
|
want: "UNAVAIL", // UNAVAIL takes priority
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got := HostStateStr(tt.state)
|
|
if got != tt.want {
|
|
t.Errorf("HostStateStr(%d) = %q, want %q", tt.state, got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
} |