mirror of
https://github.com/kubernetes-sigs/kustomize.git
synced 2026-05-17 18:25:26 +00:00
138 lines
3.5 KiB
Go
138 lines
3.5 KiB
Go
// Copyright 2019 The Kubernetes Authors.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package loader_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"sigs.k8s.io/kustomize/api/internal/loader"
|
|
. "sigs.k8s.io/kustomize/api/internal/plugins/loader"
|
|
"sigs.k8s.io/kustomize/api/provider"
|
|
"sigs.k8s.io/kustomize/api/resmap"
|
|
kusttest_test "sigs.k8s.io/kustomize/api/testutils/kusttest"
|
|
valtest_test "sigs.k8s.io/kustomize/api/testutils/valtest"
|
|
"sigs.k8s.io/kustomize/api/types"
|
|
"sigs.k8s.io/kustomize/kyaml/filesys"
|
|
)
|
|
|
|
const (
|
|
secretGenerator = `
|
|
apiVersion: builtin
|
|
kind: SecretGenerator
|
|
metadata:
|
|
name: secretGenerator
|
|
name: mySecret
|
|
behavior: merge
|
|
envFiles:
|
|
- a.env
|
|
- b.env
|
|
valueFiles:
|
|
- longsecret.txt
|
|
literals:
|
|
- FRUIT=apple
|
|
- VEGETABLE=carrot
|
|
`
|
|
someServiceGenerator = `
|
|
apiVersion: someteam.example.com/v1
|
|
kind: SomeServiceGenerator
|
|
metadata:
|
|
name: myServiceGenerator
|
|
service: my-service
|
|
port: "12345"
|
|
`
|
|
)
|
|
|
|
func TestLoader(t *testing.T) {
|
|
th := kusttest_test.MakeEnhancedHarness(t).
|
|
BuildGoPlugin("builtin", "", "SecretGenerator").
|
|
BuildGoPlugin("someteam.example.com", "v1", "SomeServiceGenerator")
|
|
defer th.Reset()
|
|
p := provider.NewDefaultDepProvider()
|
|
rmF := resmap.NewFactory(p.GetResourceFactory())
|
|
fsys := filesys.MakeFsInMemory()
|
|
fLdr, err := loader.NewLoader(
|
|
loader.RestrictionRootOnly,
|
|
filesys.Separator, fsys)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
generatorConfigs, err := rmF.NewResMapFromBytes([]byte(
|
|
someServiceGenerator + "---\n" + secretGenerator))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, behavior := range []types.BuiltinPluginLoadingOptions{
|
|
/* types.BploUseStaticallyLinked,
|
|
types.BploLoadFromFileSys */} {
|
|
c := types.EnabledPluginConfig(behavior)
|
|
pLdr := NewLoader(c, rmF, fsys)
|
|
if pLdr == nil {
|
|
t.Fatal("expect non-nil loader")
|
|
}
|
|
_, err = pLdr.LoadGenerators(
|
|
fLdr, valtest_test.MakeFakeValidator(), generatorConfigs)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestLoaderWithWorkingDir(t *testing.T) {
|
|
p := provider.NewDefaultDepProvider()
|
|
rmF := resmap.NewFactory(p.GetResourceFactory())
|
|
fsys := filesys.MakeFsInMemory()
|
|
c := types.EnabledPluginConfig(types.BploLoadFromFileSys)
|
|
pLdr := NewLoader(c, rmF, fsys)
|
|
npLdr := pLdr.LoaderWithWorkingDir("/tmp/dummy")
|
|
require.Equal(t,
|
|
"",
|
|
pLdr.Config().FnpLoadingOptions.WorkingDir,
|
|
"the plugin working dir should not change")
|
|
require.Equal(t,
|
|
"/tmp/dummy",
|
|
npLdr.Config().FnpLoadingOptions.WorkingDir,
|
|
"the plugin working dir is not updated")
|
|
}
|
|
|
|
func TestLoaderWithStorageMounts(t *testing.T) {
|
|
const storageMountTransformer = `
|
|
apiVersion: com.example.kustomize/v1
|
|
kind: Test
|
|
metadata:
|
|
name: test-transformer
|
|
annotations:
|
|
config.kubernetes.io/function: |
|
|
container:
|
|
image: test
|
|
mounts:
|
|
- type: bind
|
|
src: ../
|
|
dst: /mount
|
|
`
|
|
p := provider.NewDefaultDepProvider()
|
|
rmF := resmap.NewFactory(p.GetResourceFactory())
|
|
fsys := filesys.MakeFsInMemory()
|
|
fLdr, err := loader.NewLoader(
|
|
loader.RestrictionRootOnly,
|
|
filesys.Separator, fsys)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
configs, err := rmF.NewResMapFromBytes([]byte(storageMountTransformer))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
c := types.EnabledPluginConfig(types.BploLoadFromFileSys)
|
|
pLdr := NewLoader(c, rmF, fsys)
|
|
if pLdr == nil {
|
|
t.Fatal("expect non-nil loader")
|
|
}
|
|
_, err = pLdr.LoadTransformers(
|
|
fLdr, valtest_test.MakeFakeValidator(), configs)
|
|
if err == nil { // should fail because src specified is outside root
|
|
t.Fatal("the loader allowed a mount outside root")
|
|
}
|
|
}
|