Merge pull request #28 from anupcshan/master

Fix test compilation error
vektra · Jul 20, 2015 · a43d734 · a43d734
2 parents 71878bd + ee70206
commit a43d734
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/mockery.go → cmd/mockery/mockery.go b/mockery.go → cmd/mockery/mockery.go
diff --git a/doc.go b/doc.go
@@ -0,0 +1,4 @@
+package mockery
+
+// Empty file to make sure that importing "github.com/vektra/mockery" builds
+// successfully.
diff --git a/mockery/generator_test.go b/mockery/generator_test.go
@@ -145,7 +145,7 @@ func TestGeneratorPrologue(t *testing.T) {
 
 	gen := NewGenerator(iface)
 
-	gen.GeneratePrologue()
+	gen.GeneratePrologue("mocks")
 
 	expected := `package mocks
 
@@ -166,7 +166,7 @@ func TestGeneratorProloguewithImports(t *testing.T) {
 
 	gen := NewGenerator(iface)
 
-	gen.GeneratePrologue()
+	gen.GeneratePrologue("mocks")
 
 	expected := `package mocks