chore(): increase test coverage and update dependencies
Some checks are pending
ci/woodpecker/pr/pr Pipeline is pending
Some checks are pending
ci/woodpecker/pr/pr Pipeline is pending
This commit is contained in:
@@ -1,9 +1,12 @@
|
||||
package hetzner
|
||||
|
||||
import (
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"git.uploadfilter24.eu/covidnetes/woodpecker-autoscaler/internal/config"
|
||||
"github.com/hetznercloud/hcloud-go/hcloud"
|
||||
)
|
||||
|
||||
func TestGenerateUserData(t *testing.T) {
|
||||
@@ -54,3 +57,78 @@ runcmd:
|
||||
t.Errorf("got:\n%v\n, wanted:\n%v", got, wanted)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGenerateUserData_MultipleCases(t *testing.T) {
|
||||
base := config.Config{
|
||||
WoodpeckerGrpc: "grpc-test.woodpecker.test.tld:443",
|
||||
WoodpeckerLabelSelector: "uploadfilter24.eu/instance-role=WoodpeckerTest",
|
||||
WoodpeckerAgentVersion: "latest",
|
||||
}
|
||||
|
||||
cases := []struct {
|
||||
name string
|
||||
cfg config.Config
|
||||
agentName string
|
||||
agentToken string
|
||||
wantContains []string
|
||||
}{
|
||||
{
|
||||
name: "basic",
|
||||
cfg: base,
|
||||
agentName: "test-instance",
|
||||
agentToken: "Geheim1!",
|
||||
wantContains: []string{
|
||||
"image: woodpeckerci/woodpecker-agent:latest",
|
||||
"- WOODPECKER_AGENT_SECRET=Geheim1!",
|
||||
"- WOODPECKER_FILTER_LABELS=uploadfilter24.eu/instance-role=WoodpeckerTest",
|
||||
"- WOODPECKER_SERVER=grpc-test.woodpecker.test.tld:443",
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "empty token",
|
||||
cfg: base,
|
||||
agentName: "no-token",
|
||||
agentToken: "",
|
||||
wantContains: []string{
|
||||
"image: woodpeckerci/woodpecker-agent:latest",
|
||||
"- WOODPECKER_AGENT_SECRET=",
|
||||
"- WOODPECKER_HOSTNAME=no-token",
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tc := range cases {
|
||||
got, err := generateConfig(&tc.cfg, tc.agentName, tc.agentToken)
|
||||
if err != nil {
|
||||
t.Fatalf("%s: generateConfig returned error: %v", tc.name, err)
|
||||
}
|
||||
for _, want := range tc.wantContains {
|
||||
if !strings.Contains(got, want) {
|
||||
t.Errorf("%s: expected generated userdata to contain %q, got:\n%s", tc.name, want, got)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestCheckRuntime_MockedRefresh(t *testing.T) {
|
||||
// Mock refreshNodeInfo to return a server with a known Created time
|
||||
orig := refreshNodeInfo
|
||||
defer func() { refreshNodeInfo = orig }()
|
||||
|
||||
created := time.Date(2020, 1, 2, 3, 4, 5, 0, time.UTC)
|
||||
refreshNodeInfo = func(cfg *config.Config, serverID int) (*hcloud.Server, error) {
|
||||
return &hcloud.Server{Created: created}, nil
|
||||
}
|
||||
|
||||
cfg := config.Config{}
|
||||
// Capture minute before call to avoid flakiness across minute boundary
|
||||
minute := time.Now().Minute()
|
||||
got, err := CheckRuntime(&cfg, &hcloud.Server{ID: 123})
|
||||
if err != nil {
|
||||
t.Fatalf("CheckRuntime returned error: %v", err)
|
||||
}
|
||||
want := created.Add(time.Duration(minute))
|
||||
if !got.Equal(want) {
|
||||
t.Fatalf("unexpected runtime: got %v, want %v", got, want)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user