Re-generate mock tests

- add go generate comments to files to autogenerate mock tests
opencontrol · Jun 13, 2018 · 6109840 · 6109840
1 parent 97dfc06
commit 6109840
Show file tree

Hide file tree

Showing 22 changed files with 106 additions and 91 deletions.
diff --git a/pkg/cli/get/resources/downloader.go b/pkg/cli/get/resources/downloader.go
@@ -5,6 +5,8 @@ import (
 	"github.com/opencontrol/compliance-masonry/tools/vcs"
 )
 
+//go:generate mockery -name Downloader
+
 // Downloader is a generic interface for how to download entries.
 type Downloader interface {
 	DownloadRepo(common.RemoteSource, string) error

diff --git a/pkg/cli/get/resources/getter.go b/pkg/cli/get/resources/getter.go
@@ -11,6 +11,8 @@ import (
 	"path/filepath"
 )
 
+//go:generate mockery -name Getter
+
 // getAllLocalResources will get try to get the resources that are in the current "source" directory and place them
 // in the final "destination" workspace directory.
 func getAllLocalResources(source string, destination string, opencontrol common.OpenControl, getter Getter) error {

diff --git a/pkg/cli/get/resources/mocks/Downloader.go b/pkg/cli/get/resources/mocks/Downloader.go
diff --git a/pkg/cli/get/resources/mocks/Getter.go b/pkg/cli/get/resources/mocks/Getter.go
diff --git a/pkg/lib/common/certification.go b/pkg/lib/common/certification.go
@@ -1,5 +1,7 @@
 package common
 
+//go:generate mockery -name Certification
+
 // Certification is the interface for getting all the attributes for a given certification.
 // Schema info: https://github.com/opencontrol/schemas#certifications
 //

diff --git a/pkg/lib/common/component.go b/pkg/lib/common/component.go
@@ -2,6 +2,10 @@ package common
 
 import "github.com/blang/semver"
 
+//go:generate mockery -name Component
+//go:generate mockery -name Satisfies
+//go:generate mockery -name Section
+
 // Component is the basic building block for all systems.
 // Schema info: https://github.com/opencontrol/schemas#component-yaml
 //

diff --git a/pkg/lib/common/control.go b/pkg/lib/common/control.go
@@ -1,5 +1,7 @@
 package common
 
+//go:generate mockery -name Control
+
 // Control is the interface for getting all the attributes for a given control.
 // Schema info: https://github.com/opencontrol/schemas#standards-documentation
 //

diff --git a/pkg/lib/common/mocks/Certification.go b/pkg/lib/common/mocks/Certification.go
diff --git a/pkg/lib/common/mocks/Component.go b/pkg/lib/common/mocks/Component.go
diff --git a/pkg/lib/common/mocks/Control.go b/pkg/lib/common/mocks/Control.go
diff --git a/pkg/lib/common/mocks/OpenControl.go b/pkg/lib/common/mocks/OpenControl.go
diff --git a/pkg/lib/common/mocks/RemoteSource.go b/pkg/lib/common/mocks/RemoteSource.go
diff --git a/pkg/lib/common/mocks/Satisfies.go b/pkg/lib/common/mocks/Satisfies.go
diff --git a/pkg/lib/common/mocks/Section.go b/pkg/lib/common/mocks/Section.go
diff --git a/pkg/lib/common/opencontrol.go b/pkg/lib/common/opencontrol.go
@@ -1,5 +1,8 @@
 package common
 
+//go:generate mockery -name OpenControl
+//go:generate mockery -name RemoteSource
+
 // OpenControl is an interface that every OpenControl yaml version should implement.
 // Schema info: https://github.com/opencontrol/schemas#opencontrolyaml
 //

diff --git a/pkg/lib/common/standard.go b/pkg/lib/common/standard.go
@@ -1,5 +1,7 @@
 package common
 
+//go:generate mockery -name Standard -testonly
+
 // Standard is the container of all the information for a particular Standard.
 // Schema info: https://github.com/opencontrol/schemas#standards-documentation
 //

diff --git a/pkg/lib/common/workspace.go b/pkg/lib/common/workspace.go
@@ -1,5 +1,7 @@
 package common
 
+//go:generate mockery -name Workspace -testonly
+
 // Workspace represents all the information such as components, standards, and certification as well as
 // the result information such as the justifications.
 type Workspace interface {

diff --git a/pkg/lib/opencontrol/opencontrol.go b/pkg/lib/opencontrol/opencontrol.go
@@ -4,6 +4,8 @@ import (
 	"github.com/opencontrol/compliance-masonry/pkg/lib/common"
 )
 
+//go:generate mockery -name SchemaParser -testonly
+
 // Base is the common struct that all schemas must have.
 type Base struct {
 	// SchemaVersion contains the schema version.

diff --git a/tools/fs/fs.go b/tools/fs/fs.go
@@ -9,6 +9,8 @@ import (
 	metaleapfs "github.com/metaleap/go-util/fs"
 )
 
+//go:generate mockery -name Util
+
 // Util is an interface for helper file system utilities.
 type Util interface {
 	OpenAndReadFile(file string) ([]byte, error)