github.com/ActiveState/cli@v0.0.0-20240508170324-6801f60cd051/scripts/constants-generator/main_test.go (about)

     1  package main
     2  
     3  import (
     4  	"os"
     5  	"path/filepath"
     6  	"testing"
     7  
     8  	"github.com/stretchr/testify/assert"
     9  	"github.com/stretchr/testify/require"
    10  )
    11  
    12  func init() {
    13  	inTest = true
    14  }
    15  
    16  func TestGenerate(t *testing.T) {
    17  	targetDir, err := os.MkdirTemp("", "constants-generator-test")
    18  	require.NoError(t, err)
    19  	target := filepath.Join(targetDir, "generated.go")
    20  	if _, err := os.Stat(target); err == nil {
    21  		err = os.Remove(target)
    22  		require.NoError(t, err, "Removed generated file")
    23  	}
    24  
    25  	run([]string{"", target})
    26  	assert.FileExists(t, target, "File is generated")
    27  
    28  	err = os.Remove(target)
    29  	require.NoError(t, err, "Removed generated file")
    30  
    31  	run([]string{"", "--", target})
    32  	assert.FileExists(t, target, "File is generated")
    33  }