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

     1  package k8scontext
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/assert"
     7  
     8  	"github.com/windmilleng/tilt/internal/k8s"
     9  	"github.com/windmilleng/tilt/internal/tiltfile/starkit"
    10  )
    11  
    12  func TestAllowK8sContext(t *testing.T) {
    13  	f := NewFixture(t, "gke-blorg", k8s.EnvGKE)
    14  	f.File("Tiltfile", `
    15  allow_k8s_contexts('gke-blorg')
    16  `)
    17  	model, err := f.ExecFile("Tiltfile")
    18  	assert.NoError(t, err)
    19  	assert.Equal(t, []k8s.KubeContext{"gke-blorg"}, MustState(model).allowed)
    20  	assert.True(t, MustState(model).IsAllowed())
    21  
    22  	model, err = f.ExecFile("Tiltfile")
    23  	assert.NoError(t, err)
    24  	assert.Equal(t, []k8s.KubeContext{"gke-blorg"}, MustState(model).allowed)
    25  }
    26  
    27  func NewFixture(tb testing.TB, ctx k8s.KubeContext, env k8s.Env) *starkit.Fixture {
    28  	return starkit.NewFixture(tb, NewExtension(ctx, env))
    29  }