mirror of
https://github.com/FairwindsOps/polaris.git
synced 2026-05-13 20:56:59 +00:00
f753fc91f2
* able to run multi-resource tests * start passing resource provider through * working end-to-end * better support for go templating * fix tests * delint * add test * add json annotations * remove panics * fix annotation * fix for groupkinds * add comment * add docs * change jsonSchema field to schemaString * rename check * add pdb to tests * add ingress to tests * update deps * fix up policy import * update go * fix check name * funk it up * better docs
112 lines
4.3 KiB
Go
112 lines
4.3 KiB
Go
package kube
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io/ioutil"
|
|
"testing"
|
|
"time"
|
|
|
|
conf "github.com/fairwindsops/polaris/pkg/config"
|
|
"github.com/fairwindsops/polaris/test"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetResourcesFromPath(t *testing.T) {
|
|
provider, err := CreateResourceProviderFromPath("./test_files/test_1")
|
|
|
|
assert.Equal(t, nil, err, "Error should be nil")
|
|
|
|
assert.Equal(t, "Path", provider.SourceType, "Should have type Path")
|
|
assert.Equal(t, "./test_files/test_1", provider.SourceName, "Should have filename as name")
|
|
assert.Equal(t, "unknown", provider.ServerVersion, "Server version should be unknown")
|
|
assert.IsType(t, time.Now(), provider.CreationTime, "Creation time should be set")
|
|
|
|
assert.Equal(t, 0, len(provider.Nodes), "Should not have any nodes")
|
|
|
|
assert.Equal(t, 1, len(provider.Namespaces), "Should have a namespace")
|
|
assert.Equal(t, "two", provider.Namespaces[0].ObjectMeta.Name)
|
|
|
|
namespaceCount := map[string]int{}
|
|
for _, resources := range provider.Resources {
|
|
for _, controller := range resources {
|
|
namespaceCount[controller.ObjectMeta.GetNamespace()]++
|
|
}
|
|
}
|
|
assert.Equal(t, 11, provider.Resources.GetLength())
|
|
assert.Equal(t, 10, namespaceCount[""])
|
|
assert.Equal(t, 1, namespaceCount["two"])
|
|
}
|
|
|
|
func TestGetMultipleResourceFromSingleFile(t *testing.T) {
|
|
resources, err := CreateResourceProviderFromPath("./test_files/test_2/multi.yaml")
|
|
|
|
assert.Equal(t, nil, err, "Error should be nil")
|
|
|
|
assert.Equal(t, "Path", resources.SourceType, "Should have type Path")
|
|
assert.Equal(t, "./test_files/test_2/multi.yaml", resources.SourceName, "Should have filename as name")
|
|
assert.Equal(t, "unknown", resources.ServerVersion, "Server version should be unknown")
|
|
assert.IsType(t, time.Now(), resources.CreationTime, "Creation time should be set")
|
|
|
|
assert.Equal(t, 0, len(resources.Nodes), "Should not have any nodes")
|
|
|
|
assert.Equal(t, 1, len(resources.Resources["extensions/Deployment"]), "Should have one controller")
|
|
assert.Equal(t, "dashboard", resources.Resources["extensions/Deployment"][0].PodSpec.Containers[0].Name)
|
|
|
|
assert.Equal(t, 2, len(resources.Namespaces), "Should have a namespace")
|
|
assert.Equal(t, "polaris", resources.Namespaces[0].ObjectMeta.Name)
|
|
assert.Equal(t, "polaris-2", resources.Namespaces[1].ObjectMeta.Name)
|
|
}
|
|
|
|
func TestGetMultipleResourceFromBadFile(t *testing.T) {
|
|
_, err := CreateResourceProviderFromPath("./test_files/test_3")
|
|
assert.NotEqual(t, nil, err, "CreateResource From Path should fail with bad yaml")
|
|
}
|
|
|
|
func TestAddResourcesFromReader(t *testing.T) {
|
|
contents, err := ioutil.ReadFile("./test_files/test_2/multi.yaml")
|
|
assert.NoError(t, err)
|
|
reader := bytes.NewBuffer(contents)
|
|
resources := newResourceProvider("unknown", "Path", "-")
|
|
err = resources.addResourcesFromReader(reader)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, 0, len(resources.Nodes), "Should not have any nodes")
|
|
|
|
assert.Equal(t, 1, len(resources.Resources["extensions/Deployment"]), "Should have one controller")
|
|
assert.Equal(t, "dashboard", resources.Resources["extensions/Deployment"][0].PodSpec.Containers[0].Name)
|
|
|
|
assert.Equal(t, 2, len(resources.Namespaces), "Should have a namespace")
|
|
assert.Equal(t, "polaris", resources.Namespaces[0].ObjectMeta.Name)
|
|
assert.Equal(t, "polaris-2", resources.Namespaces[1].ObjectMeta.Name)
|
|
}
|
|
|
|
func TestGetResourceFromAPI(t *testing.T) {
|
|
k8s, dynamicInterface := test.SetupTestAPI(test.GetMockControllers("test")...)
|
|
resources, err := CreateResourceProviderFromAPI(context.Background(), k8s, "test", &dynamicInterface, conf.Configuration{})
|
|
assert.Equal(t, nil, err, "Error should be nil")
|
|
|
|
assert.Equal(t, "Cluster", resources.SourceType, "Should have type Path")
|
|
assert.Equal(t, "test", resources.SourceName, "Should have source name")
|
|
assert.IsType(t, time.Now(), resources.CreationTime, "Creation time should be set")
|
|
|
|
assert.Equal(t, 0, len(resources.Nodes), "Should not have any nodes")
|
|
assert.Equal(t, 5, len(resources.Resources), "Should have 5 controllers")
|
|
|
|
expectedNames := map[string]bool{
|
|
"deploy": false,
|
|
"job": false,
|
|
"cronjob": false,
|
|
"statefulset": false,
|
|
"daemonset": false,
|
|
}
|
|
for _, controllers := range resources.Resources {
|
|
for _, ctrl := range controllers {
|
|
expectedNames[ctrl.ObjectMeta.GetName()] = true
|
|
}
|
|
}
|
|
for name, val := range expectedNames {
|
|
assert.Equal(t, true, val, name)
|
|
}
|
|
}
|