feat(discovery): parse Compose-managed labels

2026-05-22 11:44:10 +02:00
parent cff6c1baff
commit 13ccb79ef5
2 changed files with 159 additions and 0 deletions
@@ -0,0 +1,88 @@
+package discovery_test
+
+import (
+	"testing"
+
+	"github.com/shcizo/package-updater/internal/discovery"
+	"github.com/stretchr/testify/require"
+)
+
+func TestParseComposeLabels_Success(t *testing.T) {
+	labels := map[string]string{
+		"com.docker.compose.project":                  "myapp-prod",
+		"com.docker.compose.service":                  "web",
+		"com.docker.compose.project.working_dir":      "/home/shcizo/self-hosted/myapp-prod",
+		"com.docker.compose.project.config_files":     "/home/shcizo/self-hosted/myapp-prod/docker-compose.yml",
+		"se.shcizo.auto-update":                       "true",
+	}
+	got, err := discovery.ParseComposeLabels(labels)
+	require.NoError(t, err)
+	require.Equal(t, "myapp-prod", got.Project)
+	require.Equal(t, "web", got.Service)
+	require.Equal(t, "/home/shcizo/self-hosted/myapp-prod", got.WorkingDir)
+	require.Equal(t, []string{"/home/shcizo/self-hosted/myapp-prod/docker-compose.yml"}, got.ConfigFiles)
+}
+
+func TestParseComposeLabels_MultipleConfigFiles(t *testing.T) {
+	labels := map[string]string{
+		"com.docker.compose.project":                  "myapp",
+		"com.docker.compose.service":                  "web",
+		"com.docker.compose.project.working_dir":      "/srv/myapp",
+		"com.docker.compose.project.config_files":     "/srv/myapp/docker-compose.yml,/srv/myapp/docker-compose.prod.yml",
+	}
+	got, err := discovery.ParseComposeLabels(labels)
+	require.NoError(t, err)
+	require.Equal(t, []string{
+		"/srv/myapp/docker-compose.yml",
+		"/srv/myapp/docker-compose.prod.yml",
+	}, got.ConfigFiles)
+}
+
+func TestParseComposeLabels_MissingFieldsError(t *testing.T) {
+	cases := []struct {
+		name    string
+		missing string
+		labels  map[string]string
+	}{
+		{"project", "com.docker.compose.project", map[string]string{
+			"com.docker.compose.service":              "web",
+			"com.docker.compose.project.working_dir":  "/x",
+			"com.docker.compose.project.config_files": "/x/y.yml",
+		}},
+		{"service", "com.docker.compose.service", map[string]string{
+			"com.docker.compose.project":              "p",
+			"com.docker.compose.project.working_dir":  "/x",
+			"com.docker.compose.project.config_files": "/x/y.yml",
+		}},
+		{"working_dir", "com.docker.compose.project.working_dir", map[string]string{
+			"com.docker.compose.project":              "p",
+			"com.docker.compose.service":              "web",
+			"com.docker.compose.project.config_files": "/x/y.yml",
+		}},
+		{"config_files", "com.docker.compose.project.config_files", map[string]string{
+			"com.docker.compose.project":             "p",
+			"com.docker.compose.service":             "web",
+			"com.docker.compose.project.working_dir": "/x",
+		}},
+	}
+	for _, c := range cases {
+		t.Run(c.name, func(t *testing.T) {
+			_, err := discovery.ParseComposeLabels(c.labels)
+			require.Error(t, err)
+			require.Contains(t, err.Error(), c.missing)
+		})
+	}
+}
+
+func TestHasOptIn(t *testing.T) {
+	require.True(t, discovery.HasOptIn(map[string]string{
+		"se.shcizo.auto-update": "true",
+	}, "se.shcizo.auto-update"))
+	require.False(t, discovery.HasOptIn(map[string]string{
+		"se.shcizo.auto-update": "false",
+	}, "se.shcizo.auto-update"))
+	require.False(t, discovery.HasOptIn(map[string]string{
+		"se.shcizo.auto-update": "TRUE",
+	}, "se.shcizo.auto-update"))
+	require.False(t, discovery.HasOptIn(map[string]string{}, "se.shcizo.auto-update"))
+}