Cut more ties to k8sdeps

Add tests for all the plugins.
This commit is contained in:
jregan
2019-05-22 11:22:31 -07:00
parent 7dc1eae40f
commit fe67bcdb8b
15 changed files with 295 additions and 51 deletions

View File

@@ -0,0 +1,54 @@
// Copyright 2019 The Kubernetes Authors.
// SPDX-License-Identifier: Apache-2.0
package main
import (
"sigs.k8s.io/kustomize/pkg/ifc"
"sigs.k8s.io/kustomize/pkg/resmap"
"sigs.k8s.io/kustomize/pkg/types"
"sigs.k8s.io/yaml"
)
// A secret generator example that gets data
// from a database (simulated by a hardcoded map).
type plugin struct {
rf *resmap.Factory
ldr ifc.Loader
Name string `json:"name,omitempty" yaml:"name,omitempty"`
Namespace string `json:"namespace,omitempty" yaml:"namespace,omitempty"`
// List of keys to use in database lookups
Keys []string `json:"keys,omitempty" yaml:"keys,omitempty"`
}
var KustomizePlugin plugin
var database = map[string]string{
"TREE": "oak",
"ROCKET": "SaturnV",
"FRUIT": "apple",
"VEGETABLE": "carrot",
"SIMPSON": "homer",
}
func (p *plugin) Config(
ldr ifc.Loader, rf *resmap.Factory, c []byte) error {
p.rf = rf
return yaml.Unmarshal(c, p)
}
// The plan here is to convert the plugin's input
// into the format used by the builtin secret generator plugin.
func (p *plugin) Generate() (resmap.ResMap, error) {
args := types.SecretArgs{}
args.Name = p.Name
args.Namespace = p.Namespace
for _, k := range p.Keys {
if v, ok := database[k]; ok {
args.LiteralSources = append(
args.LiteralSources, k+"="+v)
}
}
return p.rf.FromSecretArgs(p.ldr, nil, args)
}

View File

@@ -0,0 +1,44 @@
// Copyright 2019 The Kubernetes Authors.
// SPDX-License-Identifier: Apache-2.0
package main_test
import (
"testing"
"sigs.k8s.io/kustomize/pkg/kusttest"
"sigs.k8s.io/kustomize/plugin"
)
func TestSecretsFromDatabasePlugin(t *testing.T) {
tc := plugin.NewEnvForTest(t).Set()
defer tc.Reset()
tc.BuildGoPlugin(
"someteam.example.com", "v1", "SecretsFromDatabase")
th := kusttest_test.NewKustTestPluginHarness(t, "/app")
m := th.LoadAndRunGenerator(`
apiVersion: someteam.example.com/v1
kind: SecretsFromDatabase
metadata:
name: mySecretGenerator
name: forbiddenValues
namespace: production
keys:
- ROCKET
- VEGETABLE
`)
th.AssertActualEqualsExpected(m, `
apiVersion: v1
data:
ROCKET: U2F0dXJuVg==
VEGETABLE: Y2Fycm90
kind: Secret
metadata:
name: forbiddenValues
namespace: production
type: Opaque
`)
}