From fc157a141f8fbad9c2f40479329e6bfd751936b4 Mon Sep 17 00:00:00 2001
From: Mat Ryer <thatmatryer@gmail.com>
Date: Mon, 31 Jul 2017 15:09:54 +0100
Subject: [PATCH] added go fmt

---
 pkg/moq/moq.go | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/pkg/moq/moq.go b/pkg/moq/moq.go
index 3844e67..fc417e3 100644
--- a/pkg/moq/moq.go
+++ b/pkg/moq/moq.go
@@ -1,9 +1,11 @@
 package moq
 
 import (
+	"bytes"
 	"errors"
 	"fmt"
 	"go/ast"
+	"go/format"
 	"go/parser"
 	"go/token"
 	"go/types"
@@ -109,10 +111,18 @@ func (m *Mocker) Mock(w io.Writer, name ...string) error {
 	for pkgToImport := range m.imports {
 		doc.Imports = append(doc.Imports, pkgToImport)
 	}
-	err := m.tmpl.Execute(w, doc)
+	var buf bytes.Buffer
+	err := m.tmpl.Execute(&buf, doc)
 	if err != nil {
 		return err
 	}
+	formatted, err := format.Source(buf.Bytes())
+	if err != nil {
+		return fmt.Errorf("go/format: %s", err)
+	}
+	if _, err := w.Write(formatted); err != nil {
+		return err
+	}
 	return nil
 }