Move scheduler to own package.

pkg/scheduler/scheduler_test.go

@@ -0,0 +1,115 @@
+/*
+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 registry
+
+import (
+	"math/rand"
+	"testing"
+
+	"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
+)
+
+func expectSchedule(scheduler Scheduler, pod api.Pod, expected string, t *testing.T) {
+	actual, err := scheduler.Schedule(pod)
+	expectNoError(t, err)
+	if actual != expected {
+		t.Errorf("Unexpected scheduling value: %v, expected %v", actual, expected)
+	}
+}
+
+func TestRoundRobinScheduler(t *testing.T) {
+	scheduler := MakeRoundRobinScheduler(MakeMinionRegistry([]string{"m1", "m2", "m3", "m4"}))
+	expectSchedule(scheduler, api.Pod{}, "m1", t)
+	expectSchedule(scheduler, api.Pod{}, "m2", t)
+	expectSchedule(scheduler, api.Pod{}, "m3", t)
+	expectSchedule(scheduler, api.Pod{}, "m4", t)
+}
+
+func TestRandomScheduler(t *testing.T) {
+	random := rand.New(rand.NewSource(0))
+	scheduler := MakeRandomScheduler(MakeMinionRegistry([]string{"m1", "m2", "m3", "m4"}), *random)
+	_, err := scheduler.Schedule(api.Pod{})
+	expectNoError(t, err)
+}
+
+func TestFirstFitSchedulerNothingScheduled(t *testing.T) {
+	mockRegistry := MockPodRegistry{}
+	r := rand.New(rand.NewSource(0))
+	scheduler := MakeFirstFitScheduler(MakeMinionRegistry([]string{"m1", "m2", "m3"}), &mockRegistry, r)
+	expectSchedule(scheduler, api.Pod{}, "m3", t)
+}
+
+func makePod(host string, hostPorts ...int) api.Pod {
+	networkPorts := []api.Port{}
+	for _, port := range hostPorts {
+		networkPorts = append(networkPorts, api.Port{HostPort: port})
+	}
+	return api.Pod{
+		CurrentState: api.PodState{
+			Host: host,
+		},
+		DesiredState: api.PodState{
+			Manifest: api.ContainerManifest{
+				Containers: []api.Container{
+					{
+						Ports: networkPorts,
+					},
+				},
+			},
+		},
+	}
+}
+
+func TestFirstFitSchedulerFirstScheduled(t *testing.T) {
+	mockRegistry := MockPodRegistry{
+		pods: []api.Pod{
+			makePod("m1", 8080),
+		},
+	}
+	r := rand.New(rand.NewSource(0))
+	scheduler := MakeFirstFitScheduler(MakeMinionRegistry([]string{"m1", "m2", "m3"}), &mockRegistry, r)
+	expectSchedule(scheduler, makePod("", 8080), "m3", t)
+}
+
+func TestFirstFitSchedulerFirstScheduledComplicated(t *testing.T) {
+	mockRegistry := MockPodRegistry{
+		pods: []api.Pod{
+			makePod("m1", 80, 8080),
+			makePod("m2", 8081, 8082, 8083),
+			makePod("m3", 80, 443, 8085),
+		},
+	}
+	r := rand.New(rand.NewSource(0))
+	scheduler := MakeFirstFitScheduler(MakeMinionRegistry([]string{"m1", "m2", "m3"}), &mockRegistry, r)
+	expectSchedule(scheduler, makePod("", 8080, 8081), "m3", t)
+}
+
+func TestFirstFitSchedulerFirstScheduledImpossible(t *testing.T) {
+	mockRegistry := MockPodRegistry{
+		pods: []api.Pod{
+			makePod("m1", 8080),
+			makePod("m2", 8081),
+			makePod("m3", 8080),
+		},
+	}
+	r := rand.New(rand.NewSource(0))
+	scheduler := MakeFirstFitScheduler(MakeMinionRegistry([]string{"m1", "m2", "m3"}), &mockRegistry, r)
+	_, err := scheduler.Schedule(makePod("", 8080, 8081))
+	if err == nil {
+		t.Error("Unexpected non-error.")
+	}
+}