integration_test.go

package tests

import (
	"fmt"
	"go/build"
	"os"
	"os/exec"
	"path"
	"path/filepath"
	"strings"
	"testing"
)

func TestBootstrapReadme(t *testing.T) {
	defer cleanup("./readme/*")
	if err := goagen("./readme", "bootstrap", "-d", "github.com/shogo82148/goa-v1/_integration_tests/readme/design"); err != nil {
		t.Error(err.Error())
	}
	if err := gobuild("./readme"); err != nil {
		t.Error(err.Error())
	}
}

func TestDefaultMedia(t *testing.T) {
	defer cleanup("./media/*")
	if err := goagen("./media", "bootstrap", "-d", "github.com/shogo82148/goa-v1/_integration_tests/media/design"); err != nil {
		t.Error(err.Error())
	}
	if err := gobuild("./media"); err != nil {
		t.Error(err.Error())
	}
	b, err := os.ReadFile("./media/app/contexts.go")
	if err != nil {
		t.Fatal("failed to load contexts.go")
	}
	expected := `// CreateGreetingPayload is the Greeting create action payload.
type CreateGreetingPayload struct {
	// A required string field in the parent type.
	Message string ` + "`" + `form:"message" json:"message" yaml:"message" xml:"message"` + "`" + `
	// An optional boolean field in the parent type.
	ParentOptional *bool ` + "`" + `form:"parent_optional,omitempty" json:"parent_optional,omitempty" yaml:"parent_optional,omitempty" xml:"parent_optional,omitempty"` + "`" + `
}
`
	if !strings.Contains(string(b), expected) {
		t.Errorf("DefaultMedia attribute definitions reference failed. Generated context:\n%s", string(b))
	}
}

func TestCellar(t *testing.T) {
	defer cleanup("./goa-cellar/*")
	if err := goagen("./goa-cellar", "bootstrap", "-d", "github.com/shogo82148/goa-v1/_integration_tests/goa-cellar/design"); err != nil {
		t.Error(err.Error())
	}
	if err := gobuild("./goa-cellar"); err != nil {
		t.Error(err.Error())
	}
	if err := gobuild("./goa-cellar/tool/cellar-cli"); err != nil {
		t.Error(err.Error())
	}
}

func TestCustomFieldName(t *testing.T) {
	defer cleanup("./field/*")
	if err := goagen("./field", "bootstrap", "-d", "github.com/shogo82148/goa-v1/_integration_tests/field/design"); err != nil {
		t.Error(err.Error())
	}
	if err := gobuild("./field"); err != nil {
		t.Error(err.Error())
	}
	b, err := os.ReadFile("./field/app/user_types.go")
	if err != nil {
		t.Fatal("failed to load user_types.go")
	}

	expected := `// UploadPayload user type.
type UploadPayload struct {
	// A required file field in the parent type.
	FilePrimary *multipart.FileHeader ` + "`" + `form:"file1" json:"file1" yaml:"file1" xml:"file1"` + "`" + `
	// An optional file field in the parent type.
	FileSecondary *multipart.FileHeader ` + "`" + `form:"file2,omitempty" json:"file2,omitempty" yaml:"file2,omitempty" xml:"file2,omitempty"` + "`" + `
	// A required int field in the parent type.
	ID int ` + "`" + `form:"id" json:"id" yaml:"id" xml:"id"` + "`" + `
}
`
	if !strings.Contains(string(b), expected) {
		t.Errorf("UploadPayload attribute definitions reference failed. Generated user_types:\n%s", string(b))
	}

	b, err = os.ReadFile("./field/app/media_types.go")
	if err != nil {
		t.Fatal("failed to load media_types.go")
	}

	expected = `// Multimedia media type (default view)
//
// Identifier: application/vnd.multimedia+json; view=default
type Multimedia struct {
	// Media ID
	MediaID int ` + "`" + `form:"id" json:"id" yaml:"id" xml:"id"` + "`" + `
	// An optional string field in the Multimedia
	Note *string ` + "`" + `form:"optional_note,omitempty" json:"optional_note,omitempty" yaml:"optional_note,omitempty" xml:"optional_note,omitempty"` + "`" + `
	// Media URL
	MediaURL string ` + "`" + `form:"url" json:"url" yaml:"url" xml:"url"` + "`" + `
}
`
	if !strings.Contains(string(b), expected) {
		t.Errorf("Multimedia attribute definitions reference failed. Generated media_types:\n%s", string(b))
	}

	expected = `// multimedia list (default view)
//
// Identifier: application/vnd.multimedialist+json; view=default
type Multimedialist struct {
	// A required array field in the parent media type
	MediaList []*Multimedia ` + "`" + `form:"media" json:"media" yaml:"media" xml:"media"` + "`" + `
}
`
	if !strings.Contains(string(b), expected) {
		t.Errorf("Multimedialist attribute definitions reference failed. Generated media_types:\n%s", string(b))
	}
}

func TestIssue161(t *testing.T) {
	defer cleanup("./issue161/*")
	if err := goagen("./issue161", "bootstrap", "-d", "github.com/shogo82148/goa-v1/_integration_tests/issue161/design"); err != nil {
		t.Error(err.Error())
	}
	if err := gobuild("./issue161"); err != nil {
		t.Error(err.Error())
	}
}

func goagen(dir, command string, args ...string) error {
	pkg, err := build.Import("github.com/shogo82148/goa-v1/goagen", "", 0)
	if err != nil {
		return err
	}
	cmd := exec.Command("go", "run")
	for _, f := range pkg.GoFiles {
		cmd.Args = append(cmd.Args, path.Join(pkg.Dir, f))
	}
	cmd.Dir = dir
	cmd.Args = append(cmd.Args, command)
	cmd.Args = append(cmd.Args, args...)
	out, err := cmd.CombinedOutput()
	if err != nil {
		return fmt.Errorf("%s\n%s", err.Error(), out)
	}
	return nil
}

func gobuild(dir string) error {
	cmd := exec.Command("go", "build", "./...")
	cmd.Dir = dir
	out, err := cmd.CombinedOutput()
	if err != nil {
		return fmt.Errorf("%s\n%s", err.Error(), out)
	}
	return nil
}

func cleanup(dir string) {
	files, err := filepath.Glob(dir)
	if err != nil {
		return
	}
	for _, f := range files {
		if strings.HasSuffix(f, "design") {
			continue
		}
		os.RemoveAll(f)
	}
}