docker-compose/pkg/compose/loader_test.go

/*
   Copyright 2020 Docker Compose CLI authors

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
*/

package compose

import (
	"context"
	"os"
	"path/filepath"
	"testing"

	"github.com/compose-spec/compose-go/v2/cli"
	"github.com/docker/compose/v5/pkg/api"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestLoadProject_Basic(t *testing.T) {
	// Create a temporary compose file
	tmpDir := t.TempDir()
	composeFile := filepath.Join(tmpDir, "compose.yaml")
	composeContent := `
name: test-project
services:
  web:
    image: nginx:latest
    ports:
      - "8080:80"
  db:
    image: postgres:latest
    environment:
      POSTGRES_PASSWORD: secret
`
	err := os.WriteFile(composeFile, []byte(composeContent), 0o644)
	require.NoError(t, err)

	// Create compose service
	service, err := NewComposeService(nil)
	require.NoError(t, err)

	// Load the project
	ctx := context.Background()
	project, err := service.LoadProject(ctx, api.ProjectLoadOptions{
		ConfigPaths: []string{composeFile},
	})

	// Assertions
	require.NoError(t, err)
	assert.NotNil(t, project)
	assert.Equal(t, "test-project", project.Name)
	assert.Len(t, project.Services, 2)
	assert.Contains(t, project.Services, "web")
	assert.Contains(t, project.Services, "db")

	// Check labels were applied
	webService := project.Services["web"]
	assert.Equal(t, "test-project", webService.CustomLabels[api.ProjectLabel])
	assert.Equal(t, "web", webService.CustomLabels[api.ServiceLabel])
}

func TestLoadProject_WithEnvironmentResolution(t *testing.T) {
	tmpDir := t.TempDir()
	composeFile := filepath.Join(tmpDir, "compose.yaml")
	composeContent := `
services:
  app:
    image: myapp:latest
    environment:
      - TEST_VAR=${TEST_VAR}
      - LITERAL_VAR=literal_value
`
	err := os.WriteFile(composeFile, []byte(composeContent), 0o644)
	require.NoError(t, err)

	// Set environment variable
	require.NoError(t, os.Setenv("TEST_VAR", "resolved_value"))
	t.Cleanup(func() {
		require.NoError(t, os.Unsetenv("TEST_VAR"))
	})

	service, err := NewComposeService(nil)
	require.NoError(t, err)

	ctx := context.Background()

	// Test with environment resolution (default)
	t.Run("WithResolution", func(t *testing.T) {
		project, err := service.LoadProject(ctx, api.ProjectLoadOptions{
			ConfigPaths: []string{composeFile},
		})
		require.NoError(t, err)

		appService := project.Services["app"]
		// Environment should be resolved
		assert.NotNil(t, appService.Environment["TEST_VAR"])
		assert.Equal(t, "resolved_value", *appService.Environment["TEST_VAR"])
		assert.NotNil(t, appService.Environment["LITERAL_VAR"])
		assert.Equal(t, "literal_value", *appService.Environment["LITERAL_VAR"])
	})

	// Test without environment resolution
	t.Run("WithoutResolution", func(t *testing.T) {
		project, err := service.LoadProject(ctx, api.ProjectLoadOptions{
			ConfigPaths:       []string{composeFile},
			ProjectOptionsFns: []cli.ProjectOptionsFn{cli.WithoutEnvironmentResolution},
		})
		require.NoError(t, err)

		appService := project.Services["app"]
		// Environment should NOT be resolved, keeping raw values
		// Note: This depends on compose-go behavior, which may still have some resolution
		assert.NotNil(t, appService.Environment)
	})
}

func TestLoadProject_ServiceSelection(t *testing.T) {
	tmpDir := t.TempDir()
	composeFile := filepath.Join(tmpDir, "compose.yaml")
	composeContent := `
services:
  web:
    image: nginx:latest
  db:
    image: postgres:latest
  cache:
    image: redis:latest
`
	err := os.WriteFile(composeFile, []byte(composeContent), 0o644)
	require.NoError(t, err)

	service, err := NewComposeService(nil)
	require.NoError(t, err)

	ctx := context.Background()

	// Load only specific services
	project, err := service.LoadProject(ctx, api.ProjectLoadOptions{
		ConfigPaths: []string{composeFile},
		Services:    []string{"web", "db"},
	})

	require.NoError(t, err)
	assert.Len(t, project.Services, 2)
	assert.Contains(t, project.Services, "web")
	assert.Contains(t, project.Services, "db")
	assert.NotContains(t, project.Services, "cache")
}

func TestLoadProject_WithProfiles(t *testing.T) {
	tmpDir := t.TempDir()
	composeFile := filepath.Join(tmpDir, "compose.yaml")
	composeContent := `
services:
  web:
    image: nginx:latest
  debug:
    image: busybox:latest
    profiles: ["debug"]
`
	err := os.WriteFile(composeFile, []byte(composeContent), 0o644)
	require.NoError(t, err)

	service, err := NewComposeService(nil)
	require.NoError(t, err)

	ctx := context.Background()

	// Without debug profile
	t.Run("WithoutProfile", func(t *testing.T) {
		project, err := service.LoadProject(ctx, api.ProjectLoadOptions{
			ConfigPaths: []string{composeFile},
		})
		require.NoError(t, err)
		assert.Len(t, project.Services, 1)
		assert.Contains(t, project.Services, "web")
	})

	// With debug profile
	t.Run("WithProfile", func(t *testing.T) {
		project, err := service.LoadProject(ctx, api.ProjectLoadOptions{
			ConfigPaths: []string{composeFile},
			Profiles:    []string{"debug"},
		})
		require.NoError(t, err)
		assert.Len(t, project.Services, 2)
		assert.Contains(t, project.Services, "web")
		assert.Contains(t, project.Services, "debug")
	})
}

func TestLoadProject_WithLoadListeners(t *testing.T) {
	tmpDir := t.TempDir()
	composeFile := filepath.Join(tmpDir, "compose.yaml")
	composeContent := `
services:
  web:
    image: nginx:latest
`
	err := os.WriteFile(composeFile, []byte(composeContent), 0o644)
	require.NoError(t, err)

	service, err := NewComposeService(nil)
	require.NoError(t, err)

	ctx := context.Background()

	// Track events received
	var events []string
	listener := func(event string, metadata map[string]any) {
		events = append(events, event)
	}

	project, err := service.LoadProject(ctx, api.ProjectLoadOptions{
		ConfigPaths:   []string{composeFile},
		LoadListeners: []api.LoadListener{listener},
	})

	require.NoError(t, err)
	assert.NotNil(t, project)

	// Listeners should have been called (exact events depend on compose-go implementation)
	// The slice itself is always initialized (non-nil), even if empty
	_ = events // events may or may not have entries depending on compose-go behavior
}

func TestLoadProject_ProjectNameInference(t *testing.T) {
	tmpDir := t.TempDir()
	composeFile := filepath.Join(tmpDir, "compose.yaml")
	composeContent := `
services:
  web:
    image: nginx:latest
`
	err := os.WriteFile(composeFile, []byte(composeContent), 0o644)
	require.NoError(t, err)

	service, err := NewComposeService(nil)
	require.NoError(t, err)

	ctx := context.Background()

	// Without explicit project name
	t.Run("InferredName", func(t *testing.T) {
		project, err := service.LoadProject(ctx, api.ProjectLoadOptions{
			ConfigPaths: []string{composeFile},
		})
		require.NoError(t, err)
		// Project name should be inferred from directory
		assert.NotEmpty(t, project.Name)
	})

	// With explicit project name
	t.Run("ExplicitName", func(t *testing.T) {
		project, err := service.LoadProject(ctx, api.ProjectLoadOptions{
			ConfigPaths: []string{composeFile},
			ProjectName: "my-custom-project",
		})
		require.NoError(t, err)
		assert.Equal(t, "my-custom-project", project.Name)
	})
}

func TestLoadProject_Compatibility(t *testing.T) {
	tmpDir := t.TempDir()
	composeFile := filepath.Join(tmpDir, "compose.yaml")
	composeContent := `
services:
  web:
    image: nginx:latest
`
	err := os.WriteFile(composeFile, []byte(composeContent), 0o644)
	require.NoError(t, err)

	service, err := NewComposeService(nil)
	require.NoError(t, err)

	ctx := context.Background()

	// With compatibility mode
	project, err := service.LoadProject(ctx, api.ProjectLoadOptions{
		ConfigPaths:   []string{composeFile},
		Compatibility: true,
	})

	require.NoError(t, err)
	assert.NotNil(t, project)
	// In compatibility mode, separator should be "_"
	assert.Equal(t, "_", api.Separator)

	// Reset separator
	api.Separator = "-"
}

func TestLoadProject_InvalidComposeFile(t *testing.T) {
	tmpDir := t.TempDir()
	composeFile := filepath.Join(tmpDir, "compose.yaml")
	composeContent := `
this is not valid yaml: [[[
`
	err := os.WriteFile(composeFile, []byte(composeContent), 0o644)
	require.NoError(t, err)

	service, err := NewComposeService(nil)
	require.NoError(t, err)

	ctx := context.Background()

	// Should return an error for invalid YAML
	project, err := service.LoadProject(ctx, api.ProjectLoadOptions{
		ConfigPaths: []string{composeFile},
	})

	require.Error(t, err)
	assert.Nil(t, project)
}

func TestLoadProject_MissingComposeFile(t *testing.T) {
	service, err := NewComposeService(nil)
	require.NoError(t, err)

	ctx := context.Background()

	// Should return an error for missing file
	project, err := service.LoadProject(ctx, api.ProjectLoadOptions{
		ConfigPaths: []string{"/nonexistent/compose.yaml"},
	})

	require.Error(t, err)
	assert.Nil(t, project)
}