102 lines
3.2 KiB
Go
102 lines
3.2 KiB
Go
package document_test
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"sigs.k8s.io/kustomize/v3/pkg/gvk"
|
|
"sigs.k8s.io/kustomize/v3/pkg/types"
|
|
|
|
"opendev.org/airship/airshipctl/testutil"
|
|
)
|
|
|
|
func TestNewBundle(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
bundle := testutil.NewTestBundle(t, "testdata")
|
|
require.NotNil(bundle)
|
|
}
|
|
|
|
func TestBundleDocumentFiltering(t *testing.T) {
|
|
assert := assert.New(t)
|
|
require := require.New(t)
|
|
|
|
bundle := testutil.NewTestBundle(t, "testdata")
|
|
|
|
t.Run("GetKustomizeResourceMap", func(t *testing.T) {
|
|
r := bundle.GetKustomizeResourceMap()
|
|
|
|
// ensure it is populated
|
|
assert.NotZero(r.Size())
|
|
})
|
|
|
|
t.Run("GetByGvk", func(t *testing.T) {
|
|
docs, err := bundle.GetByGvk("apps", "v1", "Deployment")
|
|
require.NoError(err)
|
|
|
|
assert.Len(docs, 3, "GetGvk returned the wrong number of resources")
|
|
})
|
|
|
|
t.Run("GetByAnnotation", func(t *testing.T) {
|
|
docs, err := bundle.GetByAnnotation("airshipit.org/clustertype=ephemeral")
|
|
require.NoError(err, "Error trying to GetByAnnotation")
|
|
|
|
assert.Len(docs, 4, "GetByAnnotation returned wrong number of resources")
|
|
})
|
|
|
|
t.Run("GetByLabel", func(t *testing.T) {
|
|
docs, err := bundle.GetByLabel("app=workflow-controller")
|
|
require.NoError(err, "Error trying to GetByLabel")
|
|
|
|
assert.Len(docs, 1, "GetByLabel returned wrong number of resources")
|
|
})
|
|
|
|
t.Run("SelectGvk", func(t *testing.T) {
|
|
// Select* tests test the Kustomize selector, which requires we build Kustomize
|
|
// selector objects which is useful for advanced searches that
|
|
// need to stack filters
|
|
g := gvk.Gvk{Group: "apps", Version: "v1", Kind: "Deployment"}
|
|
selector := types.Selector{Gvk: g}
|
|
docs, err := bundle.Select(selector)
|
|
require.NoError(err, "Error trying to select resources")
|
|
|
|
assert.Len(docs, 3, "SelectGvk returned wrong number of resources")
|
|
})
|
|
|
|
t.Run("SelectAnnotations", func(t *testing.T) {
|
|
// Find documents with a particular annotation, namely airshipit.org/clustertype=ephemeral
|
|
selector := types.Selector{AnnotationSelector: "airshipit.org/clustertype=ephemeral"}
|
|
docs, err := bundle.Select(selector)
|
|
require.NoError(err, "Error trying to select annotated resources")
|
|
|
|
assert.Len(docs, 4, "SelectAnnotations returned wrong number of resources")
|
|
})
|
|
|
|
t.Run("SelectLabels", func(t *testing.T) {
|
|
// Find documents with a particular label, namely app=workflow-controller
|
|
// note how this will only find resources labeled at the top most level (metadata.labels)
|
|
// and not spec templates with this label (spec.template.metadata.labels)
|
|
selector := types.Selector{LabelSelector: "app=workflow-controller"}
|
|
docs, err := bundle.Select(selector)
|
|
require.NoError(err, "Error trying to select labeled resources")
|
|
assert.Len(docs, 1, "SelectLabels returned wrong number of resources")
|
|
})
|
|
|
|
t.Run("Write", func(t *testing.T) {
|
|
// Ensure we can write out a bundle
|
|
//
|
|
// alanmeadows(TODO) improve validation what we write looks correct
|
|
var b bytes.Buffer
|
|
|
|
err := bundle.Write(&b)
|
|
require.NoError(err, "Failed to write bundle out")
|
|
|
|
// b.String() will contain all kustomize built YAML
|
|
// so we for now we just search for an expected string
|
|
// obviously, this should be improved
|
|
assert.Contains(b.String(), "workflow-controller")
|
|
})
|
|
}
|