59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
package model
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/shcizo/package-updater/internal/protocol"
|
|
)
|
|
|
|
func TestLocationKey(t *testing.T) {
|
|
cases := []struct {
|
|
name, agentKey, want string
|
|
}{
|
|
{"compose is per agent", "compose|slusk|web", "compose|7|slusk|web"},
|
|
{"container is per agent", "container|adhoc", "container|7|adhoc"},
|
|
{"swarm is per cluster", "swarm|gitea_gitea", "swarm|cluster-1|gitea_gitea"},
|
|
}
|
|
for _, c := range cases {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
assert.Equal(t, c.want, LocationKey(7, "cluster-1", c.agentKey))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDisplayName(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
svc Service
|
|
want string
|
|
}{
|
|
{"compose", Service{Kind: protocol.KindCompose, ComposeProject: "slusk", ComposeService: "web"}, "slusk/web"},
|
|
{"swarm", Service{Kind: protocol.KindSwarm, SwarmService: "gitea_gitea"}, "gitea_gitea"},
|
|
{"container", Service{Kind: protocol.KindContainer, ContainerName: "adhoc"}, "adhoc"},
|
|
}
|
|
for _, c := range cases {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
assert.Equal(t, c.want, c.svc.DisplayName())
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestJob_Active(t *testing.T) {
|
|
cases := []struct {
|
|
status JobStatus
|
|
want bool
|
|
}{
|
|
{JobQueued, true},
|
|
{JobRunning, true},
|
|
{JobSucceeded, false},
|
|
{JobFailed, false},
|
|
}
|
|
for _, c := range cases {
|
|
t.Run(string(c.status), func(t *testing.T) {
|
|
assert.Equal(t, c.want, Job{Status: c.status}.Active())
|
|
})
|
|
}
|
|
}
|