mirror of https://github.com/k3s-io/k3s
365 lines
9.7 KiB
Go
365 lines
9.7 KiB
Go
/*
|
|
Copyright 2014 Google Inc. All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package config
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"os"
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/api/v1beta1"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/api/validation"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/kubelet"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/types"
|
|
)
|
|
|
|
func ExampleManifestAndPod(id string) (v1beta1.ContainerManifest, api.Pod) {
|
|
manifest := v1beta1.ContainerManifest{
|
|
ID: id,
|
|
UUID: types.UID(id),
|
|
Containers: []v1beta1.Container{
|
|
{
|
|
Name: "c" + id,
|
|
Image: "foo",
|
|
TerminationMessagePath: "/somepath",
|
|
},
|
|
},
|
|
Volumes: []v1beta1.Volume{
|
|
{
|
|
Name: "host-dir",
|
|
Source: v1beta1.VolumeSource{
|
|
HostDir: &v1beta1.HostPathVolumeSource{"/dir/path"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
expectedPod := api.Pod{
|
|
ObjectMeta: api.ObjectMeta{
|
|
Name: id,
|
|
UID: types.UID(id),
|
|
Namespace: kubelet.NamespaceDefault,
|
|
},
|
|
Spec: api.PodSpec{
|
|
Containers: []api.Container{
|
|
{
|
|
Name: "c" + id,
|
|
Image: "foo",
|
|
},
|
|
},
|
|
Volumes: []api.Volume{
|
|
{
|
|
Name: "host-dir",
|
|
VolumeSource: api.VolumeSource{
|
|
HostPath: &api.HostPathVolumeSource{"/dir/path"},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
return manifest, expectedPod
|
|
}
|
|
|
|
func TestExtractFromNonExistentFile(t *testing.T) {
|
|
ch := make(chan interface{}, 1)
|
|
c := sourceFile{"/some/fake/file", ch}
|
|
err := c.extractFromPath()
|
|
if err == nil {
|
|
t.Errorf("Expected error")
|
|
}
|
|
}
|
|
|
|
func TestUpdateOnNonExistentFile(t *testing.T) {
|
|
ch := make(chan interface{})
|
|
NewSourceFile("random_non_existent_path", time.Millisecond, ch)
|
|
select {
|
|
case got := <-ch:
|
|
update := got.(kubelet.PodUpdate)
|
|
expected := CreatePodUpdate(kubelet.SET, kubelet.FileSource)
|
|
if !api.Semantic.DeepDerivative(expected, update) {
|
|
t.Fatalf("Expected %#v, Got %#v", expected, update)
|
|
}
|
|
|
|
case <-time.After(time.Second):
|
|
t.Errorf("Expected update, timeout instead")
|
|
}
|
|
}
|
|
|
|
func writeTestFile(t *testing.T, dir, name string, contents string) *os.File {
|
|
file, err := ioutil.TempFile(os.TempDir(), "test_pod_config")
|
|
if err != nil {
|
|
t.Fatalf("Unable to create test file %#v", err)
|
|
}
|
|
file.Close()
|
|
if err := ioutil.WriteFile(file.Name(), []byte(contents), 0555); err != nil {
|
|
t.Fatalf("Unable to write test file %#v", err)
|
|
}
|
|
return file
|
|
}
|
|
|
|
func TestReadFromFile(t *testing.T) {
|
|
file := writeTestFile(t, os.TempDir(), "test_pod_config",
|
|
`{
|
|
"version": "v1beta1",
|
|
"uuid": "12345",
|
|
"id": "test",
|
|
"containers": [{ "image": "test/image", imagePullPolicy: "PullAlways"}]
|
|
}`)
|
|
defer os.Remove(file.Name())
|
|
|
|
ch := make(chan interface{})
|
|
NewSourceFile(file.Name(), time.Millisecond, ch)
|
|
select {
|
|
case got := <-ch:
|
|
update := got.(kubelet.PodUpdate)
|
|
expected := CreatePodUpdate(kubelet.SET, kubelet.FileSource, api.Pod{
|
|
ObjectMeta: api.ObjectMeta{
|
|
Name: "",
|
|
UID: "12345",
|
|
Namespace: kubelet.NamespaceDefault,
|
|
SelfLink: "",
|
|
},
|
|
Spec: api.PodSpec{Containers: []api.Container{{Image: "test/image"}}},
|
|
})
|
|
|
|
if !strings.HasPrefix(update.Pods[0].Name, "test-") {
|
|
t.Errorf("Unexpected name: %s", update.Pods[0].Name)
|
|
}
|
|
// There's no way to provide namespace in ContainerManifest, so
|
|
// it will be defaulted.
|
|
if update.Pods[0].Namespace != kubelet.NamespaceDefault {
|
|
t.Errorf("Unexpected namespace: %s", update.Pods[0].Namespace)
|
|
}
|
|
// SelfLink depends on namespace.
|
|
if !strings.HasPrefix(update.Pods[0].SelfLink, "/api/") {
|
|
t.Errorf("Unexpected selflink: %s", update.Pods[0].SelfLink)
|
|
}
|
|
|
|
// Reset the fileds that we don't want to compare.
|
|
update.Pods[0].Name = ""
|
|
update.Pods[0].SelfLink = ""
|
|
if !api.Semantic.DeepDerivative(expected, update) {
|
|
t.Fatalf("Expected %#v, Got %#v", expected, update)
|
|
}
|
|
|
|
case <-time.After(time.Second):
|
|
t.Errorf("Expected update, timeout instead")
|
|
}
|
|
}
|
|
|
|
func TestReadFromFileWithoutID(t *testing.T) {
|
|
file := writeTestFile(t, os.TempDir(), "test_pod_config",
|
|
`{
|
|
"version": "v1beta1",
|
|
"uuid": "12345",
|
|
"containers": [{ "image": "test/image", imagePullPolicy: "PullAlways"}]
|
|
}`)
|
|
defer os.Remove(file.Name())
|
|
|
|
ch := make(chan interface{})
|
|
NewSourceFile(file.Name(), time.Millisecond, ch)
|
|
select {
|
|
case got := <-ch:
|
|
update := got.(kubelet.PodUpdate)
|
|
expected := CreatePodUpdate(kubelet.SET, kubelet.FileSource, api.Pod{
|
|
ObjectMeta: api.ObjectMeta{
|
|
Name: "",
|
|
UID: "12345",
|
|
Namespace: kubelet.NamespaceDefault,
|
|
SelfLink: "",
|
|
},
|
|
Spec: api.PodSpec{Containers: []api.Container{{Image: "test/image"}}},
|
|
})
|
|
|
|
if len(update.Pods[0].ObjectMeta.Name) == 0 {
|
|
t.Errorf("Name did not get defaulted")
|
|
}
|
|
// Reset the fileds that we don't want to compare.
|
|
update.Pods[0].Name = ""
|
|
update.Pods[0].SelfLink = ""
|
|
if !api.Semantic.DeepDerivative(expected, update) {
|
|
t.Fatalf("Expected %#v, Got %#v", expected, update)
|
|
}
|
|
|
|
case <-time.After(time.Second):
|
|
t.Errorf("Expected update, timeout instead")
|
|
}
|
|
}
|
|
|
|
func TestReadV1Beta2FromFile(t *testing.T) {
|
|
file := writeTestFile(t, os.TempDir(), "test_pod_config",
|
|
`{
|
|
"version": "v1beta2",
|
|
"uuid": "12345",
|
|
"id": "test",
|
|
"containers": [{ "image": "test/image", imagePullPolicy: "PullAlways"}]
|
|
}`)
|
|
defer os.Remove(file.Name())
|
|
|
|
ch := make(chan interface{})
|
|
NewSourceFile(file.Name(), time.Millisecond, ch)
|
|
select {
|
|
case got := <-ch:
|
|
update := got.(kubelet.PodUpdate)
|
|
expected := CreatePodUpdate(kubelet.SET, kubelet.FileSource, api.Pod{
|
|
ObjectMeta: api.ObjectMeta{
|
|
Name: "",
|
|
UID: "12345",
|
|
Namespace: kubelet.NamespaceDefault,
|
|
SelfLink: "",
|
|
},
|
|
Spec: api.PodSpec{Containers: []api.Container{{Image: "test/image"}}},
|
|
})
|
|
|
|
// Reset the fileds that we don't want to compare.
|
|
update.Pods[0].Name = ""
|
|
update.Pods[0].SelfLink = ""
|
|
if !api.Semantic.DeepDerivative(expected, update) {
|
|
t.Fatalf("Expected %#v, Got %#v", expected, update)
|
|
}
|
|
|
|
case <-time.After(time.Second):
|
|
t.Errorf("Expected update, timeout instead")
|
|
}
|
|
}
|
|
|
|
func TestReadFromFileWithDefaults(t *testing.T) {
|
|
file := writeTestFile(t, os.TempDir(), "test_pod_config",
|
|
`{
|
|
"version": "v1beta1",
|
|
"id": "test",
|
|
"containers": [{ "image": "test/image" }]
|
|
}`)
|
|
defer os.Remove(file.Name())
|
|
|
|
ch := make(chan interface{})
|
|
NewSourceFile(file.Name(), time.Millisecond, ch)
|
|
select {
|
|
case got := <-ch:
|
|
update := got.(kubelet.PodUpdate)
|
|
if update.Pods[0].UID == "" {
|
|
t.Errorf("Unexpected UID: %s", update.Pods[0].UID)
|
|
}
|
|
|
|
case <-time.After(time.Second):
|
|
t.Errorf("Expected update, timeout instead")
|
|
}
|
|
}
|
|
|
|
func TestExtractFromBadDataFile(t *testing.T) {
|
|
file := writeTestFile(t, os.TempDir(), "test_pod_config", string([]byte{1, 2, 3}))
|
|
defer os.Remove(file.Name())
|
|
|
|
ch := make(chan interface{}, 1)
|
|
c := sourceFile{file.Name(), ch}
|
|
err := c.extractFromPath()
|
|
if err == nil {
|
|
t.Fatalf("Expected error")
|
|
}
|
|
expectEmptyChannel(t, ch)
|
|
}
|
|
|
|
func TestExtractFromEmptyDir(t *testing.T) {
|
|
dirName, err := ioutil.TempDir("", "foo")
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
defer os.RemoveAll(dirName)
|
|
|
|
ch := make(chan interface{}, 1)
|
|
c := sourceFile{dirName, ch}
|
|
err = c.extractFromPath()
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
|
|
update := (<-ch).(kubelet.PodUpdate)
|
|
expected := CreatePodUpdate(kubelet.SET, kubelet.FileSource)
|
|
if !api.Semantic.DeepDerivative(expected, update) {
|
|
t.Errorf("Expected %#v, Got %#v", expected, update)
|
|
}
|
|
}
|
|
|
|
func TestExtractFromDir(t *testing.T) {
|
|
manifest, expectedPod := ExampleManifestAndPod("1")
|
|
manifest2, expectedPod2 := ExampleManifestAndPod("2")
|
|
|
|
manifests := []v1beta1.ContainerManifest{manifest, manifest2}
|
|
pods := []api.Pod{expectedPod, expectedPod2}
|
|
files := make([]*os.File, len(manifests))
|
|
|
|
dirName, err := ioutil.TempDir("", "foo")
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
|
|
for i, manifest := range manifests {
|
|
data, err := json.Marshal(manifest)
|
|
if err != nil {
|
|
t.Errorf("Unexpected error: %v", err)
|
|
continue
|
|
}
|
|
file, err := ioutil.TempFile(dirName, manifest.ID)
|
|
if err != nil {
|
|
t.Errorf("Unexpected error: %v", err)
|
|
continue
|
|
}
|
|
name := file.Name()
|
|
if err := file.Close(); err != nil {
|
|
t.Errorf("Unexpected error: %v", err)
|
|
continue
|
|
}
|
|
ioutil.WriteFile(name, data, 0755)
|
|
files[i] = file
|
|
}
|
|
|
|
ch := make(chan interface{}, 1)
|
|
c := sourceFile{dirName, ch}
|
|
err = c.extractFromPath()
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
|
|
update := (<-ch).(kubelet.PodUpdate)
|
|
for i := range update.Pods {
|
|
// Pod name is generated with hash and is unique. Skip the comparision
|
|
// here by setting it to a simple value.
|
|
update.Pods[i].Name = manifests[i].ID
|
|
update.Pods[i].SelfLink = ""
|
|
}
|
|
expected := CreatePodUpdate(kubelet.SET, kubelet.FileSource, pods...)
|
|
for i := range expected.Pods {
|
|
// Pod name is generated with hash and is unique. Skip the comparision
|
|
// here by setting it to a simple value.
|
|
expected.Pods[i].Name = manifests[i].ID
|
|
}
|
|
sort.Sort(sortedPods(update.Pods))
|
|
sort.Sort(sortedPods(expected.Pods))
|
|
if !api.Semantic.DeepDerivative(expected, update) {
|
|
t.Fatalf("Expected %#v, Got %#v", expected, update)
|
|
}
|
|
for i := range update.Pods {
|
|
if errs := validation.ValidatePod(&update.Pods[i]); len(errs) != 0 {
|
|
t.Errorf("Expected no validation errors on %#v, Got %q", update.Pods[i], errs)
|
|
}
|
|
}
|
|
}
|