env_unix_test.go

Documentation: os

     1  // Copyright 2013 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  //go:build aix || darwin || dragonfly || freebsd || linux || netbsd || openbsd || solaris
     6  // +build aix darwin dragonfly freebsd linux netbsd openbsd solaris
     7  
     8  package os_test
     9  
    10  import (
    11  	"fmt"
    12  	. "os"
    13  	"testing"
    14  )
    15  
    16  var setenvEinvalTests = []struct {
    17  	k, v string
    18  }{
    19  	{"", ""},      // empty key
    20  	{"k=v", ""},   // '=' in key
    21  	{"\x00", ""},  // '\x00' in key
    22  	{"k", "\x00"}, // '\x00' in value
    23  }
    24  
    25  func TestSetenvUnixEinval(t *testing.T) {
    26  	for _, tt := range setenvEinvalTests {
    27  		err := Setenv(tt.k, tt.v)
    28  		if err == nil {
    29  			t.Errorf(`Setenv(%q, %q) == nil, want error`, tt.k, tt.v)
    30  		}
    31  	}
    32  }
    33  
    34  var shellSpecialVarTests = []struct {
    35  	k, v string
    36  }{
    37  	{"*", "asterisk"},
    38  	{"#", "pound"},
    39  	{"$", "dollar"},
    40  	{"@", "at"},
    41  	{"!", "exclamation mark"},
    42  	{"?", "question mark"},
    43  	{"-", "dash"},
    44  }
    45  
    46  func TestExpandEnvShellSpecialVar(t *testing.T) {
    47  	for _, tt := range shellSpecialVarTests {
    48  		Setenv(tt.k, tt.v)
    49  		defer Unsetenv(tt.k)
    50  
    51  		argRaw := fmt.Sprintf("$%s", tt.k)
    52  		argWithBrace := fmt.Sprintf("${%s}", tt.k)
    53  		if gotRaw, gotBrace := ExpandEnv(argRaw), ExpandEnv(argWithBrace); gotRaw != gotBrace {
    54  			t.Errorf("ExpandEnv(%q) = %q, ExpandEnv(%q) = %q; expect them to be equal", argRaw, gotRaw, argWithBrace, gotBrace)
    55  		}
    56  	}
    57  }
    58
View as plain text