Added cleanup script into test env setup

Author: domendobnikar

.github/workflows/test.yml

@@ -117,6 +117,7 @@ jobs:
           . ./internal/provider/tests/acceptance/setup/env.txt
           go run ./internal/provider/tests/acceptance/setup/acceptance_test_env_prepare.go
           go test -v -cover -coverpkg=github.com/hashicorp/terraform-provider-hypercore/internal/provider ./internal/provider/tests/acceptance/
+          go run ./internal/provider/tests/acceptance/setup/acceptance_test_env_prepare.go "cleanup"
         timeout-minutes: 10
   unit-test:
     name: Go Unit Tests

internal/provider/tests/acceptance/setup/acceptance_test_env_prepare.go

@@ -142,22 +142,53 @@ func DoesVirtualDiskExist(host string) bool {
 	return resp.StatusCode == http.StatusOK
 }
 
-func main() {
-	host := os.Getenv("HC_HOST")
-
-	if !AreEnvVariablesLoaded() {
-		log.Fatal("Environment variables aren't loaded, check env file in /acceptance/setup directory")
+func CleanupEnv(host string) {
+	client := SetHTTPClient()
+	data := []byte(fmt.Sprintf(`[{"virDomainUUID": "%s", "actionType": "STOP", "cause": "INTERNAL"}]`, source_vm_uuid))
+	req, err := http.NewRequest("POST", fmt.Sprintf("%s/rest/v1/VirDomain/action", host), bytes.NewBuffer(data))
+	if err != nil {
+		log.Fatal(err)
 	}
+	req = SetHTTPHeader(req)
 
-	if !DoesTestVMExist(host) {
-		log.Fatal("Acceptance test VM is missing in your testing environment")
+	// Execute the request
+	resp, err := client.Do(req)
+	if err != nil {
+		log.Fatal(err)
 	}
+	defer resp.Body.Close()
 
-	if IsTestVMRunning(host) {
-		log.Fatal("Acceptance test VM is RUNNING and should be turned off before the testing begins")
+	// Read and print the response
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		log.Fatal(err)
 	}
 
-	if !DoesVirtualDiskExist(host) {
-		log.Fatal("Acceptance test Virtual disk is missing in your testing environment")
+	fmt.Println("Response Status:", resp.Status)
+	fmt.Println("Response Body:", string(body))
+}
+
+func main() {
+	host := os.Getenv("HC_HOST")
+	isCleanup := len(os.Args) < 1 && os.Args[1] == "cleanup"
+
+	if isCleanup {
+		CleanupEnv(host)
+	} else {
+		if !AreEnvVariablesLoaded() {
+			log.Fatal("Environment variables aren't loaded, check env file in /acceptance/setup directory")
+		}
+
+		if !DoesTestVMExist(host) {
+			log.Fatal("Acceptance test VM is missing in your testing environment")
+		}
+
+		if IsTestVMRunning(host) {
+			log.Fatal("Acceptance test VM is RUNNING and should be turned off before the testing begins")
+		}
+
+		if !DoesVirtualDiskExist(host) {
+			log.Fatal("Acceptance test Virtual disk is missing in your testing environment")
+		}
 	}
 }