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 }