Merge pull request #1047 from monopole/pluginTests

Add test for builtin secretgenerator plugin.

pkg/kusttest/kusttestharness.go

@@ -111,6 +111,23 @@ func (th *KustTestHarness) WriteDefaultConfigs(fName string) {
 	}
 }
 
+func (th *KustTestHarness) LoadAndRunGenerator(
+	config string) resmap.ResMap {
+	res, err := th.rf.RF().FromBytes([]byte(config))
+	if err != nil {
+		th.t.Fatalf("Err: %v", err)
+	}
+	g, err := th.pl.LoadGenerator(th.ldr, res)
+	if err != nil {
+		th.t.Fatalf("Err: %v", err)
+	}
+	rm, err := g.Generate()
+	if err != nil {
+		th.t.Fatalf("Err: %v", err)
+	}
+	return rm
+}
+
 func tabToSpace(input string) string {
 	var result []string
 	for _, i := range input {

pkg/plugins/loader.go

@@ -48,36 +48,54 @@ func (l *Loader) LoadGenerators(
 	ldr ifc.Loader, rm resmap.ResMap) ([]transformers.Generator, error) {
 	var result []transformers.Generator
 	for _, res := range rm {
-		c, err := l.loadAndConfigurePlugin(ldr, res)
+		g, err := l.LoadGenerator(ldr, res)
 		if err != nil {
 			return nil, err
 		}
-		g, ok := c.(transformers.Generator)
-		if !ok {
-			return nil, fmt.Errorf("plugin %s not a generator", res.Id())
-		}
 		result = append(result, g)
 	}
 	return result, nil
 }
 
+func (l *Loader) LoadGenerator(
+	ldr ifc.Loader, res *resource.Resource) (transformers.Generator, error) {
+	c, err := l.loadAndConfigurePlugin(ldr, res)
+	if err != nil {
+		return nil, err
+	}
+	g, ok := c.(transformers.Generator)
+	if !ok {
+		return nil, fmt.Errorf("plugin %s not a generator", res.Id())
+	}
+	return g, nil
+}
+
 func (l *Loader) LoadTransformers(
 	ldr ifc.Loader, rm resmap.ResMap) ([]transformers.Transformer, error) {
 	var result []transformers.Transformer
 	for _, res := range rm {
-		c, err := l.loadAndConfigurePlugin(ldr, res)
+		t, err := l.LoadTransformer(ldr, res)
 		if err != nil {
 			return nil, err
 		}
-		t, ok := c.(transformers.Transformer)
-		if !ok {
-			return nil, fmt.Errorf("plugin %s not a transformer", res.Id())
-		}
 		result = append(result, t)
 	}
 	return result, nil
 }
 
+func (l *Loader) LoadTransformer(
+	ldr ifc.Loader, res *resource.Resource) (transformers.Transformer, error) {
+	c, err := l.loadAndConfigurePlugin(ldr, res)
+	if err != nil {
+		return nil, err
+	}
+	t, ok := c.(transformers.Transformer)
+	if !ok {
+		return nil, fmt.Errorf("plugin %s not a transformer", res.Id())
+	}
+	return t, nil
+}
+
 func pluginPath(id resid.ResId) string {
 	return filepath.Join(id.Gvk().Group, id.Gvk().Version, id.Gvk().Kind)
 }

pkg/resmap/legacygeneratorargs.go

@@ -0,0 +1,26 @@
+package resmap
+
+import (
+	"sigs.k8s.io/kustomize/pkg/ifc"
+	"sigs.k8s.io/kustomize/pkg/types"
+)
+
+func GeneratorArgsFromKunstruct(k ifc.Kunstructured) (
+	result types.GeneratorArgs, err error) {
+	result.Name = k.GetName()
+	// TODO: validate behavior values.
+	result.Behavior, err = k.GetFieldValue("behavior")
+	if err != nil {
+		return
+	}
+	result.EnvSources, err = k.GetStringSlice("envFiles")
+	if err != nil {
+		return
+	}
+	result.FileSources, err = k.GetStringSlice("valueFiles")
+	if err != nil {
+		return
+	}
+	result.LiteralSources, err = k.GetStringSlice("literals")
+	return
+}

pkg/target/generatorplugin_test.go

@@ -71,7 +71,7 @@ func writeSecretGeneratorConfig(th *kusttest_test.KustTestHarness, root string)
 apiVersion: builtin
 kind: SecretGenerator
 metadata:
-  name: secretGenerator
+  name: mySecret
 behavior: merge
 envFiles:
 - a.env
@@ -90,13 +90,10 @@ DB_PASSWORD=iloveyou
 `)
 	th.WriteF(filepath.Join(root, "longsecret.txt"), `
 Lorem ipsum dolor sit amet,
-consectetur adipiscing elit,
-sed do eiusmod tempor incididunt
-ut labore et dolore magna aliqua.
+consectetur adipiscing elit.
 `)
 }
 
-// nolint:lll
 func TestSecretGenerator(t *testing.T) {
 	tc := plugintest_test.NewPluginTestEnv(t).Set()
 	defer tc.Reset()
@@ -122,10 +119,10 @@ data:
   FRUIT: YXBwbGU=
   ROUTER_PASSWORD: YWRtaW4=
   VEGETABLE: Y2Fycm90
-  longsecret.txt: CkxvcmVtIGlwc3VtIGRvbG9yIHNpdCBhbWV0LApjb25zZWN0ZXR1ciBhZGlwaXNjaW5nIGVsaXQsCnNlZCBkbyBlaXVzbW9kIHRlbXBvciBpbmNpZGlkdW50CnV0IGxhYm9yZSBldCBkb2xvcmUgbWFnbmEgYWxpcXVhLgo=
+  longsecret.txt: CkxvcmVtIGlwc3VtIGRvbG9yIHNpdCBhbWV0LApjb25zZWN0ZXR1ciBhZGlwaXNjaW5nIGVsaXQuCg==
 kind: Secret
 metadata:
-  name: -ktm999dkcc
+  name: mySecret-g4g4kh4f7t
 type: Opaque
 `)
 }