From 338d8597a9c3670ef984ba82b5ee742d35f01226 Mon Sep 17 00:00:00 2001
From: haitham911 <haitham911eg@gmail.com>
Date: Sun, 5 Jan 2025 01:56:58 +0200
Subject: [PATCH] add integration test for terraform apply and terraform clean

---
 tests/cli_test.go | 97 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)

diff --git a/tests/cli_test.go b/tests/cli_test.go
index 255503cc8..18d50dfb8 100644
--- a/tests/cli_test.go
+++ b/tests/cli_test.go
@@ -248,3 +248,100 @@ func verifyFileContains(t *testing.T, filePatterns map[string][]string) bool {
 	}
 	return success
 }
+func TestCLITerraformClean(t *testing.T) {
+	// Capture the starting working directory
+	startingDir, err := os.Getwd()
+	if err != nil {
+		t.Fatalf("Failed to get the current working directory: %v", err)
+	}
+
+	// Initialize PathManager and update PATH
+	pathManager := NewPathManager()
+	pathManager.Prepend("../build", "..")
+	err = pathManager.Apply()
+	if err != nil {
+		t.Fatalf("Failed to apply updated PATH: %v", err)
+	}
+	fmt.Printf("Updated PATH: %s\n", pathManager.GetPath())
+	defer func() {
+		// Change back to the original working directory after the test
+		if err := os.Chdir(startingDir); err != nil {
+			t.Fatalf("Failed to change back to the starting directory: %v", err)
+		}
+	}()
+	workDir := "../examples/quick-start-simple"
+	err = os.Chdir(workDir)
+	if err != nil {
+		t.Fatalf("Failed to change directory to %q: %v", workDir, err)
+	}
+	binaryPath, err := exec.LookPath("atmos")
+	if err != nil {
+		t.Fatalf("Binary not found: %s. Current PATH: %s", "atmos", pathManager.GetPath())
+	}
+	cmd := exec.Command(binaryPath, "terraform", "apply", "station", "-s", "dev")
+	var stdout, stderr bytes.Buffer
+	cmd.Stdout = &stdout
+	cmd.Stderr = &stderr
+	// ATMOS_COMPONENTS_TERRAFORM_APPLY_AUTO_APPROVE
+	envVars := os.Environ()
+	envVars = append(envVars, "ATMOS_COMPONENTS_TERRAFORM_APPLY_AUTO_APPROVE=true")
+	cmd.Env = envVars
+
+	// run terraform apply station -s dev and terraform apply station -s prod
+	err = cmd.Run()
+	if err != nil {
+		t.Log(stdout.String())
+		t.Fatalf("Failed to run terraform apply station -s dev: %v", stderr.String())
+		return
+	}
+	cmd = exec.Command(binaryPath, "terraform", "apply", "station", "-s", "dev")
+	err = cmd.Run()
+	if err != nil {
+		t.Log(stdout.String())
+		t.Fatalf("Failed to run terraform apply station -s prod: %v", stderr.String())
+		return
+	}
+	// get command error sta
+	// check if the state files and directories for the component and stack are exist
+	stateFiles := []string{
+		"./components/terraform/weather/.terraform",
+		"./components/terraform/weather/terraform.tfstate.d",
+		"./components/terraform/weather/.terraform.lock.hcl",
+	}
+	for _, file := range stateFiles {
+		fileAbs, err := filepath.Abs(file)
+		if err != nil {
+			t.Fatalf("Failed to resolve absolute path for %q: %v", file, err)
+		}
+		if _, err := os.Stat(fileAbs); errors.Is(err, os.ErrNotExist) {
+			t.Errorf("Reason: Expected file exist: %q", fileAbs)
+			return
+		}
+	}
+
+	// run atmos terraform clean
+	cmd = exec.Command(binaryPath, "terraform", "clean", "--force")
+	err = cmd.Run()
+	if err != nil {
+		t.Log(stdout.String())
+		t.Fatalf("Failed to run atmos terraform clean: %v", stderr.String())
+	}
+	// check if the state files and directories for the component and stack are deleted
+	for _, file := range stateFiles {
+		fileAbs, err := filepath.Abs(file)
+		if err != nil {
+			t.Fatalf("Failed to resolve absolute path for %q: %v", file, err)
+		}
+		_, err = os.Stat(fileAbs)
+		if err != nil {
+			if errors.Is(err, os.ErrNotExist) {
+				continue
+			}
+			t.Errorf("Reason: error %q", fileAbs)
+			return
+
+		}
+		return
+	}
+
+}