add mock validator

kubernetes-sigs · Dec 2, 2024 · a02b8df · a02b8df
1 parent 272b958
commit a02b8df
Show file tree

Hide file tree

Showing 39 changed files with 905 additions and 4 deletions.
diff --git a/pkg/azclient/accountclient/custom_test.go b/pkg/azclient/accountclient/custom_test.go
diff --git a/pkg/azclient/accountclient/mock.go b/pkg/azclient/accountclient/mock.go
diff --git a/pkg/azclient/accountclient/mock_accountclient/interface.go b/pkg/azclient/accountclient/mock_accountclient/interface.go
diff --git a/pkg/azclient/availabilitysetclient/mock.go b/pkg/azclient/availabilitysetclient/mock.go
diff --git a/pkg/azclient/blobcontainerclient/mock.go b/pkg/azclient/blobcontainerclient/mock.go
diff --git a/pkg/azclient/blobservicepropertiesclient/mock.go b/pkg/azclient/blobservicepropertiesclient/mock.go
diff --git a/pkg/azclient/client-gen/generator/mock.go b/pkg/azclient/client-gen/generator/mock.go
@@ -21,6 +21,7 @@ import (
 	"go/ast"
 	"os"
 	"os/exec"
+	"text/template"
 
 	"sigs.k8s.io/controller-tools/pkg/genall"
 	"sigs.k8s.io/controller-tools/pkg/loader"
@@ -62,7 +63,6 @@ func (generator MockGenerator) Generate(ctx *genall.GenerationContext) error {
 			if marker == nil {
 				return
 			}
-
 			mockFile, err := ctx.OutputRule.Open(root, "mock_"+root.Name+"/interface.go")
 			if err != nil {
 				root.AddError(err)
@@ -78,7 +78,24 @@ func (generator MockGenerator) Generate(ctx *genall.GenerationContext) error {
 				root.AddError(err)
 				return
 			}
-
+			mockVerifier, err := ctx.OutputRule.Open(root, "mock.go")
+			if err != nil {
+				root.AddError(err)
+				return
+			}
+			defer mockVerifier.Close()
+			var importList = make(map[string]map[string]struct{})
+			importList[root.ID+"/mock_"+root.Name] = make(map[string]struct{})
+			err = DumpHeaderToWriter(ctx, mockVerifier, generator.HeaderFile, importList, root.Name)
+			if err != nil {
+				root.AddError(err)
+				return
+			}
+			err = MockVerifyTemplate.Execute(mockVerifier, root.Name)
+			if err != nil {
+				root.AddError(err)
+				return
+			}
 		})
 		if err != nil {
 			break
@@ -109,3 +126,8 @@ func (MockGenerator) CheckFilter() loader.NodeFilter {
 		return isIface
 	}
 }
+
+var MockVerifyTemplate = template.Must(template.New("object-scaffolding-mock-verify").Parse(
+	`
+var _ Interface = &mock_{{.}}.MockInterface{}
+`))
diff --git a/pkg/azclient/deploymentclient/mock.go b/pkg/azclient/deploymentclient/mock.go
diff --git a/pkg/azclient/diskclient/mock.go b/pkg/azclient/diskclient/mock.go
diff --git a/pkg/azclient/fileservicepropertiesclient/mock.go b/pkg/azclient/fileservicepropertiesclient/mock.go
diff --git a/pkg/azclient/fileshareclient/mock.go b/pkg/azclient/fileshareclient/mock.go
diff --git a/pkg/azclient/identityclient/mock.go b/pkg/azclient/identityclient/mock.go
diff --git a/pkg/azclient/interfaceclient/mock.go b/pkg/azclient/interfaceclient/mock.go
diff --git a/pkg/azclient/ipgroupclient/mock.go b/pkg/azclient/ipgroupclient/mock.go
diff --git a/pkg/azclient/loadbalancerclient/mock.go b/pkg/azclient/loadbalancerclient/mock.go