Runner/act/container/parse_env_file_test.go

// Copyright 2026 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package container

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

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"golang.org/x/text/encoding"
	"golang.org/x/text/encoding/unicode"
)

func newTestHostEnv(t *testing.T) (*HostEnvironment, string) {
	t.Helper()
	e := &HostEnvironment{Path: t.TempDir()}
	return e, filepath.Join(e.Path, "envfile")
}

func TestParseEnvFileSingleLine(t *testing.T) {
	e, envPath := newTestHostEnv(t)
	require.NoError(t, os.WriteFile(envPath, []byte("FOO=bar\nBAZ=qux\n"), 0o600))

	env := map[string]string{}
	require.NoError(t, parseEnvFile(e, envPath, &env)(context.Background()))
	assert.Equal(t, "bar", env["FOO"])
	assert.Equal(t, "qux", env["BAZ"])
}

func TestParseEnvFileMultiLine(t *testing.T) {
	e, envPath := newTestHostEnv(t)
	content := "FOO<<EOF\nline1\nline2\nEOF\n"
	require.NoError(t, os.WriteFile(envPath, []byte(content), 0o600))

	env := map[string]string{}
	require.NoError(t, parseEnvFile(e, envPath, &env)(context.Background()))
	assert.Equal(t, "line1\nline2", env["FOO"])
}

func TestParseEnvFileLargeValueWithinLimit(t *testing.T) {
	e, envPath := newTestHostEnv(t)
	big := strings.Repeat("x", 2*1024*1024)
	content := "FOO<<EOF\n" + big + "\nEOF\n"
	require.NoError(t, os.WriteFile(envPath, []byte(content), 0o600))

	env := map[string]string{}
	require.NoError(t, parseEnvFile(e, envPath, &env)(context.Background()))
	assert.Equal(t, big, env["FOO"])
}

func TestParseEnvFileLineExceedsBufferReportsScannerError(t *testing.T) {
	e, envPath := newTestHostEnv(t)
	tooBig := strings.Repeat("x", 17*1024*1024) // over the 16 MiB cap
	content := "FOO<<EOF\n" + tooBig + "\nEOF\n"
	require.NoError(t, os.WriteFile(envPath, []byte(content), 0o600))

	env := map[string]string{}
	err := parseEnvFile(e, envPath, &env)(context.Background())
	require.ErrorIs(t, err, bufio.ErrTooLong)
	assert.Contains(t, err.Error(), "reading env file")
}

// Regression test: a blank line used to fail the job at "Complete Job", after
// every step had already been recorded as successful.
func TestParseEnvFileBlankLines(t *testing.T) {
	e, envPath := newTestHostEnv(t)
	require.NoError(t, os.WriteFile(envPath, []byte("\nFOO=bar\n\n   \nBAZ=qux\n\n"), 0o600))

	env := map[string]string{}
	require.NoError(t, parseEnvFile(e, envPath, &env)(context.Background()))
	assert.Equal(t, "bar", env["FOO"])
	assert.Equal(t, "qux", env["BAZ"])
}

// blank lines inside a heredoc value are content, not separators
func TestParseEnvFileMultiLineKeepsBlankLines(t *testing.T) {
	e, envPath := newTestHostEnv(t)
	require.NoError(t, os.WriteFile(envPath, []byte("FOO<<EOF\nline1\n\nline2\nEOF\n"), 0o600))

	env := map[string]string{}
	require.NoError(t, parseEnvFile(e, envPath, &env)(context.Background()))
	assert.Equal(t, "line1\n\nline2", env["FOO"])

	require.NoError(t, os.WriteFile(envPath, []byte("FOO<<EOF\n\nline2\nEOF\n"), 0o600))
	env = map[string]string{}
	require.NoError(t, parseEnvFile(e, envPath, &env)(context.Background()))
	assert.Equal(t, "\nline2", env["FOO"])
}

func TestParseEnvFileUTF8BOM(t *testing.T) {
	e, envPath := newTestHostEnv(t)
	content := append([]byte{0xEF, 0xBB, 0xBF}, []byte("FOO=bar\n")...)
	require.NoError(t, os.WriteFile(envPath, content, 0o600))

	env := map[string]string{}
	require.NoError(t, parseEnvFile(e, envPath, &env)(context.Background()))
	assert.Equal(t, "bar", env["FOO"])
}

// Windows host mode: PowerShell 5.1 redirection writes UTF-16, which used to be
// unrecognisable as KEY=VALUE, so the writes were silently ignored.
func TestParseEnvFileUTF16(t *testing.T) {
	tests := []struct {
		name    string
		encoder *encoding.Encoder
	}{
		{"little endian", unicode.UTF16(unicode.LittleEndian, unicode.UseBOM).NewEncoder()},
		{"big endian", unicode.UTF16(unicode.BigEndian, unicode.UseBOM).NewEncoder()},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			e, envPath := newTestHostEnv(t)
			content, err := tt.encoder.Bytes([]byte("FOO=bar\r\nMULTI<<EOF\r\nline1\r\nEOF\r\n"))
			require.NoError(t, err)
			require.NoError(t, os.WriteFile(envPath, content, 0o600))

			env := map[string]string{}
			require.NoError(t, parseEnvFile(e, envPath, &env)(context.Background()))
			assert.Equal(t, "bar", env["FOO"])
			assert.Equal(t, "line1", env["MULTI"])
		})
	}
}

func TestParseEnvFileMissingDelimiter(t *testing.T) {
	e, envPath := newTestHostEnv(t)
	require.NoError(t, os.WriteFile(envPath, []byte("FOO<<EOF\nline1\nline2\n"), 0o600))

	env := map[string]string{}
	err := parseEnvFile(e, envPath, &env)(context.Background())
	require.Error(t, err)
	assert.Contains(t, err.Error(), "delimiter")
}