Carlos Florêncio
46 lines
1.4 KiB
Go
46 lines
1.4 KiB
Go
package backendapp
|
|
|
|
import (
|
|
"testing"
|
|
|
|
agentdocker "github.com/kandev/kandev/internal/agent/docker"
|
|
)
|
|
|
|
func TestScopedDockerLabelsAddsOnlyTheE2EScope(t *testing.T) {
|
|
t.Setenv("KANDEV_E2E_DOCKER_SCOPE", "e2e-test-scope")
|
|
input := map[string]string{"kandev.managed": "true"}
|
|
|
|
got := scopedDockerLabels(input)
|
|
|
|
if got["kandev.e2e.run"] != "e2e-test-scope" {
|
|
t.Fatalf("scope label = %q, want e2e-test-scope", got["kandev.e2e.run"])
|
|
}
|
|
if got["kandev.managed"] != "true" {
|
|
t.Fatalf("managed label = %q, want true", got["kandev.managed"])
|
|
}
|
|
if _, ok := input["kandev.e2e.run"]; ok {
|
|
t.Fatal("scoping must not mutate the caller's label map")
|
|
}
|
|
}
|
|
|
|
func TestScopeDockerUsageFiltersOnlyContainerRecords(t *testing.T) {
|
|
t.Setenv("KANDEV_E2E_DOCKER_SCOPE", "e2e-test-scope")
|
|
usage := agentdocker.DiskUsage{
|
|
ImageLayerBytes: 10,
|
|
Images: []agentdocker.ImageUsage{{ID: "image"}},
|
|
Containers: []agentdocker.ContainerUsage{
|
|
{ID: "owned", Labels: map[string]string{"kandev.e2e.run": "e2e-test-scope"}},
|
|
{ID: "other", Labels: map[string]string{"kandev.e2e.run": "other-scope"}},
|
|
},
|
|
}
|
|
|
|
scoped := scopeDockerUsage(usage)
|
|
|
|
if len(scoped.Containers) != 1 || scoped.Containers[0].ID != "owned" {
|
|
t.Fatalf("scoped containers = %#v, want only owned record", scoped.Containers)
|
|
}
|
|
if len(scoped.Images) != 1 || scoped.ImageLayerBytes != 10 {
|
|
t.Fatalf("image usage changed while scoping containers: %#v", scoped)
|
|
}
|
|
}
|