github.com/google/cloudprober@v0.11.3/logger/logger_test.go (about)

     1  package logger
     2  
     3  import (
     4  	"fmt"
     5  	"os"
     6  	"testing"
     7  )
     8  
     9  func TestEnvVarSet(t *testing.T) {
    10  	varName := "TEST_VAR"
    11  
    12  	testRows := []struct {
    13  		v        string
    14  		expected bool
    15  	}{
    16  		{"1", true},
    17  		{"yes", true},
    18  		{"not_set", false},
    19  		{"no", false},
    20  		{"false", false},
    21  	}
    22  
    23  	for _, row := range testRows {
    24  		t.Run(fmt.Sprintf("Val: %s, should be set: %v", row.v, row.expected), func(t *testing.T) {
    25  			os.Unsetenv(varName)
    26  			if row.v != "not_set" {
    27  				os.Setenv(varName, row.v)
    28  			}
    29  
    30  			got := envVarSet(varName)
    31  			if got != row.expected {
    32  				t.Errorf("Variable set: got=%v, expected=%v", got, row.expected)
    33  			}
    34  		})
    35  	}
    36  }