github.com/hashicorp/go-getter/v2@v2.2.2/util_test.go (about)

     1  package getter
     2  
     3  import (
     4  	"os"
     5  	"testing"
     6  )
     7  
     8  // tempEnv sets the env var temporarily and returns a function that should
     9  // be deferred to clean it up.
    10  func tempEnv(t *testing.T, k, v string) func() {
    11  	old := os.Getenv(k)
    12  
    13  	// Set env
    14  	if err := os.Setenv(k, v); err != nil {
    15  		t.Fatalf("err: %s", err)
    16  	}
    17  
    18  	// Easy cleanup
    19  	return func() {
    20  		if err := os.Setenv(k, old); err != nil {
    21  			t.Fatalf("err: %s", err)
    22  		}
    23  	}
    24  }