k3s/pkg/kubelet/kubelet_test.go

866 lines
21 KiB
Go
Raw Normal View History

2014-06-06 23:40:48 +00:00
/*
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 kubelet
import (
"bytes"
2014-06-06 23:40:48 +00:00
"encoding/json"
"fmt"
"io/ioutil"
2014-06-06 23:40:48 +00:00
"net/http/httptest"
"reflect"
"strings"
2014-06-06 23:40:48 +00:00
"sync"
"testing"
"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
"github.com/GoogleCloudPlatform/kubernetes/pkg/registry"
"github.com/GoogleCloudPlatform/kubernetes/pkg/util"
"github.com/coreos/go-etcd/etcd"
"github.com/fsouza/go-dockerclient"
)
// TODO: This doesn't reduce typing enough to make it worth the less readable errors. Remove.
func expectNoError(t *testing.T, err error) {
if err != nil {
t.Errorf("Unexpected error: %#v", err)
}
}
// These are used for testing extract json (below)
type TestData struct {
Value string
Number int
}
type TestObject struct {
Name string
Data TestData
}
func verifyStringEquals(t *testing.T, actual, expected string) {
if actual != expected {
t.Errorf("Verification failed. Expected: %s, Found %s", expected, actual)
}
}
func verifyIntEquals(t *testing.T, actual, expected int) {
if actual != expected {
t.Errorf("Verification failed. Expected: %d, Found %d", expected, actual)
}
}
func verifyNoError(t *testing.T, e error) {
if e != nil {
t.Errorf("Expected no error, found %#v", e)
}
}
func verifyError(t *testing.T, e error) {
if e == nil {
t.Errorf("Expected error, found nil")
}
}
func TestExtractJSON(t *testing.T) {
obj := TestObject{}
kubelet := Kubelet{}
data := `{ "name": "foo", "data": { "value": "bar", "number": 10 } }`
kubelet.ExtractYAMLData([]byte(data), &obj)
verifyStringEquals(t, obj.Name, "foo")
verifyStringEquals(t, obj.Data.Value, "bar")
verifyIntEquals(t, obj.Data.Number, 10)
}
type FakeDockerClient struct {
containerList []docker.APIContainers
container *docker.Container
err error
called []string
}
func (f *FakeDockerClient) clearCalls() {
f.called = []string{}
}
func (f *FakeDockerClient) appendCall(call string) {
f.called = append(f.called, call)
}
func (f *FakeDockerClient) ListContainers(options docker.ListContainersOptions) ([]docker.APIContainers, error) {
f.appendCall("list")
return f.containerList, f.err
}
func (f *FakeDockerClient) InspectContainer(id string) (*docker.Container, error) {
f.appendCall("inspect")
return f.container, f.err
}
func (f *FakeDockerClient) CreateContainer(docker.CreateContainerOptions) (*docker.Container, error) {
f.appendCall("create")
return nil, nil
}
func (f *FakeDockerClient) StartContainer(id string, hostConfig *docker.HostConfig) error {
f.appendCall("start")
return nil
}
func (f *FakeDockerClient) StopContainer(id string, timeout uint) error {
f.appendCall("stop")
return nil
}
func verifyCalls(t *testing.T, fakeDocker FakeDockerClient, calls []string) {
verifyStringArrayEquals(t, fakeDocker.called, calls)
}
func verifyStringArrayEquals(t *testing.T, actual, expected []string) {
invalid := len(actual) != len(expected)
for ix, value := range actual {
if expected[ix] != value {
invalid = true
}
}
if invalid {
t.Errorf("Expected: %#v, Actual: %#v", expected, actual)
}
}
func verifyPackUnpack(t *testing.T, manifestId, containerName string) {
name := manifestAndContainerToDockerName(
&api.ContainerManifest{Id: manifestId},
&api.Container{Name: containerName},
)
returnedManifestId, returnedContainerName := dockerNameToManifestAndContainer(name)
if manifestId != returnedManifestId || containerName != returnedContainerName {
t.Errorf("For (%s, %s), unpacked (%s, %s)", manifestId, containerName, returnedManifestId, returnedContainerName)
}
}
func TestContainerManifestNaming(t *testing.T) {
verifyPackUnpack(t, "manifest1234", "container5678")
verifyPackUnpack(t, "manifest--", "container__")
verifyPackUnpack(t, "--manifest", "__container")
verifyPackUnpack(t, "m___anifest_", "container-_-")
verifyPackUnpack(t, "_m___anifest", "-_-container")
}
func TestContainerExists(t *testing.T) {
fakeDocker := FakeDockerClient{
err: nil,
}
kubelet := Kubelet{
DockerClient: &fakeDocker,
}
manifest := api.ContainerManifest{
Id: "qux",
}
container := api.Container{
Name: "foo",
}
fakeDocker.containerList = []docker.APIContainers{
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"foo--qux--1234"},
},
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"bar--qux--1234"},
},
}
fakeDocker.container = &docker.Container{
ID: "foobar",
}
exists, _, err := kubelet.ContainerExists(&manifest, &container)
verifyCalls(t, fakeDocker, []string{"list", "list", "inspect"})
if !exists {
t.Errorf("Failed to find container %#v", container)
}
if err != nil {
t.Errorf("Unexpected error: %#v", err)
}
fakeDocker.clearCalls()
missingManifest := api.ContainerManifest{Id: "foobar"}
exists, _, err = kubelet.ContainerExists(&missingManifest, &container)
verifyCalls(t, fakeDocker, []string{"list"})
if exists {
t.Errorf("Failed to not find container %#v, missingManifest")
}
2014-06-06 23:40:48 +00:00
}
func TestGetContainerID(t *testing.T) {
fakeDocker := FakeDockerClient{
err: nil,
}
kubelet := Kubelet{
DockerClient: &fakeDocker,
}
fakeDocker.containerList = []docker.APIContainers{
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"foo"},
ID: "1234",
},
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"bar"},
ID: "4567",
},
}
id, err := kubelet.GetContainerID("foo")
verifyStringEquals(t, id, "1234")
verifyNoError(t, err)
verifyCalls(t, fakeDocker, []string{"list"})
fakeDocker.clearCalls()
id, err = kubelet.GetContainerID("bar")
verifyStringEquals(t, id, "4567")
verifyNoError(t, err)
verifyCalls(t, fakeDocker, []string{"list"})
fakeDocker.clearCalls()
id, err = kubelet.GetContainerID("NotFound")
verifyError(t, err)
verifyCalls(t, fakeDocker, []string{"list"})
}
func TestGetContainerByName(t *testing.T) {
fakeDocker := FakeDockerClient{
err: nil,
}
kubelet := Kubelet{
DockerClient: &fakeDocker,
}
fakeDocker.containerList = []docker.APIContainers{
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"foo"},
},
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"bar"},
},
}
fakeDocker.container = &docker.Container{
ID: "foobar",
}
container, err := kubelet.GetContainerByName("foo")
verifyCalls(t, fakeDocker, []string{"list", "inspect"})
if container == nil {
t.Errorf("Unexpected nil container")
}
verifyStringEquals(t, container.ID, "foobar")
verifyNoError(t, err)
}
func TestListContainers(t *testing.T) {
fakeDocker := FakeDockerClient{
err: nil,
}
kubelet := Kubelet{
DockerClient: &fakeDocker,
}
fakeDocker.containerList = []docker.APIContainers{
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"foo"},
},
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"bar"},
},
}
containers, err := kubelet.ListContainers()
verifyStringArrayEquals(t, containers, []string{"foo", "bar"})
verifyNoError(t, err)
verifyCalls(t, fakeDocker, []string{"list"})
}
func TestKillContainerWithError(t *testing.T) {
fakeDocker := FakeDockerClient{
err: fmt.Errorf("Sample Error"),
containerList: []docker.APIContainers{
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"foo"},
},
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"bar"},
},
},
}
kubelet := Kubelet{
DockerClient: &fakeDocker,
}
err := kubelet.KillContainer("foo")
verifyError(t, err)
verifyCalls(t, fakeDocker, []string{"list"})
}
func TestKillContainer(t *testing.T) {
fakeDocker := FakeDockerClient{
err: nil,
}
kubelet := Kubelet{
DockerClient: &fakeDocker,
}
fakeDocker.containerList = []docker.APIContainers{
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"foo"},
},
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"bar"},
},
}
fakeDocker.container = &docker.Container{
ID: "foobar",
}
err := kubelet.KillContainer("foo")
verifyNoError(t, err)
verifyCalls(t, fakeDocker, []string{"list", "stop"})
}
func TestResponseToContainersNil(t *testing.T) {
kubelet := Kubelet{}
list, err := kubelet.ResponseToManifests(&etcd.Response{Node: nil})
if len(list) != 0 {
t.Errorf("Unexpected non-zero list: %#v", list)
}
if err == nil {
t.Error("Unexpected non-error")
}
}
func TestResponseToManifests(t *testing.T) {
kubelet := Kubelet{}
list, err := kubelet.ResponseToManifests(&etcd.Response{
Node: &etcd.Node{
Value: util.MakeJSONString([]api.ContainerManifest{
2014-06-12 21:09:40 +00:00
{Id: "foo"},
{Id: "bar"},
2014-06-06 23:40:48 +00:00
}),
},
})
if len(list) != 2 || list[0].Id != "foo" || list[1].Id != "bar" {
t.Errorf("Unexpected list: %#v", list)
}
expectNoError(t, err)
}
type channelReader struct {
list [][]api.ContainerManifest
wg sync.WaitGroup
}
func startReading(channel <-chan []api.ContainerManifest) *channelReader {
cr := &channelReader{}
cr.wg.Add(1)
go func() {
for {
containers, ok := <-channel
if !ok {
break
}
cr.list = append(cr.list, containers)
}
cr.wg.Done()
}()
return cr
}
func (cr *channelReader) GetList() [][]api.ContainerManifest {
cr.wg.Wait()
return cr.list
}
func TestGetKubeletStateFromEtcdNoData(t *testing.T) {
fakeClient := registry.MakeFakeEtcdClient(t)
kubelet := Kubelet{
Client: fakeClient,
}
channel := make(chan []api.ContainerManifest)
reader := startReading(channel)
fakeClient.Data["/registry/hosts/machine/kubelet"] = registry.EtcdResponseWithError{
R: &etcd.Response{},
E: nil,
}
err := kubelet.getKubeletStateFromEtcd("/registry/hosts/machine", channel)
if err == nil {
t.Error("Unexpected no err.")
}
close(channel)
list := reader.GetList()
if len(list) != 0 {
t.Errorf("Unexpected list: %#v", list)
}
}
func TestGetKubeletStateFromEtcd(t *testing.T) {
fakeClient := registry.MakeFakeEtcdClient(t)
kubelet := Kubelet{
Client: fakeClient,
}
channel := make(chan []api.ContainerManifest)
reader := startReading(channel)
fakeClient.Data["/registry/hosts/machine/kubelet"] = registry.EtcdResponseWithError{
R: &etcd.Response{
Node: &etcd.Node{
Value: util.MakeJSONString([]api.Container{}),
},
},
E: nil,
}
err := kubelet.getKubeletStateFromEtcd("/registry/hosts/machine", channel)
expectNoError(t, err)
close(channel)
list := reader.GetList()
if len(list) != 1 {
t.Errorf("Unexpected list: %#v", list)
}
}
func TestGetKubeletStateFromEtcdNotFound(t *testing.T) {
fakeClient := registry.MakeFakeEtcdClient(t)
kubelet := Kubelet{
Client: fakeClient,
}
channel := make(chan []api.ContainerManifest)
reader := startReading(channel)
fakeClient.Data["/registry/hosts/machine/kubelet"] = registry.EtcdResponseWithError{
R: &etcd.Response{},
E: &etcd.EtcdError{
ErrorCode: 100,
},
}
err := kubelet.getKubeletStateFromEtcd("/registry/hosts/machine", channel)
expectNoError(t, err)
close(channel)
list := reader.GetList()
if len(list) != 0 {
t.Errorf("Unexpected list: %#v", list)
}
}
func TestGetKubeletStateFromEtcdError(t *testing.T) {
fakeClient := registry.MakeFakeEtcdClient(t)
kubelet := Kubelet{
Client: fakeClient,
}
channel := make(chan []api.ContainerManifest)
reader := startReading(channel)
fakeClient.Data["/registry/hosts/machine/kubelet"] = registry.EtcdResponseWithError{
R: &etcd.Response{},
E: &etcd.EtcdError{
ErrorCode: 200, // non not found error
},
}
err := kubelet.getKubeletStateFromEtcd("/registry/hosts/machine", channel)
if err == nil {
t.Error("Unexpected non-error")
}
close(channel)
list := reader.GetList()
if len(list) != 0 {
t.Errorf("Unexpected list: %#v", list)
}
}
func TestSyncManifestsDoesNothing(t *testing.T) {
fakeDocker := FakeDockerClient{
err: nil,
}
fakeDocker.containerList = []docker.APIContainers{
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
// format is <container-id>--<manifest-id>
Names: []string{"bar--foo"},
ID: "1234",
},
}
fakeDocker.container = &docker.Container{
ID: "1234",
}
kubelet := Kubelet{
DockerClient: &fakeDocker,
}
err := kubelet.SyncManifests([]api.ContainerManifest{
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Id: "foo",
Containers: []api.Container{
2014-06-12 21:09:40 +00:00
{Name: "bar"},
2014-06-06 23:40:48 +00:00
},
},
})
expectNoError(t, err)
if len(fakeDocker.called) != 4 ||
fakeDocker.called[0] != "list" ||
fakeDocker.called[1] != "list" ||
fakeDocker.called[2] != "inspect" ||
fakeDocker.called[3] != "list" {
t.Errorf("Unexpected call sequence: %#v", fakeDocker.called)
}
}
func TestSyncManifestsDeletes(t *testing.T) {
fakeDocker := FakeDockerClient{
err: nil,
}
fakeDocker.containerList = []docker.APIContainers{
2014-06-12 21:09:40 +00:00
{
2014-06-06 23:40:48 +00:00
Names: []string{"foo"},
ID: "1234",
},
}
kubelet := Kubelet{
DockerClient: &fakeDocker,
}
err := kubelet.SyncManifests([]api.ContainerManifest{})
expectNoError(t, err)
if len(fakeDocker.called) != 3 ||
fakeDocker.called[0] != "list" ||
fakeDocker.called[1] != "list" ||
fakeDocker.called[2] != "stop" {
t.Errorf("Unexpected call sequence: %#v", fakeDocker.called)
}
}
func TestEventWriting(t *testing.T) {
fakeEtcd := registry.MakeFakeEtcdClient(t)
kubelet := &Kubelet{
Client: fakeEtcd,
}
expectedEvent := api.Event{
Event: "test",
Container: &api.Container{
Name: "foo",
},
}
err := kubelet.LogEvent(&expectedEvent)
expectNoError(t, err)
if fakeEtcd.Ix != 1 {
t.Errorf("Unexpected number of children added: %d, expected 1", fakeEtcd.Ix)
}
response, err := fakeEtcd.Get("/events/foo/1", false, false)
expectNoError(t, err)
var event api.Event
err = json.Unmarshal([]byte(response.Node.Value), &event)
expectNoError(t, err)
if event.Event != expectedEvent.Event ||
event.Container.Name != expectedEvent.Container.Name {
t.Errorf("Event's don't match. Expected: %#v Saw: %#v", expectedEvent, event)
}
}
func TestEventWritingError(t *testing.T) {
fakeEtcd := registry.MakeFakeEtcdClient(t)
kubelet := &Kubelet{
Client: fakeEtcd,
}
fakeEtcd.Err = fmt.Errorf("Test error")
err := kubelet.LogEvent(&api.Event{
Event: "test",
Container: &api.Container{
Name: "foo",
},
})
if err == nil {
t.Errorf("Unexpected non-error")
}
}
func TestMakeCommandLine(t *testing.T) {
expected := []string{"echo", "hello", "world"}
container := api.Container{
Command: strings.Join(expected, " "),
}
cmdLine := makeCommandLine(&container)
if !reflect.DeepEqual(expected, cmdLine) {
t.Error("Unexpected command line. Expected %#v, got %#v", expected, cmdLine)
}
}
func TestMakeEnvVariables(t *testing.T) {
container := api.Container{
Env: []api.EnvVar{
api.EnvVar{
Name: "foo",
Value: "bar",
},
api.EnvVar{
Name: "baz",
Value: "blah",
},
},
}
vars := makeEnvironmentVariables(&container)
if len(vars) != len(container.Env) {
t.Errorf("Vars don't match. Expected: %#v Found: %#v", container.Env, vars)
}
for ix, env := range container.Env {
value := fmt.Sprintf("%s=%s", env.Name, env.Value)
if value != vars[ix] {
t.Errorf("Unexpected value: %s. Expected: %s", vars[ix], value)
}
}
}
func TestMakeVolumesAndBinds(t *testing.T) {
container := api.Container{
VolumeMounts: []api.VolumeMount{
api.VolumeMount{
MountPath: "/mnt/path",
Name: "disk",
ReadOnly: false,
},
api.VolumeMount{
MountPath: "/mnt/path2",
Name: "disk2",
ReadOnly: true,
},
},
}
volumes, binds := makeVolumesAndBinds(&container)
if len(volumes) != len(container.VolumeMounts) ||
len(binds) != len(container.VolumeMounts) {
t.Errorf("Unexpected volumes and binds: %#v %#v. Container was: %#v", volumes, binds, container)
}
for ix, volume := range container.VolumeMounts {
expectedBind := "/exports/" + volume.Name + ":" + volume.MountPath
if volume.ReadOnly {
expectedBind = expectedBind + ":ro"
}
if binds[ix] != expectedBind {
t.Errorf("Unexpected bind. Expected %s. Found %s", expectedBind, binds[ix])
}
if _, ok := volumes[volume.MountPath]; !ok {
t.Errorf("Map is missing key: %s. %#v", volume.MountPath, volumes)
}
}
}
func TestMakePortsAndBindings(t *testing.T) {
container := api.Container{
Ports: []api.Port{
api.Port{
ContainerPort: 80,
HostPort: 8080,
},
api.Port{
ContainerPort: 443,
HostPort: 443,
},
},
}
exposedPorts, bindings := makePortsAndBindings(&container)
if len(container.Ports) != len(exposedPorts) ||
len(container.Ports) != len(bindings) {
t.Errorf("Unexpected ports and bindings, %#v %#v %#v", container, exposedPorts, bindings)
}
}
func TestExtractFromNonExistentFile(t *testing.T) {
kubelet := Kubelet{}
changeChannel := make(chan api.ContainerManifest)
lastData := []byte{1, 2, 3}
data, err := kubelet.extractFromFile(lastData, "/some/fake/file", changeChannel)
if err == nil {
t.Error("Unexpected non-error.")
}
if !bytes.Equal(data, lastData) {
t.Errorf("Unexpected data response. Expected %#v, found %#v", lastData, data)
}
}
func TestExtractFromBadDataFile(t *testing.T) {
kubelet := Kubelet{}
changeChannel := make(chan api.ContainerManifest)
lastData := []byte{1, 2, 3}
file, err := ioutil.TempFile("", "foo")
expectNoError(t, err)
name := file.Name()
file.Close()
ioutil.WriteFile(name, lastData, 0755)
data, err := kubelet.extractFromFile(lastData, name, changeChannel)
if err == nil {
t.Error("Unexpected non-error.")
}
if !bytes.Equal(data, lastData) {
t.Errorf("Unexpected data response. Expected %#v, found %#v", lastData, data)
}
}
func TestExtractFromSameDataFile(t *testing.T) {
kubelet := Kubelet{}
changeChannel := make(chan api.ContainerManifest)
manifest := api.ContainerManifest{
Id: "foo",
}
lastData, err := json.Marshal(manifest)
expectNoError(t, err)
file, err := ioutil.TempFile("", "foo")
expectNoError(t, err)
name := file.Name()
expectNoError(t, file.Close())
ioutil.WriteFile(name, lastData, 0755)
data, err := kubelet.extractFromFile(lastData, name, changeChannel)
expectNoError(t, err)
if !bytes.Equal(data, lastData) {
t.Errorf("Unexpected data response. Expected %#v, found %#v", lastData, data)
}
}
func TestExtractFromChangedDataFile(t *testing.T) {
kubelet := Kubelet{}
changeChannel := make(chan api.ContainerManifest)
reader := startReadingSingle(changeChannel)
oldManifest := api.ContainerManifest{
Id: "foo",
}
newManifest := api.ContainerManifest{
Id: "bar",
}
lastData, err := json.Marshal(oldManifest)
expectNoError(t, err)
newData, err := json.Marshal(newManifest)
expectNoError(t, err)
file, err := ioutil.TempFile("", "foo")
expectNoError(t, err)
name := file.Name()
expectNoError(t, file.Close())
ioutil.WriteFile(name, newData, 0755)
data, err := kubelet.extractFromFile(lastData, name, changeChannel)
close(changeChannel)
expectNoError(t, err)
if !bytes.Equal(data, newData) {
t.Errorf("Unexpected data response. Expected %#v, found %#v", lastData, data)
}
read := reader.GetList()
if len(read) != 1 {
t.Errorf("Unexpected channel traffic: %#v", read)
}
if !reflect.DeepEqual(read[0], newManifest) {
t.Errorf("Unexpected difference. Expected %#v, got %#v", newManifest, read[0])
}
}
func TestExtractFromHttpBadness(t *testing.T) {
kubelet := Kubelet{}
lastData := []byte{1, 2, 3}
changeChannel := make(chan api.ContainerManifest)
data, err := kubelet.extractFromHTTP(lastData, "http://localhost:12345", changeChannel)
if err == nil {
t.Error("Unexpected non-error.")
}
if !bytes.Equal(lastData, data) {
t.Errorf("Unexpected difference. Expected: %#v, Saw: %#v", lastData, data)
}
}
func TestExtractFromHttpNoChange(t *testing.T) {
kubelet := Kubelet{}
changeChannel := make(chan api.ContainerManifest)
manifest := api.ContainerManifest{
Id: "foo",
}
lastData, err := json.Marshal(manifest)
fakeHandler := util.FakeHandler{
StatusCode: 200,
ResponseBody: string(lastData),
}
testServer := httptest.NewServer(&fakeHandler)
data, err := kubelet.extractFromHTTP(lastData, testServer.URL, changeChannel)
if err != nil {
t.Errorf("Unexpected error: %#v", err)
}
if !bytes.Equal(lastData, data) {
t.Errorf("Unexpected difference. Expected: %#v, Saw: %#v", lastData, data)
}
}
func TestExtractFromHttpChanges(t *testing.T) {
kubelet := Kubelet{}
changeChannel := make(chan api.ContainerManifest)
reader := startReadingSingle(changeChannel)
manifest := api.ContainerManifest{
Id: "foo",
}
newManifest := api.ContainerManifest{
Id: "bar",
}
lastData, _ := json.Marshal(manifest)
newData, _ := json.Marshal(newManifest)
fakeHandler := util.FakeHandler{
StatusCode: 200,
ResponseBody: string(newData),
}
testServer := httptest.NewServer(&fakeHandler)
data, err := kubelet.extractFromHTTP(lastData, testServer.URL, changeChannel)
close(changeChannel)
read := reader.GetList()
if err != nil {
t.Errorf("Unexpected error: %#v", err)
}
if len(read) != 1 {
t.Errorf("Unexpected list: %#v", read)
}
if !bytes.Equal(newData, data) {
t.Errorf("Unexpected difference. Expected: %#v, Saw: %#v", lastData, data)
}
if !reflect.DeepEqual(newManifest, read[0]) {
t.Errorf("Unexpected difference. Expected: %#v, Saw: %#v", newManifest, read[0])
}
}
func TestWatchEtcd(t *testing.T) {
watchChannel := make(chan *etcd.Response)
changeChannel := make(chan []api.ContainerManifest)
kubelet := Kubelet{}
reader := startReading(changeChannel)
manifest := []api.ContainerManifest{
api.ContainerManifest{
Id: "foo",
},
}
data, err := json.Marshal(manifest)
expectNoError(t, err)
go kubelet.WatchEtcd(watchChannel, changeChannel)
watchChannel <- &etcd.Response{
Node: &etcd.Node{
Value: string(data),
},
}
close(watchChannel)
close(changeChannel)
read := reader.GetList()
if len(read) != 1 ||
!reflect.DeepEqual(read[0], manifest) {
t.Errorf("Unexpected manifest(s) %#v %#v", read[0], manifest)
}
}