github.com/grahambrereton-form3/tilt@v0.10.18/internal/tiltfile/os/os_test.go (about)

     1  package os
     2  
     3  import (
     4  	"os"
     5  	"testing"
     6  
     7  	"github.com/stretchr/testify/assert"
     8  
     9  	"github.com/windmilleng/tilt/internal/tiltfile/starkit"
    10  )
    11  
    12  func TestEnviron(t *testing.T) {
    13  	f := NewFixture(t)
    14  	os.Setenv("FAKE_ENV_VARIABLE", "fakeValue")
    15  	defer os.Unsetenv("FAKE_ENV_VARIABLE")
    16  
    17  	f.File("Tiltfile", `
    18  print(os.environ['FAKE_ENV_VARIABLE'])
    19  `)
    20  
    21  	_, err := f.ExecFile("Tiltfile")
    22  	assert.NoError(t, err)
    23  	assert.Equal(t, "fakeValue\n", f.PrintOutput())
    24  }
    25  
    26  func NewFixture(tb testing.TB) *starkit.Fixture {
    27  	return starkit.NewFixture(tb, NewExtension())
    28  }