mirror of
https://github.com/spf13/cobra
synced 2024-11-04 21:07:19 +00:00
9388e79fb4
Pretty major change in behavior, but with modules a change is needed. Now cobra can initialize and add from within any Go module. The experience is simplified and streamlined, but requires `go mod init` to happen first.
82 lines
1.7 KiB
Go
82 lines
1.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
func getProject() *Project {
|
|
wd, _ := os.Getwd()
|
|
return &Project{
|
|
AbsolutePath: fmt.Sprintf("%s/testproject", wd),
|
|
Legal: getLicense(),
|
|
Copyright: copyrightLine(),
|
|
AppName: "cmd",
|
|
PkgName: "github.com/spf13/cobra/cobra/cmd/cmd",
|
|
Viper: true,
|
|
}
|
|
}
|
|
|
|
func TestGoldenInitCmd(t *testing.T) {
|
|
|
|
dir, err := ioutil.TempDir("", "cobra-init")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
tests := []struct {
|
|
name string
|
|
args []string
|
|
pkgName string
|
|
expectErr bool
|
|
}{
|
|
{
|
|
name: "successfully creates a project based on module",
|
|
args: []string{"testproject"},
|
|
pkgName: "github.com/spf13/testproject",
|
|
expectErr: false,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
viper.Set("useViper", true)
|
|
viper.Set("license", "apache")
|
|
projectPath, err := initializeProject(tt.args)
|
|
defer func() {
|
|
if projectPath != "" {
|
|
os.RemoveAll(projectPath)
|
|
}
|
|
}()
|
|
|
|
if !tt.expectErr && err != nil {
|
|
t.Fatalf("did not expect an error, got %s", err)
|
|
}
|
|
if tt.expectErr {
|
|
if err == nil {
|
|
t.Fatal("expected an error but got none")
|
|
} else {
|
|
// got an expected error nothing more to do
|
|
return
|
|
}
|
|
}
|
|
|
|
expectedFiles := []string{"LICENSE", "main.go", "cmd/root.go"}
|
|
for _, f := range expectedFiles {
|
|
generatedFile := fmt.Sprintf("%s/%s", projectPath, f)
|
|
goldenFile := fmt.Sprintf("testdata/%s.golden", filepath.Base(f))
|
|
err := compareFiles(generatedFile, goldenFile)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|