tmpl/registry/template_test.go

package registry

import (
	"io/ioutil"
	"os"
	"path/filepath"
	"testing"

	"github.com/matryer/is"
)

var (
	tmplContents = `{{title name}} (@{{username}}) {{if .bool}}{{.year}}{{end}} {{org}}`
	tmplTemplate = `
prompts:
  - id: name
    default: john olheiser
  - id: year
    default: ${TMPL_TEST} # 2020
  - id: package
    default: pkg
  - id: bool
    default: true
  - id: username
    default: username
  - id: org
    default: ${TMPL_PROMPT_USERNAME}/org
`
	tmplGold    = "John Olheiser (@jolheiser) 2020 jolheiser/org"
	tmplNewGold = "DO NOT OVERWRITE!"
)

func testExecute(t *testing.T) {
	assert := is.New(t)
	destDir := t.TempDir()

	// Set environment variable
	err := os.Setenv("TMPL_TEST", "2020")
	assert.NoErr(err) // Should set TMPL_TEST env

	err = os.Setenv("TMPL_VAR_USERNAME", "jolheiser")
	assert.NoErr(err) // Should set TMPL_VAR_USERNAME env

	// Get template
	tmpl, err := reg.GetTemplate("test")
	assert.NoErr(err) // Should get template

	// Execute template
	err = tmpl.Execute(destDir, true, true)
	assert.NoErr(err) // Should execute template

	// Check contents of file
	testPath := filepath.Join(destDir, "TEST")
	contents, err := ioutil.ReadFile(testPath)
	assert.NoErr(err)                        // Should be able to read TEST file
	assert.Equal(string(contents), tmplGold) // Template should match golden file

	// Check if directory was created
	pkgPath := filepath.Join(destDir, "PKG")
	_, err = os.Lstat(pkgPath)
	assert.NoErr(err) // PKG directory should exist

	// Check for .tmplkeep
	tmplKeep := filepath.Join(pkgPath, ".tmplkeep")
	_, err = os.Lstat(tmplKeep)
	assert.True(err != nil) // .tmplkeep file should NOT be retained

	// Change file to test non-overwrite
	err = ioutil.WriteFile(testPath, []byte(tmplNewGold), os.ModePerm)
	assert.NoErr(err) // Writing file should succeed

	err = tmpl.Execute(destDir, true, false)
	assert.NoErr(err) // Should execute template

	contents, err = os.ReadFile(testPath)
	assert.NoErr(err)                           // Should be able to read file
	assert.Equal(string(contents), tmplNewGold) // Template should match new golden file
}