Skip to content

Commit

Permalink
added test for error
Browse files Browse the repository at this point in the history
  • Loading branch information
deadcheat committed Jun 12, 2018
1 parent 5c0b606 commit 803e273
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 8 deletions.
11 changes: 3 additions & 8 deletions generator/presenter/file/presenter.go
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ func (p *Presenter) action(c *cli.Context) error {
if err != nil {
return err
}
t, err := template.New("asset").Funcs(
t, _ := template.New("asset").Funcs(
template.FuncMap{
"sha1": func(s string) string {
h := sha1.New()
Expand All @@ -54,20 +54,15 @@ func (p *Presenter) action(c *cli.Context) error {
},
},
).Parse(pt.AssetFileTemplate)
if err != nil {
return err
}

var b bytes.Buffer
if err = t.Execute(&b, &pt.Assets{
_ = t.Execute(&b, &pt.Assets{
PackageName: c.String("package"),
VarName: c.String("name"),
DirMap: e.DirMap,
FileMap: e.FileMap,
Paths: e.Paths,
}); err != nil {
return err
}
})

// gofmt
formatted, err := format.Source(b.Bytes())
Expand Down
25 changes: 25 additions & 0 deletions generator/presenter/file/presenter_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ import (
"flag"
"testing"

"github.com/deadcheat/goblet/generator"
"github.com/deadcheat/goblet/generator/mock"
"github.com/deadcheat/goblet/generator/values"
"github.com/golang/mock/gomock"
Expand Down Expand Up @@ -76,3 +77,27 @@ func TestActionFailLoadFiles(t *testing.T) {
t.Error("Mount should return errTest", err)
}
}

func TestActionFailGoformat(t *testing.T) {

// Prepare mock
c := gomock.NewController(t)
defer c.Finish()

m := mock.NewMockUseCase(c)
m.EXPECT().LoadFiles([]string{"config"}, nil).Return(&generator.Entity{FileMap: nil}, nil)

p := New(m)
a := &cli.App{}
p.Mount(a)
set := flag.NewFlagSet("test", flag.ContinueOnError)
if err := set.Parse([]string{"config"}); err != nil {
panic(err)
}

a.Flags = values.FlagDefs
ctx := cli.NewContext(a, set, nil)
if err := p.action(ctx); err == nil {
t.Error("Mount should return errTest", err)
}
}

0 comments on commit 803e273

Please sign in to comment.