github.com/brycereitano/goa@v0.0.0-20170315073847-8ffa6c85e265/goagen/codegen/import_spec_test.go

package codegen_test

import (
	. "github.com/goadesign/goa/design"
	. "github.com/goadesign/goa/design/apidsl"
	"github.com/goadesign/goa/dslengine"
	"github.com/goadesign/goa/goagen/codegen"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

var _ = Describe("AttributeImports", func() {
	Context("given an attribute definition with fields", func() {
		var att *AttributeDefinition
		var st string
		var object Object

		Context("of object", func() {

			It("produces the import slice", func() {
				var imports []*codegen.ImportSpec
				object = Object{
					"foo": &AttributeDefinition{Type: String},
					"bar": &AttributeDefinition{Type: Integer},
				}
				object["foo"].Metadata = dslengine.MetadataDefinition{
					"struct:field:type": []string{"json.RawMessage", "encoding/json"},
				}
				att = new(AttributeDefinition)
				att.Type = object
				imports = codegen.AttributeImports(att, imports, nil)

				i := []*codegen.ImportSpec{&codegen.ImportSpec{
					Path: "encoding/json",
				},
				}
				st = i[0].Path

				Ω(st).Should(Equal(imports[0].Path))
			})
		})

		Context("of recursive object", func() {

			It("produces the import slice", func() {
				var imports []*codegen.ImportSpec
				o := Object{
					"foo": &AttributeDefinition{Type: String},
				}
				o["foo"].Metadata = dslengine.MetadataDefinition{
					"struct:field:type": []string{"json.RawMessage", "encoding/json"},
				}
				child := &AttributeDefinition{Type: o}

				po := Object{"child": child}
				po["child"].Metadata = dslengine.MetadataDefinition{
					"struct:field:type": []string{"json.RawMessage", "encoding/json"},
				}
				parent := &AttributeDefinition{Type: po}

				o["parent"] = parent

				att = new(AttributeDefinition)
				att.Type = po
				imports = codegen.AttributeImports(att, imports, nil)

				i := []*codegen.ImportSpec{&codegen.ImportSpec{
					Path: "encoding/json",
				},
				}
				st = i[0].Path
				l := len(imports)

				Ω(st).Should(Equal(imports[0].Path))
				Ω(l).Should(Equal(1))
			})
		})

		Context("of hash", func() {

			It("produces the import slice", func() {
				var imports []*codegen.ImportSpec
				elemType := &AttributeDefinition{Type: Integer}
				elemType.Metadata = dslengine.MetadataDefinition{
					"struct:field:type": []string{"json.RawMessage", "encoding/json"},
				}
				keyType := &AttributeDefinition{Type: Integer}
				elemType.Metadata = dslengine.MetadataDefinition{
					"struct:field:type": []string{"json.RawMessage", "encoding/json"},
				}
				hash := &Hash{KeyType: keyType, ElemType: elemType}

				att = new(AttributeDefinition)
				att.Type = hash
				imports = codegen.AttributeImports(att, imports, nil)

				i := []*codegen.ImportSpec{&codegen.ImportSpec{
					Path: "encoding/json",
				},
				}
				st = i[0].Path
				l := len(imports)

				Ω(st).Should(Equal(imports[0].Path))
				Ω(l).Should(Equal(1))
			})
		})

		Context("of array", func() {
			It("produces the import slice", func() {
				var imports []*codegen.ImportSpec
				elemType := &AttributeDefinition{Type: Integer}
				elemType.Metadata = dslengine.MetadataDefinition{
					"struct:field:type": []string{"json.RawMessage", "encoding/json"},
				}
				array := &Array{ElemType: elemType}

				att = new(AttributeDefinition)
				att.Type = array
				imports = codegen.AttributeImports(att, imports, nil)

				i := []*codegen.ImportSpec{&codegen.ImportSpec{
					Path: "encoding/json",
				},
				}
				st = i[0].Path

				Ω(st).Should(Equal(imports[0].Path))
			})
		})

		Context("of UserTypeDefinition", func() {

			It("produces the import slice", func() {
				var imports []*codegen.ImportSpec
				object = Object{
					"bar": &AttributeDefinition{Type: String},
				}
				object["bar"].Metadata = dslengine.MetadataDefinition{
					"struct:field:type": []string{"json.RawMessage", "encoding/json"},
				}

				u := &UserTypeDefinition{
					AttributeDefinition: &AttributeDefinition{Type: object},
				}

				att = u.AttributeDefinition
				imports = codegen.AttributeImports(att, imports, nil)

				i := []*codegen.ImportSpec{&codegen.ImportSpec{
					Path: "encoding/json",
				},
				}
				st = i[0].Path

				Ω(st).Should(Equal(imports[0].Path))
			})
		})

		Context("of MediaTypeDefinition", func() {
			It("produces the import slice", func() {
				var imports []*codegen.ImportSpec
				elemType := &AttributeDefinition{Type: Integer}
				elemType.Metadata = dslengine.MetadataDefinition{
					"struct:field:type": []string{"json.RawMessage", "encoding/json"},
				}
				array := &Array{ElemType: elemType}
				u := &UserTypeDefinition{
					AttributeDefinition: &AttributeDefinition{Type: array},
				}
				m := &MediaTypeDefinition{
					UserTypeDefinition: u,
				}

				att = m.AttributeDefinition
				imports = codegen.AttributeImports(att, imports, nil)

				i := []*codegen.ImportSpec{&codegen.ImportSpec{
					Path: "encoding/json",
				},
				}
				st = i[0].Path

				Ω(st).Should(Equal(imports[0].Path))
			})
		})
	})
})