Add unit test for ProcessFile

Author: shihanng

generator/file_test.go

@@ -2,6 +2,7 @@ package generator
 
 import (
 	"bytes"
+	"io/ioutil"
 	"path/filepath"
 	"runtime"
 	"time"
@@ -124,6 +125,52 @@ var _ = Describe("File", func() {
 		)
 	})
 
+	Describe("ProcessFile", func() {
+		Context("when get a header", func() {
+			var f *descriptor.FileDescriptorProto
+
+			BeforeEach(func() {
+				f = &descriptor.FileDescriptorProto{
+					Options: &descriptor.FileOptions{},
+					Name:    sp("product.proto"),
+					Package: sp("pb"),
+					MessageType: []*descriptor.DescriptorProto{
+						{
+							Name: sp("Product"),
+							Field: []*descriptor.FieldDescriptorProto{
+								&descriptor.FieldDescriptorProto{
+									Name:     sp("ID"),
+									Number:   nil,
+									Label:    nil,
+									Type:     &typInt64,
+									TypeName: nil,
+									Options:  &descriptor.FieldOptions{},
+								},
+							},
+							Options: &descriptor.MessageOptions{},
+						},
+					},
+				}
+
+				err := proto.SetExtension(f.Options, options.E_GoModelsFilePath, sp("testdata/model.go"))
+				Expect(err).NotTo(HaveOccurred())
+
+				err = proto.SetExtension(f.MessageType[0].Options, options.E_GoStruct, sp("Product"))
+				Expect(err).NotTo(HaveOccurred())
+			})
+
+			It("returns generated code", func() {
+				expectedContent, err := ioutil.ReadFile("testdata/processfile.go.golden")
+				Expect(err).NotTo(HaveOccurred())
+
+				absPath, content, err := ProcessFile(f, sp("product"), sp("helper-package"), map[string]MessageOption{}, false, false)
+				Expect(err).NotTo(HaveOccurred())
+				Expect(content).To(Equal(string(expectedContent)))
+				Expect(absPath).To(Equal("product_transformer.go"))
+			})
+		})
+	})
+
 	Describe("modelPath", func() {
 
 		Context("when there is no option go_models_file_path in file", func() {

generator/testdata/model.go

@@ -0,0 +1,5 @@
+package model
+
+type Product struct {
+	ID int `db:"id" json:"id"`
+}

generator/testdata/processfile.go.golden

@@ -0,0 +1,142 @@
+// Code generated by protoc-gen-struct-transformer, version: v0.0.1. DO NOT EDIT.
+// source file: product.proto
+// source package: pb
+
+package product
+func PbToProductPtr(src *pb1.Product, opts ...TransformParam) *repo1.Product {
+	if src == nil {
+		return nil
+	}
+
+	d := PbToProduct(*src, opts...)
+	return &d
+}
+
+func PbToProductPtrList(src []*pb1.Product, opts ...TransformParam) []*repo1.Product {
+	resp := make([]*repo1.Product, len(src))
+
+	for i, s := range src {
+		resp[i] = PbToProductPtr(s, opts...)
+	}
+
+	return resp
+}
+
+func PbToProductPtrVal(src *pb1.Product, opts ...TransformParam) repo1.Product {
+	if src == nil {
+		return repo1.Product{}
+	}
+
+	return PbToProduct(*src, opts...)
+}
+
+func PbToProductPtrValList(src []*pb1.Product, opts ...TransformParam) []repo1.Product {
+	resp := make([]repo1.Product, len(src))
+
+	for i, s := range src {
+		resp[i] = PbToProduct(*s)
+		}
+
+	return resp
+}
+
+// PbToProductList is DEPRECATED. Use PbToProductPtrValList instead.
+func PbToProductList(src []*pb1.Product, opts ...TransformParam) []repo1.Product {
+	return PbToProductPtrValList(src)
+}
+
+func PbToProduct(src pb1.Product, opts ...TransformParam) repo1.Product {
+	s := repo1.Product{
+			ID:  int(src.Id ),
+	}
+
+	applyOptions(opts...)
+
+
+	return s
+}
+
+func PbToProductValPtr(src pb1.Product, opts ...TransformParam) *repo1.Product {
+	d := PbToProduct(src, opts...)
+	return &d
+}
+
+func PbToProductValList(src []pb1.Product, opts ...TransformParam) []repo1.Product {
+	resp := make([]repo1.Product, len(src))
+
+	for i, s := range src {
+		resp[i] = PbToProduct(s, opts...)
+	}
+
+	return resp
+}
+
+func ProductToPbPtr(src *repo1.Product, opts ...TransformParam) *pb1.Product {
+	if src == nil {
+		return nil
+	}
+
+	d := ProductToPb(*src, opts...)
+	return &d
+}
+
+func ProductToPbPtrList(src []*repo1.Product, opts ...TransformParam) []*pb1.Product {
+	resp := make([]*pb1.Product, len(src))
+
+	for i, s := range src {
+		resp[i] = ProductToPbPtr(s, opts...)
+	}
+
+	return resp
+}
+
+func ProductToPbPtrVal(src *repo1.Product, opts ...TransformParam) pb1.Product {
+	if src == nil {
+		return pb1.Product{}
+	}
+
+	return ProductToPb(*src, opts...)
+}
+
+func ProductToPbValPtrList(src []repo1.Product, opts ...TransformParam) []*pb1.Product {
+	resp := make([]*pb1.Product, len(src))
+
+	for i, s := range src {
+		g := ProductToPb(s, opts...)
+		resp[i] = &g
+		}
+
+	return resp
+}
+
+// ProductToPbList is DEPRECATED. Use ProductToPbValPtrList instead.
+func ProductToPbList(src []repo1.Product, opts ...TransformParam) []*pb1.Product {
+	return ProductToPbValPtrList(src)
+}
+
+func ProductToPb(src repo1.Product, opts ...TransformParam) pb1.Product {
+	s := pb1.Product{
+			Id:  int64(src.ID ),
+	}
+
+	applyOptions(opts...)
+
+
+	return s
+}
+
+func ProductToPbValPtr(src repo1.Product, opts ...TransformParam) *pb1.Product {
+	d := ProductToPb(src, opts...)
+	return &d
+}
+
+func ProductToPbValList(src []repo1.Product, opts ...TransformParam) []pb1.Product {
+	resp := make([]pb1.Product, len(src))
+
+	for i, s := range src {
+		resp[i] = ProductToPb(s, opts...)
+	}
+
+	return resp
+}
+