added

cmd/protoc-gen-orm/main.go

@@ -11,37 +11,43 @@ import (
 )
 
 func main() {
-
-	reg := descriptor.NewRegistry()
-
 	req, err := generator.ParseRequest(os.Stdin)
 	if err != nil {
 		glog.Fatal(err)
 	}
 
+	out, err := generate(req)
+	if err != nil {
+		emitError(err)
+	}
+
+	emitFiles(out)
+}
+
+func generate(req *plugin.CodeGeneratorRequest) ([]*plugin.CodeGeneratorResponse_File, error) {
+	var err error
+	reg := descriptor.NewRegistry()
 	g := generator.New(reg)
 
 	if err = reg.Load(req); err != nil {
-		emitError(err)
-		return
+		return nil, err
 	}
 
 	var targets []*descriptor.File
 	for _, target := range req.FileToGenerate {
 		var f *descriptor.File
 		if f, err = reg.LookupFile(target); err != nil {
-			glog.Fatal(err)
+			return nil, err
 		}
 		targets = append(targets, f)
 	}
 
 	out, err := g.Generate(targets)
 	glog.V(1).Info("Processed code generator request")
 	if err != nil {
-		emitError(err)
-		return
+		return nil, err
 	}
-	emitFiles(out)
+	return out, nil
 }
 
 func emitFiles(out []*plugin.CodeGeneratorResponse_File) {

cmd/protoc-gen-orm/main_test.go

@@ -3,9 +3,26 @@ package main
 import (
 	"testing"
 
-	"github.com/stretchr/testify/assert"
+	"github.com/golang/protobuf/proto"
+	plugin "github.com/golang/protobuf/protoc-gen-go/plugin"
 )
 
-func TestMainNotNil(t *testing.T) {
-	assert.NotNil(t, main)
+func TestGenerate(t *testing.T) {
+	src := `
+		file_to_generate: 'a.proto'
+		proto_file <
+			name: 'a.proto'
+			message_type <
+				name: 'Example'
+			>
+		>
+	`
+	req := new(plugin.CodeGeneratorRequest)
+	if err := proto.UnmarshalText(src, req); err != nil {
+		t.Fatalf("proto.UnmarshalText(%s, &file) failed with %v; want success", src, err)
+	}
+
+	if _, err := generate(req); err != nil {
+		t.Errorf("failed to generate with error %v", err)
+	}
 }