Home | History | Annotate | Download | only in os 
      1 // Copyright 2010 The Go Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style
      3 // license that can be found in the LICENSE file.
      4 
      5 package os_test
      6 
      7 import (
      8 	. "os"
      9 	"reflect"
     10 	"strings"
     11 	"testing"
     12 )
     13 
     14 // testGetenv gives us a controlled set of variables for testing Expand.
     15 func testGetenv(s string) string {
     16 	switch s {
     17 	case "*":
     18 		return "all the args"
     19 	case "#":
     20 		return "NARGS"
     21 	case "$":
     22 		return "PID"
     23 	case "1":
     24 		return "ARGUMENT1"
     25 	case "HOME":
     26 		return "/usr/gopher"
     27 	case "H":
     28 		return "(Value of H)"
     29 	case "home_1":
     30 		return "/usr/foo"
     31 	case "_":
     32 		return "underscore"
     33 	}
     34 	return ""
     35 }
     36 
     37 var expandTests = []struct {
     38 	in, out string
     39 }{
     40 	{"", ""},
     41 	{"$*", "all the args"},
     42 	{"$$", "PID"},
     43 	{"${*}", "all the args"},
     44 	{"$1", "ARGUMENT1"},
     45 	{"${1}", "ARGUMENT1"},
     46 	{"now is the time", "now is the time"},
     47 	{"$HOME", "/usr/gopher"},
     48 	{"$home_1", "/usr/foo"},
     49 	{"${HOME}", "/usr/gopher"},
     50 	{"${H}OME", "(Value of H)OME"},
     51 	{"A$$$#$1$H$home_1*B", "APIDNARGSARGUMENT1(Value of H)/usr/foo*B"},
     52 }
     53 
     54 func TestExpand(t *testing.T) {
     55 	for _, test := range expandTests {
     56 		result := Expand(test.in, testGetenv)
     57 		if result != test.out {
     58 			t.Errorf("Expand(%q)=%q; expected %q", test.in, result, test.out)
     59 		}
     60 	}
     61 }
     62 
     63 func TestConsistentEnviron(t *testing.T) {
     64 	e0 := Environ()
     65 	for i := 0; i < 10; i++ {
     66 		e1 := Environ()
     67 		if !reflect.DeepEqual(e0, e1) {
     68 			t.Fatalf("environment changed")
     69 		}
     70 	}
     71 }
     72 
     73 func TestUnsetenv(t *testing.T) {
     74 	const testKey = "GO_TEST_UNSETENV"
     75 	set := func() bool {
     76 		prefix := testKey + "="
     77 		for _, key := range Environ() {
     78 			if strings.HasPrefix(key, prefix) {
     79 				return true
     80 			}
     81 		}
     82 		return false
     83 	}
     84 	if err := Setenv(testKey, "1"); err != nil {
     85 		t.Fatalf("Setenv: %v", err)
     86 	}
     87 	if !set() {
     88 		t.Error("Setenv didn't set TestUnsetenv")
     89 	}
     90 	if err := Unsetenv(testKey); err != nil {
     91 		t.Fatalf("Unsetenv: %v", err)
     92 	}
     93 	if set() {
     94 		t.Fatal("Unsetenv didn't clear TestUnsetenv")
     95 	}
     96 }
     97 
     98 func TestClearenv(t *testing.T) {
     99 	const testKey = "GO_TEST_CLEARENV"
    100 	const testValue = "1"
    101 
    102 	// reset env
    103 	defer func(origEnv []string) {
    104 		for _, pair := range origEnv {
    105 			// Environment variables on Windows can begin with =
    106 			// http://blogs.msdn.com/b/oldnewthing/archive/2010/05/06/10008132.aspx
    107 			i := strings.Index(pair[1:], "=") + 1
    108 			if err := Setenv(pair[:i], pair[i+1:]); err != nil {
    109 				t.Errorf("Setenv(%q, %q) failed during reset: %v", pair[:i], pair[i+1:], err)
    110 			}
    111 		}
    112 	}(Environ())
    113 
    114 	if err := Setenv(testKey, testValue); err != nil {
    115 		t.Fatalf("Setenv(%q, %q) failed: %v", testKey, testValue, err)
    116 	}
    117 	if _, ok := LookupEnv(testKey); !ok {
    118 		t.Errorf("Setenv(%q, %q) didn't set $%s", testKey, testValue, testKey)
    119 	}
    120 	Clearenv()
    121 	if val, ok := LookupEnv(testKey); ok {
    122 		t.Errorf("Clearenv() didn't clear $%s, remained with value %q", testKey, val)
    123 	}
    124 }
    125 
    126 func TestLookupEnv(t *testing.T) {
    127 	const smallpox = "SMALLPOX"      // No one has smallpox.
    128 	value, ok := LookupEnv(smallpox) // Should not exist.
    129 	if ok || value != "" {
    130 		t.Fatalf("%s=%q", smallpox, value)
    131 	}
    132 	defer Unsetenv(smallpox)
    133 	err := Setenv(smallpox, "virus")
    134 	if err != nil {
    135 		t.Fatalf("failed to release smallpox virus")
    136 	}
    137 	_, ok = LookupEnv(smallpox)
    138 	if !ok {
    139 		t.Errorf("smallpox release failed; world remains safe but LookupEnv is broken")
    140 	}
    141 }
    142