Omit LICENSE file when license is set to none

cobra/cmd/init_test.go

@@ -2,15 +2,18 @@ package cmd
 
 import (
 	"fmt"
+	"io/ioutil"
 	"os"
 	"path/filepath"
+	"reflect"
+	"sort"
+	"strings"
 	"testing"
 )
 
 func getProject() *Project {
-	wd, _ := os.Getwd()
 	return &Project{
-		AbsolutePath: fmt.Sprintf("%s/testproject", wd),
+		AbsolutePath: fmt.Sprintf("%s/testproject", mustTempDir()),
 		Legal:        getLicense(),
 		Copyright:    copyrightLine(),
 		AppName:      "testproject",
@@ -19,6 +22,14 @@ func getProject() *Project {
 	}
 }
 
+func mustTempDir() string {
+	dir, err := ioutil.TempDir("", "cobra_cli_test_")
+	if err != nil {
+		panic(err)
+	}
+	return dir
+}
+
 func TestGoldenInitCmd(t *testing.T) {
 	project := getProject()
 	defer os.RemoveAll(project.AbsolutePath)
@@ -37,3 +48,46 @@ func TestGoldenInitCmd(t *testing.T) {
 		}
 	}
 }
+
+func TestInitNoLicense(t *testing.T) {
+	project := getProject()
+	project.Legal = noLicense
+	defer os.RemoveAll(project.AbsolutePath)
+
+	err := project.Create()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	root := project.AbsolutePath
+
+	want := []string{"main.go", "cmd/root.go"}
+	var got []string
+	err = filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if info.IsDir() {
+			return nil
+		}
+		relpath, err := filepath.Rel(root, path)
+		if err != nil {
+			return err
+		}
+		got = append(got, relpath)
+		return nil
+	})
+	if err != nil {
+		t.Fatalf("walking path %q: %v", root, err)
+	}
+	sort.StringSlice(got).Sort()
+	sort.StringSlice(want).Sort()
+	if !reflect.DeepEqual(got, want) {
+		t.Fatalf(
+			"In %s, found:\n  %s\nwant:\n  %s",
+			root,
+			strings.Join(got, ", "),
+			strings.Join(want, ", "),
+		)
+	}
+}

cobra/cmd/licenses.go

@@ -36,9 +36,11 @@ type License struct {
 	Header          string   // License header for source files
 }
 
+var noLicense = License{"None", []string{"none", "false"}, "", ""}
+
 func init() {
 	// Allows a user to not use a license.
-	Licenses["none"] = License{"None", []string{"none", "false"}, "", ""}
+	Licenses["none"] = noLicense
 
 	initApache2()
 	initMit()

cobra/cmd/project.go

@@ -64,6 +64,9 @@ func (p *Project) Create() error {
 	}
 
 	// create license
+	if p.Legal.Name == noLicense.Name {
+		return nil
+	}
 	return p.createLicenseFile()
 }