main_hooks_and_store_integration_test.go

package main

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

	"github.com/alicebob/miniredis/v2"
)

func TestMainHooksAndStoreIntegration(t *testing.T) {
	// Run the miniredis server so we can store values across calls to main().
	s := miniredis.RunT(t)
	defer s.Close()

	redisUrl := fmt.Sprintf("redis://%s", s.Addr())
	t.Setenv("ATMOS_REDIS_URL", redisUrl)

	// Disable CI auto-detection so deploy hooks don't try to download
	// planfiles from GitHub Artifacts during tests.
	os.Unsetenv("GITHUB_ACTIONS")

	origDir, err := os.Getwd()
	if err != nil {
		t.Fatalf("failed to get current working directory: %v", err)
	}
	defer os.RemoveAll(filepath.Join(origDir, "tests", "fixtures", "scenarios", "hooks-test", ".terraform"))

	t.Chdir("tests/fixtures/scenarios/hooks-test")

	// This integration test calls run() (not main()) to avoid os.Exit() which panics in Go 1.25+.
	// run() returns an exit code instead of calling os.Exit().
	// We manipulate os.Args since run() uses cmd.Execute() which reads os.Args internally.
	origArgs := os.Args
	defer func() { os.Args = origArgs }()

	// Set the arguments for the first call to run() to deploy the `component1` component, which uses a `hook` to set a
	// value in Redis.
	os.Args = []string{"atmos", "terraform", "deploy", "component1", "-s", "test"}
	exitCode := run()
	if exitCode != 0 {
		t.Fatalf("first run() returned non-zero exit code: %d", exitCode)
	}

	// Set the arguments for the second call to run() to deploy the `component2` component, which uses a `store` to read a
	// value that was set in the first apply.
	os.Args = []string{"atmos", "terraform", "deploy", "component2", "-s", "test"}
	exitCode = run()
	if exitCode != 0 {
		t.Fatalf("second run() returned non-zero exit code: %d", exitCode)
	}
}