envvar_test.go

// Copyright 2015 The Vanadium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package ref

import (
	"os"
	"reflect"
	"testing"
)

// Set an environment variable and return a function to undo it.
// Typical usage:
//   defer setenv(t, "VAR", "VALUE")()
func setenv(t *testing.T, name, value string) func() {
	oldval := os.Getenv(name)
	if err := os.Setenv(name, value); err != nil {
		t.Fatalf("Failed to set %q to %q: %v", name, value, err)
		return func() {}
	}
	return func() {
		if err := os.Setenv(name, oldval); err != nil {
			t.Fatalf("Failed to restore %q to %q: %v", name, oldval, err)
		}
	}
}

func TestEnvNamespaceRoots(t *testing.T) {
	defer setenv(t, EnvNamespacePrefix, "NS1")()
	defer setenv(t, EnvNamespacePrefix+"_BLAH", "NS_BLAH")()

	wantm := map[string]string{
		"V23_NAMESPACE":      "NS1",
		"V23_NAMESPACE_BLAH": "NS_BLAH",
	}
	wantl := []string{"NS1", "NS_BLAH"}

	gotm, gotl := EnvNamespaceRoots()
	if !reflect.DeepEqual(wantm, gotm) {
		t.Errorf("Got %v want %v", gotm, wantm)
	}
	if !reflect.DeepEqual(wantl, gotl) {
		t.Errorf("Got %v want %v", gotl, wantl)
	}
}

func TestEnvClearCredentials(t *testing.T) {
	defer setenv(t, EnvCredentials, "FOO")()
	if got, want := os.Getenv(EnvCredentials), "FOO"; got != want {
		t.Errorf("Got %q, want %q", got, want)
	}
	EnvClearCredentials()
	if got := os.Getenv(EnvCredentials); got != "" {
		t.Errorf("Got %q, wanted empty string", got)
	}
}