mirror of
https://github.com/kubernetes-sigs/kustomize.git
synced 2026-05-17 18:25:26 +00:00
367 lines
7.4 KiB
Go
367 lines
7.4 KiB
Go
// Copyright 2019 The Kubernetes Authors.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package framework_test
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"text/template"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"sigs.k8s.io/kustomize/kyaml/fn/framework"
|
|
"sigs.k8s.io/kustomize/kyaml/fn/framework/frameworktestutil"
|
|
"sigs.k8s.io/kustomize/kyaml/testutil"
|
|
"sigs.k8s.io/kustomize/kyaml/yaml"
|
|
)
|
|
|
|
func TestCommand_dockerfile(t *testing.T) {
|
|
d, err := ioutil.TempDir("", "kustomize")
|
|
if !assert.NoError(t, err) {
|
|
t.FailNow()
|
|
}
|
|
defer os.RemoveAll(d)
|
|
|
|
// create a function
|
|
|
|
resourceList := &framework.ResourceList{}
|
|
cmd := framework.Command(resourceList, func() error { return nil })
|
|
|
|
// generate the Dockerfile
|
|
cmd.SetArgs([]string{"gen", d})
|
|
if !assert.NoError(t, cmd.Execute()) {
|
|
t.FailNow()
|
|
}
|
|
|
|
b, err := ioutil.ReadFile(filepath.Join(d, "Dockerfile"))
|
|
if !assert.NoError(t, err) {
|
|
t.FailNow()
|
|
}
|
|
|
|
expected := `FROM golang:1.13-stretch
|
|
ENV CGO_ENABLED=0
|
|
WORKDIR /go/src/
|
|
COPY . .
|
|
RUN go build -v -o /usr/local/bin/function ./
|
|
|
|
FROM alpine:latest
|
|
COPY --from=0 /usr/local/bin/function /usr/local/bin/function
|
|
CMD ["function"]
|
|
`
|
|
if !assert.Equal(t, expected, string(b)) {
|
|
t.FailNow()
|
|
}
|
|
}
|
|
|
|
// TestCommand_standalone tests the framework works in standalone mode
|
|
func TestCommand_standalone(t *testing.T) {
|
|
// TODO: make this test pass on windows -- currently failure seems spurious
|
|
testutil.SkipWindows(t)
|
|
|
|
type api = struct {
|
|
A string `json:"a" yaml:"a"`
|
|
}
|
|
var config api
|
|
|
|
resourceList := &framework.ResourceList{FunctionConfig: &config}
|
|
cmdFn := func() *cobra.Command {
|
|
return framework.Command(resourceList, func() error {
|
|
resourceList.Items = append(resourceList.Items, yaml.MustParse(`
|
|
apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar1
|
|
namespace: default
|
|
annotations:
|
|
foo: bar1
|
|
`))
|
|
for i := range resourceList.Items {
|
|
err := resourceList.Items[i].PipeE(yaml.SetAnnotation("a", config.A))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|
|
|
|
frameworktestutil.ResultsChecker{Command: cmdFn}.Assert(t)
|
|
}
|
|
|
|
func TestCommand_standalonestdin(t *testing.T) {
|
|
// TODO: make this test pass on windows -- currently failure seems spurious
|
|
testutil.SkipWindows(t)
|
|
|
|
type api = struct {
|
|
A string `json:"a" yaml:"a"`
|
|
}
|
|
var config api
|
|
|
|
resourceList := &framework.ResourceList{FunctionConfig: &config}
|
|
cmd := framework.Command(resourceList, func() error {
|
|
resourceList.Items = append(resourceList.Items, yaml.MustParse(`
|
|
apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar2
|
|
namespace: default
|
|
annotations:
|
|
foo: bar2
|
|
`))
|
|
for i := range resourceList.Items {
|
|
err := resourceList.Items[i].PipeE(yaml.SetAnnotation("a", config.A))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
cmd.SetIn(bytes.NewBufferString(`
|
|
apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar1
|
|
namespace: default
|
|
annotations:
|
|
foo: bar1
|
|
spec:
|
|
replicas: 1
|
|
`))
|
|
var out bytes.Buffer
|
|
cmd.SetOut(&out)
|
|
cmd.SetArgs([]string{filepath.Join("testdata", "command", "config.yaml"), "-"})
|
|
|
|
require.NoError(t, cmd.Execute())
|
|
|
|
require.Equal(t, strings.TrimSpace(`
|
|
apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar1
|
|
namespace: default
|
|
annotations:
|
|
foo: bar1
|
|
a: 'b'
|
|
spec:
|
|
replicas: 1
|
|
---
|
|
apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar2
|
|
namespace: default
|
|
annotations:
|
|
foo: bar2
|
|
a: 'b'
|
|
`), strings.TrimSpace(out.String()))
|
|
}
|
|
|
|
func TestCommand_PatchTemplateFn(t *testing.T) {
|
|
// TODO: make this test pass on windows -- currently failure seems spurious
|
|
testutil.SkipWindows(t)
|
|
|
|
type api = struct {
|
|
Spec struct {
|
|
A string `json:"a" yaml:"a"`
|
|
} `json:"spec" yaml:"spec"`
|
|
}
|
|
var config api
|
|
|
|
cmd := framework.TemplateCommand{
|
|
API: &config,
|
|
PatchTemplatesFn: func(_ *framework.ResourceList) ([]framework.PatchTemplate, error) {
|
|
return []framework.PatchTemplate{{
|
|
Selector: &framework.Selector{Names: []string{config.Spec.A}},
|
|
Template: template.Must(template.New("test").Parse(`
|
|
metadata:
|
|
annotations:
|
|
baz: buz
|
|
`)),
|
|
}}, nil
|
|
},
|
|
}.GetCommand()
|
|
|
|
cmd.SetIn(bytes.NewBufferString(`
|
|
apiVersion: config.kubernetes.io/v1alpha1
|
|
kind: ResourceList
|
|
items:
|
|
- apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar1
|
|
namespace: default
|
|
annotations:
|
|
foo: bar1
|
|
spec:
|
|
replicas: 1
|
|
- apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar2
|
|
namespace: default
|
|
annotations:
|
|
foo: bar2
|
|
spec:
|
|
replicas: 1
|
|
functionConfig:
|
|
apiVersion: example.com/v1alpha1
|
|
kind: Example
|
|
spec:
|
|
a: "bar1"
|
|
`))
|
|
var out bytes.Buffer
|
|
cmd.SetOut(&out)
|
|
|
|
require.NoError(t, cmd.Execute())
|
|
|
|
require.Equal(t, strings.TrimSpace(`
|
|
apiVersion: config.kubernetes.io/v1alpha1
|
|
kind: ResourceList
|
|
items:
|
|
- apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar1
|
|
namespace: default
|
|
annotations:
|
|
foo: bar1
|
|
baz: buz
|
|
config.kubernetes.io/index: '0'
|
|
spec:
|
|
replicas: 1
|
|
- apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar2
|
|
namespace: default
|
|
annotations:
|
|
foo: bar2
|
|
spec:
|
|
replicas: 1
|
|
functionConfig:
|
|
apiVersion: example.com/v1alpha1
|
|
kind: Example
|
|
spec:
|
|
a: "bar1"
|
|
`), strings.TrimSpace(out.String()))
|
|
}
|
|
|
|
func TestCommand_PatchContainerTemplatesFn(t *testing.T) {
|
|
// TODO: make this test pass on windows -- currently failure seems spurious
|
|
testutil.SkipWindows(t)
|
|
|
|
type api = struct {
|
|
Spec struct {
|
|
A string `json:"a" yaml:"a"`
|
|
} `json:"spec" yaml:"spec"`
|
|
}
|
|
var config api
|
|
|
|
cmd := framework.TemplateCommand{
|
|
API: &config,
|
|
PatchContainerTemplatesFn: func(_ *framework.ResourceList) ([]framework.ContainerPatchTemplate, error) {
|
|
return []framework.ContainerPatchTemplate{{
|
|
PatchTemplate: framework.PatchTemplate{
|
|
Selector: &framework.Selector{Names: []string{config.Spec.A}},
|
|
Template: template.Must(template.New("test").Parse(`
|
|
env:
|
|
key: Foo
|
|
value: Bar
|
|
`))},
|
|
}}, nil
|
|
},
|
|
}.GetCommand()
|
|
|
|
cmd.SetIn(bytes.NewBufferString(`
|
|
apiVersion: config.kubernetes.io/v1alpha1
|
|
kind: ResourceList
|
|
items:
|
|
- apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar1
|
|
namespace: default
|
|
annotations:
|
|
foo: bar1
|
|
spec:
|
|
template:
|
|
spec:
|
|
containers:
|
|
- name: foo
|
|
- name: bar
|
|
- apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar2
|
|
namespace: default
|
|
annotations:
|
|
foo: bar2
|
|
spec:
|
|
template:
|
|
spec:
|
|
containers:
|
|
- name: foo
|
|
- name: bar
|
|
functionConfig:
|
|
apiVersion: example.com/v1alpha1
|
|
kind: Example
|
|
spec:
|
|
a: "bar1"
|
|
`))
|
|
var out bytes.Buffer
|
|
cmd.SetOut(&out)
|
|
|
|
require.NoError(t, cmd.Execute())
|
|
|
|
require.Equal(t, strings.TrimSpace(`
|
|
apiVersion: config.kubernetes.io/v1alpha1
|
|
kind: ResourceList
|
|
items:
|
|
- apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar1
|
|
namespace: default
|
|
annotations:
|
|
foo: bar1
|
|
spec:
|
|
template:
|
|
spec:
|
|
containers:
|
|
- name: foo
|
|
env:
|
|
key: Foo
|
|
value: Bar
|
|
- name: bar
|
|
env:
|
|
key: Foo
|
|
value: Bar
|
|
- apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: bar2
|
|
namespace: default
|
|
annotations:
|
|
foo: bar2
|
|
spec:
|
|
template:
|
|
spec:
|
|
containers:
|
|
- name: foo
|
|
- name: bar
|
|
functionConfig:
|
|
apiVersion: example.com/v1alpha1
|
|
kind: Example
|
|
spec:
|
|
a: "bar1"
|
|
`), strings.TrimSpace(out.String()))
|
|
}
|