Fixes -inpkg when its a main package, all objects in the package were

prefixed with `main.`
db7 · Apr 14, 2016 · 03b2d81 · 03b2d81
1 parent 7af274c
commit 03b2d81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mockery/generator.go b/mockery/generator.go
@@ -149,7 +149,7 @@ func renderType(t types.Type) string {
 	switch t := t.(type) {
 	case *types.Named:
 		o := t.Obj()
-		if o.Pkg() == nil {
+		if o.Pkg() == nil || o.Pkg().Name() == "main" {
 			return o.Name()
 		} else {
 			return o.Pkg().Name() + "." + o.Name()