Merge pull request #20030 from kubernetes/revert-18169-smarter-scaling

Revert "kubectl: Make scaling smarter"
This commit is contained in:
Zach Loafman 2016-01-22 15:07:26 -08:00
commit 6397b781ac
6 changed files with 335 additions and 585 deletions

View File

@ -112,7 +112,7 @@ func JobHasDesiredParallelism(c ExtensionsInterface, job *extensions.Job) wait.C
} }
// desired parallelism can be either the exact number, in which case return immediately // desired parallelism can be either the exact number, in which case return immediately
if job.Spec.Parallelism != nil && job.Status.Active == *job.Spec.Parallelism { if job.Status.Active == *job.Spec.Parallelism {
return true, nil return true, nil
} }
// otherwise count successful // otherwise count successful

View File

@ -146,10 +146,6 @@ func RunScale(f *cmdutil.Factory, out io.Writer, cmd *cobra.Command, args []stri
errs := []error{} errs := []error{}
for _, info := range infos { for _, info := range infos {
if err := scaler.Scale(info.Namespace, info.Name, uint(count), precondition, retry, waitForReplicas); err != nil { if err := scaler.Scale(info.Namespace, info.Name, uint(count), precondition, retry, waitForReplicas); err != nil {
if scaleErr, ok := err.(kubectl.ScaleError); ok && scaleErr.FailureType == kubectl.AlreadyScaled {
cmdutil.PrintSuccess(mapper, shortOutput, out, info.Mapping.Resource, info.Name, "already scaled")
continue
}
errs = append(errs, err) errs = append(errs, err)
continue continue
} }

View File

@ -17,7 +17,6 @@ limitations under the License.
package kubectl package kubectl
import ( import (
goerrors "errors"
"fmt" "fmt"
"strconv" "strconv"
"time" "time"
@ -80,13 +79,8 @@ const (
ScaleGetFailure ScaleErrorType = iota ScaleGetFailure ScaleErrorType = iota
ScaleUpdateFailure ScaleUpdateFailure
ScaleUpdateInvalidFailure ScaleUpdateInvalidFailure
// AlreadyScaled is not really an error but we need a way to surface to the client that
// the scaling didn't happen because we already have the desired state the user asked for.
AlreadyScaled
) )
var alreadyScaledErr = goerrors.New("desired replicas already equals the requested replicas")
// A ScaleError is returned when a scale request passes // A ScaleError is returned when a scale request passes
// preconditions but fails to actually scale the controller. // preconditions but fails to actually scale the controller.
type ScaleError struct { type ScaleError struct {
@ -118,14 +112,12 @@ func ScaleCondition(r Scaler, precondition *ScalePrecondition, namespace, name s
case nil: case nil:
return true, nil return true, nil
case ScaleError: case ScaleError:
switch e.FailureType {
case ScaleUpdateInvalidFailure:
// if it's invalid we shouldn't keep waiting // if it's invalid we shouldn't keep waiting
if e.FailureType == ScaleUpdateInvalidFailure {
return false, err return false, err
case ScaleUpdateFailure: }
if e.FailureType == ScaleUpdateFailure {
return false, nil return false, nil
case AlreadyScaled:
return false, err
} }
} }
return false, err return false, err
@ -157,10 +149,8 @@ func (scaler *ReplicationControllerScaler) ScaleSimple(namespace, name string, p
return err return err
} }
} }
if controller.Spec.Replicas == int(newSize) {
return ScaleError{AlreadyScaled, controller.ResourceVersion, alreadyScaledErr}
}
controller.Spec.Replicas = int(newSize) controller.Spec.Replicas = int(newSize)
// TODO: do retry on 409 errors here?
if _, err := scaler.c.ReplicationControllers(namespace).Update(controller); err != nil { if _, err := scaler.c.ReplicationControllers(namespace).Update(controller); err != nil {
if errors.IsInvalid(err) { if errors.IsInvalid(err) {
return ScaleError{ScaleUpdateInvalidFailure, controller.ResourceVersion, err} return ScaleError{ScaleUpdateInvalidFailure, controller.ResourceVersion, err}
@ -226,9 +216,6 @@ func (scaler *JobScaler) ScaleSimple(namespace, name string, preconditions *Scal
} }
} }
parallelism := int(newSize) parallelism := int(newSize)
if job.Spec.Parallelism != nil && *job.Spec.Parallelism == parallelism {
return ScaleError{AlreadyScaled, job.ResourceVersion, alreadyScaledErr}
}
job.Spec.Parallelism = &parallelism job.Spec.Parallelism = &parallelism
if _, err := scaler.c.Jobs(namespace).Update(job); err != nil { if _, err := scaler.c.Jobs(namespace).Update(job); err != nil {
if errors.IsInvalid(err) { if errors.IsInvalid(err) {
@ -292,9 +279,6 @@ func (scaler *DeploymentScaler) ScaleSimple(namespace, name string, precondition
} }
} }
scale := extensions.ScaleFromDeployment(deployment) scale := extensions.ScaleFromDeployment(deployment)
if scale.Spec.Replicas == int(newSize) {
return ScaleError{AlreadyScaled, deployment.ResourceVersion, alreadyScaledErr}
}
scale.Spec.Replicas = int(newSize) scale.Spec.Replicas = int(newSize)
if _, err := scaler.c.Scales(namespace).Update("Deployment", scale); err != nil { if _, err := scaler.c.Scales(namespace).Update("Deployment", scale); err != nil {
if errors.IsInvalid(err) { if errors.IsInvalid(err) {

View File

@ -132,31 +132,6 @@ func TestReplicationControllerScaleFailsPreconditions(t *testing.T) {
} }
} }
func TestReplicationControllerAlreadyScaled(t *testing.T) {
fake := testclient.NewSimpleFake(&api.ReplicationController{
Spec: api.ReplicationControllerSpec{
Replicas: 3,
},
})
scaler := ReplicationControllerScaler{fake}
count := uint(3)
name := "foo"
err := scaler.Scale("default", name, count, nil, nil, nil)
if err == nil {
t.Fatal("expected AlreadyScaled error, got nil")
}
if scaleErr, ok := err.(ScaleError); !ok || scaleErr.FailureType != AlreadyScaled {
t.Fatalf("expected AlreadyScaled error, got %s", scaleErr.FailureType)
}
actions := fake.Actions()
if len(actions) != 1 {
t.Fatalf("unexpected actions: %v, expected 1 action (get)", actions)
}
if action, ok := actions[0].(testclient.GetAction); !ok || action.GetResource() != "replicationcontrollers" || action.GetName() != name {
t.Errorf("unexpected action: %v, expected get-replicationController %s", actions[0], name)
}
}
func TestValidateReplicationController(t *testing.T) { func TestValidateReplicationController(t *testing.T) {
tests := []struct { tests := []struct {
preconditions ScalePrecondition preconditions ScalePrecondition
@ -387,32 +362,6 @@ func TestJobScaleFailsPreconditions(t *testing.T) {
} }
} }
func TestJobAlreadyScaled(t *testing.T) {
three := 3
fake := testclient.NewSimpleFake(&extensions.Job{
Spec: extensions.JobSpec{
Parallelism: &three,
},
})
scaler := JobScaler{&testclient.FakeExperimental{fake}}
count := uint(3)
name := "foo"
err := scaler.Scale("default", name, count, nil, nil, nil)
if err == nil {
t.Fatal("expected AlreadyScaled error, got nil")
}
if scaleErr, ok := err.(ScaleError); !ok || scaleErr.FailureType != AlreadyScaled {
t.Fatalf("expected AlreadyScaled error, got %s", scaleErr.FailureType)
}
actions := fake.Actions()
if len(actions) != 1 {
t.Fatalf("unexpected actions: %v, expected 1 action (get)", actions)
}
if action, ok := actions[0].(testclient.GetAction); !ok || action.GetResource() != "jobs" || action.GetName() != name {
t.Errorf("unexpected action: %v, expected get-job %s", actions[0], name)
}
}
func TestValidateJob(t *testing.T) { func TestValidateJob(t *testing.T) {
zero, ten, twenty := 0, 10, 20 zero, ten, twenty := 0, 10, 20
tests := []struct { tests := []struct {
@ -677,31 +626,6 @@ func TestDeploymentScaleFailsPreconditions(t *testing.T) {
} }
} }
func TestDeploymentAlreadyScaled(t *testing.T) {
fake := testclient.NewSimpleFake(&extensions.Deployment{
Spec: extensions.DeploymentSpec{
Replicas: 3,
},
})
scaler := DeploymentScaler{&testclient.FakeExperimental{fake}}
count := uint(3)
name := "foo"
err := scaler.Scale("default", name, count, nil, nil, nil)
if err == nil {
t.Fatal("expected AlreadyScaled error, got nil")
}
if scaleErr, ok := err.(ScaleError); !ok || scaleErr.FailureType != AlreadyScaled {
t.Fatalf("expected AlreadyScaled error, got %s", scaleErr.FailureType)
}
actions := fake.Actions()
if len(actions) != 1 {
t.Fatalf("unexpected actions: %v, expected 1 action (get)", actions)
}
if action, ok := actions[0].(testclient.GetAction); !ok || action.GetResource() != "deployments" || action.GetName() != name {
t.Errorf("unexpected action: %v, expected get-deployment %s", actions[0], name)
}
}
func TestValidateDeployment(t *testing.T) { func TestValidateDeployment(t *testing.T) {
zero, ten, twenty := 0, 10, 20 zero, ten, twenty := 0, 10, 20
tests := []struct { tests := []struct {

View File

@ -23,6 +23,7 @@ import (
"k8s.io/kubernetes/pkg/api" "k8s.io/kubernetes/pkg/api"
"k8s.io/kubernetes/pkg/api/errors" "k8s.io/kubernetes/pkg/api/errors"
"k8s.io/kubernetes/pkg/api/meta"
"k8s.io/kubernetes/pkg/api/unversioned" "k8s.io/kubernetes/pkg/api/unversioned"
"k8s.io/kubernetes/pkg/apis/extensions" "k8s.io/kubernetes/pkg/apis/extensions"
client "k8s.io/kubernetes/pkg/client/unversioned" client "k8s.io/kubernetes/pkg/client/unversioned"
@ -82,6 +83,30 @@ func ReaperForReplicationController(c client.Interface, timeout time.Duration) (
return &ReplicationControllerReaper{c, Interval, timeout}, nil return &ReplicationControllerReaper{c, Interval, timeout}, nil
} }
type ReplicationControllerReaper struct {
client.Interface
pollInterval, timeout time.Duration
}
type DaemonSetReaper struct {
client.Interface
pollInterval, timeout time.Duration
}
type JobReaper struct {
client.Interface
pollInterval, timeout time.Duration
}
type PodReaper struct {
client.Interface
}
type ServiceReaper struct {
client.Interface
}
type objInterface interface {
Delete(name string) error
Get(name string) (meta.Object, error)
}
// getOverlappingControllers finds rcs that this controller overlaps, as well as rcs overlapping this controller. // getOverlappingControllers finds rcs that this controller overlaps, as well as rcs overlapping this controller.
func getOverlappingControllers(c client.ReplicationControllerInterface, rc *api.ReplicationController) ([]api.ReplicationController, error) { func getOverlappingControllers(c client.ReplicationControllerInterface, rc *api.ReplicationController) ([]api.ReplicationController, error) {
rcs, err := c.List(api.ListOptions{}) rcs, err := c.List(api.ListOptions{})
@ -99,11 +124,6 @@ func getOverlappingControllers(c client.ReplicationControllerInterface, rc *api.
return matchingRCs, nil return matchingRCs, nil
} }
type ReplicationControllerReaper struct {
client.Interface
pollInterval, timeout time.Duration
}
func (reaper *ReplicationControllerReaper) Stop(namespace, name string, timeout time.Duration, gracePeriod *api.DeleteOptions) error { func (reaper *ReplicationControllerReaper) Stop(namespace, name string, timeout time.Duration, gracePeriod *api.DeleteOptions) error {
rc := reaper.ReplicationControllers(namespace) rc := reaper.ReplicationControllers(namespace)
scaler, err := ScalerFor(api.Kind("ReplicationController"), *reaper) scaler, err := ScalerFor(api.Kind("ReplicationController"), *reaper)
@ -161,22 +181,15 @@ func (reaper *ReplicationControllerReaper) Stop(namespace, name string, timeout
retry := NewRetryParams(reaper.pollInterval, reaper.timeout) retry := NewRetryParams(reaper.pollInterval, reaper.timeout)
waitForReplicas := NewRetryParams(reaper.pollInterval, timeout) waitForReplicas := NewRetryParams(reaper.pollInterval, timeout)
if err = scaler.Scale(namespace, name, 0, nil, retry, waitForReplicas); err != nil { if err = scaler.Scale(namespace, name, 0, nil, retry, waitForReplicas); err != nil {
if scaleErr, ok := err.(ScaleError); !ok || scaleErr.FailureType != AlreadyScaled {
return err return err
} }
} }
}
if err := rc.Delete(name); err != nil { if err := rc.Delete(name); err != nil {
return err return err
} }
return nil return nil
} }
type DaemonSetReaper struct {
client.Interface
pollInterval, timeout time.Duration
}
func (reaper *DaemonSetReaper) Stop(namespace, name string, timeout time.Duration, gracePeriod *api.DeleteOptions) error { func (reaper *DaemonSetReaper) Stop(namespace, name string, timeout time.Duration, gracePeriod *api.DeleteOptions) error {
ds, err := reaper.Extensions().DaemonSets(namespace).Get(name) ds, err := reaper.Extensions().DaemonSets(namespace).Get(name)
if err != nil { if err != nil {
@ -214,11 +227,6 @@ func (reaper *DaemonSetReaper) Stop(namespace, name string, timeout time.Duratio
return nil return nil
} }
type JobReaper struct {
client.Interface
pollInterval, timeout time.Duration
}
func (reaper *JobReaper) Stop(namespace, name string, timeout time.Duration, gracePeriod *api.DeleteOptions) error { func (reaper *JobReaper) Stop(namespace, name string, timeout time.Duration, gracePeriod *api.DeleteOptions) error {
jobs := reaper.Extensions().Jobs(namespace) jobs := reaper.Extensions().Jobs(namespace)
pods := reaper.Pods(namespace) pods := reaper.Pods(namespace)
@ -240,10 +248,8 @@ func (reaper *JobReaper) Stop(namespace, name string, timeout time.Duration, gra
retry := NewRetryParams(reaper.pollInterval, reaper.timeout) retry := NewRetryParams(reaper.pollInterval, reaper.timeout)
waitForJobs := NewRetryParams(reaper.pollInterval, timeout) waitForJobs := NewRetryParams(reaper.pollInterval, timeout)
if err = scaler.Scale(namespace, name, 0, nil, retry, waitForJobs); err != nil { if err = scaler.Scale(namespace, name, 0, nil, retry, waitForJobs); err != nil {
if scaleErr, ok := err.(ScaleError); !ok || scaleErr.FailureType != AlreadyScaled {
return err return err
} }
}
// at this point only dead pods are left, that should be removed // at this point only dead pods are left, that should be removed
selector, _ := extensions.LabelSelectorAsSelector(job.Spec.Selector) selector, _ := extensions.LabelSelectorAsSelector(job.Spec.Selector)
options := api.ListOptions{LabelSelector: selector} options := api.ListOptions{LabelSelector: selector}
@ -270,10 +276,6 @@ func (reaper *JobReaper) Stop(namespace, name string, timeout time.Duration, gra
return nil return nil
} }
type PodReaper struct {
client.Interface
}
func (reaper *PodReaper) Stop(namespace, name string, timeout time.Duration, gracePeriod *api.DeleteOptions) error { func (reaper *PodReaper) Stop(namespace, name string, timeout time.Duration, gracePeriod *api.DeleteOptions) error {
pods := reaper.Pods(namespace) pods := reaper.Pods(namespace)
_, err := pods.Get(name) _, err := pods.Get(name)
@ -287,10 +289,6 @@ func (reaper *PodReaper) Stop(namespace, name string, timeout time.Duration, gra
return nil return nil
} }
type ServiceReaper struct {
client.Interface
}
func (reaper *ServiceReaper) Stop(namespace, name string, timeout time.Duration, gracePeriod *api.DeleteOptions) error { func (reaper *ServiceReaper) Stop(namespace, name string, timeout time.Duration, gracePeriod *api.DeleteOptions) error {
services := reaper.Services(namespace) services := reaper.Services(namespace)
_, err := services.Get(name) _, err := services.Get(name)

View File

@ -19,6 +19,7 @@ package kubectl
import ( import (
"fmt" "fmt"
"reflect" "reflect"
"strings"
"testing" "testing"
"time" "time"
@ -30,339 +31,256 @@ import (
"k8s.io/kubernetes/pkg/runtime" "k8s.io/kubernetes/pkg/runtime"
) )
const ( func TestReplicationControllerStop(t *testing.T) {
name = "foo" name := "foo"
) ns := "default"
tests := []struct {
func overlappingButSafe() *api.ReplicationControllerList { Name string
return &api.ReplicationControllerList{ Objs []runtime.Object
StopError error
ExpectedActions []string
}{
{
Name: "OnlyOneRC",
Objs: []runtime.Object{
&api.ReplicationController{ // GET
ObjectMeta: api.ObjectMeta{
Name: name,
Namespace: ns,
},
Spec: api.ReplicationControllerSpec{
Replicas: 0,
Selector: map[string]string{"k1": "v1"}},
},
&api.ReplicationControllerList{ // LIST
Items: []api.ReplicationController{ Items: []api.ReplicationController{
{ {
ObjectMeta: api.ObjectMeta{ ObjectMeta: api.ObjectMeta{
Name: name, Name: name,
Namespace: api.NamespaceDefault, Namespace: ns,
}, },
Spec: api.ReplicationControllerSpec{ Spec: api.ReplicationControllerSpec{
Replicas: 1, Replicas: 0,
Selector: map[string]string{"k1": "v1"}},
},
},
},
},
StopError: nil,
ExpectedActions: []string{"get", "list", "get", "update", "get", "get", "delete"},
},
{
Name: "NoOverlapping",
Objs: []runtime.Object{
&api.ReplicationController{ // GET
ObjectMeta: api.ObjectMeta{
Name: name,
Namespace: ns,
},
Spec: api.ReplicationControllerSpec{
Replicas: 0,
Selector: map[string]string{"k1": "v1"}},
},
&api.ReplicationControllerList{ // LIST
Items: []api.ReplicationController{
{
ObjectMeta: api.ObjectMeta{
Name: "baz",
Namespace: ns,
},
Spec: api.ReplicationControllerSpec{
Replicas: 0,
Selector: map[string]string{"k3": "v3"}},
},
{
ObjectMeta: api.ObjectMeta{
Name: name,
Namespace: ns,
},
Spec: api.ReplicationControllerSpec{
Replicas: 0,
Selector: map[string]string{"k1": "v1"}},
},
},
},
},
StopError: nil,
ExpectedActions: []string{"get", "list", "get", "update", "get", "get", "delete"},
},
{
Name: "OverlappingError",
Objs: []runtime.Object{
&api.ReplicationController{ // GET
ObjectMeta: api.ObjectMeta{
Name: name,
Namespace: ns,
},
Spec: api.ReplicationControllerSpec{
Replicas: 0,
Selector: map[string]string{"k1": "v1"}},
},
&api.ReplicationControllerList{ // LIST
Items: []api.ReplicationController{
{
ObjectMeta: api.ObjectMeta{
Name: "baz",
Namespace: ns,
},
Spec: api.ReplicationControllerSpec{
Replicas: 0,
Selector: map[string]string{"k1": "v1", "k2": "v2"}}, Selector: map[string]string{"k1": "v1", "k2": "v2"}},
}, },
{ {
ObjectMeta: api.ObjectMeta{ ObjectMeta: api.ObjectMeta{
Name: "baz", Name: name,
Namespace: api.NamespaceDefault, Namespace: ns,
}, },
Spec: api.ReplicationControllerSpec{ Spec: api.ReplicationControllerSpec{
Replicas: 2, Replicas: 0,
Selector: map[string]string{"k1": "v1"}},
},
},
},
},
StopError: fmt.Errorf("Detected overlapping controllers for rc foo: baz, please manage deletion individually with --cascade=false."),
ExpectedActions: []string{"get", "list"},
},
{
Name: "OverlappingButSafeDelete",
Objs: []runtime.Object{
&api.ReplicationController{ // GET
ObjectMeta: api.ObjectMeta{
Name: name,
Namespace: ns,
},
Spec: api.ReplicationControllerSpec{
Replicas: 0,
Selector: map[string]string{"k1": "v1", "k2": "v2"}},
},
&api.ReplicationControllerList{ // LIST
Items: []api.ReplicationController{
{
ObjectMeta: api.ObjectMeta{
Name: "baz",
Namespace: ns,
},
Spec: api.ReplicationControllerSpec{
Replicas: 0,
Selector: map[string]string{"k1": "v1", "k2": "v2", "k3": "v3"}}, Selector: map[string]string{"k1": "v1", "k2": "v2", "k3": "v3"}},
}, },
{ {
ObjectMeta: api.ObjectMeta{ ObjectMeta: api.ObjectMeta{
Name: "zaz", Name: "zaz",
Namespace: api.NamespaceDefault, Namespace: ns,
},
Spec: api.ReplicationControllerSpec{
Replicas: 3,
Selector: map[string]string{"k1": "v1"}},
},
},
}
}
func exactMatches() *api.ReplicationControllerList {
return &api.ReplicationControllerList{
Items: []api.ReplicationController{
{
ObjectMeta: api.ObjectMeta{
Name: "zaz",
Namespace: api.NamespaceDefault,
},
Spec: api.ReplicationControllerSpec{
Replicas: 3,
Selector: map[string]string{"k1": "v1"}},
},
{
ObjectMeta: api.ObjectMeta{
Name: name,
Namespace: api.NamespaceDefault,
},
Spec: api.ReplicationControllerSpec{
Replicas: 3,
Selector: map[string]string{"k1": "v1"}},
},
},
}
}
func TestReplicationControllerStop(t *testing.T) {
// test data
toBeReaped := &api.ReplicationController{
ObjectMeta: api.ObjectMeta{
Name: name,
Namespace: api.NamespaceDefault,
},
Spec: api.ReplicationControllerSpec{
Replicas: 5,
Selector: map[string]string{"k1": "v1"}},
}
reaped := &api.ReplicationController{
ObjectMeta: api.ObjectMeta{
Name: name,
Namespace: api.NamespaceDefault,
}, },
Spec: api.ReplicationControllerSpec{ Spec: api.ReplicationControllerSpec{
Replicas: 0, Replicas: 0,
Selector: map[string]string{"k1": "v1"}}, Selector: map[string]string{"k1": "v1"}},
} },
noOverlapping := &api.ReplicationController{ {
ObjectMeta: api.ObjectMeta{ ObjectMeta: api.ObjectMeta{
Name: "baz", Name: name,
Namespace: api.NamespaceDefault, Namespace: ns,
}, },
Spec: api.ReplicationControllerSpec{ Spec: api.ReplicationControllerSpec{
Replicas: 3, Replicas: 0,
Selector: map[string]string{"k3": "v3"}},
}
overlapping := &api.ReplicationController{
ObjectMeta: api.ObjectMeta{
Name: "baz",
Namespace: api.NamespaceDefault,
},
Spec: api.ReplicationControllerSpec{
Replicas: 2,
Selector: map[string]string{"k1": "v1", "k2": "v2"}}, Selector: map[string]string{"k1": "v1", "k2": "v2"}},
} },
},
},
},
// tests
tests := []struct {
Name string
Fns []testclient.ReactionFunc
StopError error
ExpectedActions []testclient.Action
}{
{
Name: "OnlyOneRC",
Fns: []testclient.ReactionFunc{
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, toBeReaped, nil
},
// LIST rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, &api.ReplicationControllerList{
Items: []api.ReplicationController{*toBeReaped}}, nil
},
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, toBeReaped, nil
},
// UPDATE rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
// DELETE rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
},
StopError: nil,
ExpectedActions: []testclient.Action{
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewListAction("replicationcontrollers", api.NamespaceDefault, api.ListOptions{}),
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewUpdateAction("replicationcontrollers", api.NamespaceDefault, reaped),
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewDeleteAction("replicationcontrollers", api.NamespaceDefault, name),
},
},
{
Name: "RCWithNoPods",
Fns: []testclient.ReactionFunc{
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
// LIST rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, &api.ReplicationControllerList{
Items: []api.ReplicationController{*reaped}}, nil
},
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
// DELETE rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
},
StopError: nil,
ExpectedActions: []testclient.Action{
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewListAction("replicationcontrollers", api.NamespaceDefault, api.ListOptions{}),
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewDeleteAction("replicationcontrollers", api.NamespaceDefault, name),
},
},
{
Name: "NoOverlapping",
Fns: []testclient.ReactionFunc{
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, toBeReaped, nil
},
// LIST rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, &api.ReplicationControllerList{
Items: []api.ReplicationController{*toBeReaped, *noOverlapping}}, nil
},
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, toBeReaped, nil
},
// UPDATE rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
// DELETE rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
},
StopError: nil,
ExpectedActions: []testclient.Action{
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewListAction("replicationcontrollers", api.NamespaceDefault, api.ListOptions{}),
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewUpdateAction("replicationcontrollers", api.NamespaceDefault, reaped),
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewDeleteAction("replicationcontrollers", api.NamespaceDefault, name),
},
},
{
Name: "OverlappingError",
Fns: []testclient.ReactionFunc{
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, toBeReaped, nil
},
// LIST rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, &api.ReplicationControllerList{
Items: []api.ReplicationController{*toBeReaped, *overlapping}}, nil
},
},
StopError: fmt.Errorf("Detected overlapping controllers for rc foo: baz, please manage deletion individually with --cascade=false."),
ExpectedActions: []testclient.Action{
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewListAction("replicationcontrollers", api.NamespaceDefault, api.ListOptions{}),
},
},
{
Name: "OverlappingButSafeDelete",
Fns: []testclient.ReactionFunc{
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, &overlappingButSafe().Items[0], nil
},
// LIST rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, overlappingButSafe(), nil
},
},
StopError: fmt.Errorf("Detected overlapping controllers for rc foo: baz,zaz, please manage deletion individually with --cascade=false."), StopError: fmt.Errorf("Detected overlapping controllers for rc foo: baz,zaz, please manage deletion individually with --cascade=false."),
ExpectedActions: []testclient.Action{ ExpectedActions: []string{"get", "list"},
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewListAction("replicationcontrollers", api.NamespaceDefault, api.ListOptions{}),
},
}, },
{ {
Name: "TwoExactMatchRCs", Name: "TwoExactMatchRCs",
Fns: []testclient.ReactionFunc{ Objs: []runtime.Object{
// GET rc
func(action testclient.Action) (handled bool, ret runtime.Object, err error) { &api.ReplicationController{ // GET
return true, toBeReaped, nil ObjectMeta: api.ObjectMeta{
Name: name,
Namespace: ns,
}, },
// LIST rc Spec: api.ReplicationControllerSpec{
func(action testclient.Action) (handled bool, ret runtime.Object, err error) { Replicas: 0,
return true, exactMatches(), nil Selector: map[string]string{"k1": "v1"}},
}, },
// GET rc &api.ReplicationControllerList{ // LIST
func(action testclient.Action) (handled bool, ret runtime.Object, err error) { Items: []api.ReplicationController{
return true, reaped, nil {
ObjectMeta: api.ObjectMeta{
Name: "zaz",
Namespace: ns,
},
Spec: api.ReplicationControllerSpec{
Replicas: 0,
Selector: map[string]string{"k1": "v1"}},
},
{
ObjectMeta: api.ObjectMeta{
Name: name,
Namespace: ns,
},
Spec: api.ReplicationControllerSpec{
Replicas: 0,
Selector: map[string]string{"k1": "v1"}},
}, },
}, },
},
},
StopError: nil, StopError: nil,
ExpectedActions: []testclient.Action{ ExpectedActions: []string{"get", "list", "delete"},
testclient.NewGetAction("replicationcontrollers", api.NamespaceDefault, name),
testclient.NewListAction("replicationcontrollers", api.NamespaceDefault, api.ListOptions{}),
testclient.NewDeleteAction("replicationcontrollers", api.NamespaceDefault, name),
},
}, },
} }
for _, test := range tests { for _, test := range tests {
toBeReaped.Spec.Replicas = 5 fake := testclient.NewSimpleFake(test.Objs...)
fake := &testclient.Fake{}
for i, reaction := range test.Fns {
fake.AddReactor(test.ExpectedActions[i].GetVerb(), test.ExpectedActions[i].GetResource(), reaction)
}
reaper := ReplicationControllerReaper{fake, time.Millisecond, time.Millisecond} reaper := ReplicationControllerReaper{fake, time.Millisecond, time.Millisecond}
err := reaper.Stop(api.NamespaceDefault, name, 0, nil) err := reaper.Stop(ns, name, 0, nil)
if !reflect.DeepEqual(err, test.StopError) { if !reflect.DeepEqual(err, test.StopError) {
t.Errorf("%s: unexpected error: %v", test.Name, err) t.Errorf("%s unexpected error: %v", test.Name, err)
continue continue
} }
actions := fake.Actions() actions := fake.Actions()
if len(test.ExpectedActions) != len(actions) { if len(actions) != len(test.ExpectedActions) {
t.Errorf("%s: unexpected actions:\n%v\nexpected\n%v\n", test.Name, actions, test.ExpectedActions) t.Errorf("%s unexpected actions: %v, expected %d actions got %d", test.Name, actions, len(test.ExpectedActions), len(actions))
continue
} }
for i, action := range actions { for i, verb := range test.ExpectedActions {
testAction := test.ExpectedActions[i] if actions[i].GetResource() != "replicationcontrollers" {
if !testAction.Matches(action.GetVerb(), action.GetResource()) { t.Errorf("%s unexpected action: %+v, expected %s-replicationController", test.Name, actions[i], verb)
t.Errorf("%s: unexpected action: %#v; expected %v", test.Name, action, testAction) }
if actions[i].GetVerb() != verb {
t.Errorf("%s unexpected action: %+v, expected %s-replicationController", test.Name, actions[i], verb)
} }
} }
} }
} }
func TestJobStop(t *testing.T) { func TestJobStop(t *testing.T) {
name := "foo"
ns := "default"
zero := 0 zero := 0
one := 1 tests := []struct {
toBeReaped := &extensions.Job{ Name string
Objs []runtime.Object
StopError error
ExpectedActions []string
}{
{
Name: "OnlyOneJob",
Objs: []runtime.Object{
&extensions.Job{ // GET
ObjectMeta: api.ObjectMeta{ ObjectMeta: api.ObjectMeta{
Name: name, Name: name,
Namespace: api.NamespaceDefault, Namespace: ns,
},
Spec: extensions.JobSpec{
Parallelism: &one,
Selector: &extensions.LabelSelector{
MatchLabels: map[string]string{"k1": "v1"},
},
},
}
reaped := &extensions.Job{
ObjectMeta: api.ObjectMeta{
Name: name,
Namespace: api.NamespaceDefault,
}, },
Spec: extensions.JobSpec{ Spec: extensions.JobSpec{
Parallelism: &zero, Parallelism: &zero,
@ -370,168 +288,98 @@ func TestJobStop(t *testing.T) {
MatchLabels: map[string]string{"k1": "v1"}, MatchLabels: map[string]string{"k1": "v1"},
}, },
}, },
} },
&extensions.JobList{ // LIST
tests := []struct { Items: []extensions.Job{
Name string
Fns []testclient.ReactionFunc
StopError error
ExpectedActions []testclient.Action
}{
{ {
Name: "OnlyOneJob", ObjectMeta: api.ObjectMeta{
Fns: []testclient.ReactionFunc{ Name: name,
// GET job Namespace: ns,
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, toBeReaped, nil
}, },
// GET job Spec: extensions.JobSpec{
func(action testclient.Action) (handled bool, ret runtime.Object, err error) { Parallelism: &zero,
return true, toBeReaped, nil Selector: &extensions.LabelSelector{
MatchLabels: map[string]string{"k1": "v1"},
}, },
// UPDATE job
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
*toBeReaped.Spec.Parallelism = 0
return true, toBeReaped, nil
}, },
// GET job
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, toBeReaped, nil
}, },
// GET job
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, toBeReaped, nil
}, },
// LIST pods
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, &api.PodList{}, nil
},
// DELETE job
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, toBeReaped, nil
}, },
}, },
StopError: nil, StopError: nil,
ExpectedActions: []testclient.Action{ ExpectedActions: []string{"get:jobs", "get:jobs", "update:jobs",
testclient.NewGetAction("jobs", api.NamespaceDefault, name), "get:jobs", "get:jobs", "list:pods", "delete:jobs"},
testclient.NewGetAction("jobs", api.NamespaceDefault, name),
testclient.NewUpdateAction("jobs", api.NamespaceDefault, toBeReaped),
testclient.NewGetAction("jobs", api.NamespaceDefault, name),
testclient.NewGetAction("jobs", api.NamespaceDefault, name),
testclient.NewListAction("pods", api.NamespaceDefault, api.ListOptions{}),
testclient.NewDeleteAction("jobs", api.NamespaceDefault, name),
},
}, },
{ {
Name: "JobWithDeadPods", Name: "JobWithDeadPods",
Fns: []testclient.ReactionFunc{ Objs: []runtime.Object{
// GET job &extensions.Job{ // GET
func(action testclient.Action) (handled bool, ret runtime.Object, err error) { ObjectMeta: api.ObjectMeta{
return true, toBeReaped, nil Name: name,
Namespace: ns,
}, },
// GET job Spec: extensions.JobSpec{
func(action testclient.Action) (handled bool, ret runtime.Object, err error) { Parallelism: &zero,
return true, toBeReaped, nil Selector: &extensions.LabelSelector{
MatchLabels: map[string]string{"k1": "v1"},
}, },
// UPDATE job
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
*toBeReaped.Spec.Parallelism = 0
return true, toBeReaped, nil
}, },
// GET job
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, toBeReaped, nil
}, },
// GET job &extensions.JobList{ // LIST
func(action testclient.Action) (handled bool, ret runtime.Object, err error) { Items: []extensions.Job{
return true, toBeReaped, nil {
ObjectMeta: api.ObjectMeta{
Name: name,
Namespace: ns,
}, },
// LIST pods Spec: extensions.JobSpec{
func(action testclient.Action) (handled bool, ret runtime.Object, err error) { Parallelism: &zero,
return true, &api.PodList{ Selector: &extensions.LabelSelector{
MatchLabels: map[string]string{"k1": "v1"},
},
},
},
},
},
&api.PodList{ // LIST
Items: []api.Pod{ Items: []api.Pod{
{ {
ObjectMeta: api.ObjectMeta{ ObjectMeta: api.ObjectMeta{
Name: "pod1", Name: "pod1",
Namespace: api.NamespaceDefault, Namespace: ns,
Labels: map[string]string{"k1": "v1"}, Labels: map[string]string{"k1": "v1"},
}, },
}, },
}, },
}, nil
},
// DELETE pod
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, nil, nil
},
// DELETE job
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, toBeReaped, nil
}, },
}, },
StopError: nil, StopError: nil,
ExpectedActions: []testclient.Action{ ExpectedActions: []string{"get:jobs", "get:jobs", "update:jobs",
testclient.NewGetAction("jobs", api.NamespaceDefault, name), "get:jobs", "get:jobs", "list:pods", "delete:pods", "delete:jobs"},
testclient.NewGetAction("jobs", api.NamespaceDefault, name),
testclient.NewUpdateAction("jobs", api.NamespaceDefault, toBeReaped),
testclient.NewGetAction("jobs", api.NamespaceDefault, name),
testclient.NewGetAction("jobs", api.NamespaceDefault, name),
testclient.NewListAction("pods", api.NamespaceDefault, api.ListOptions{}),
testclient.NewDeleteAction("pods", api.NamespaceDefault, name),
testclient.NewDeleteAction("jobs", api.NamespaceDefault, name),
},
},
{
Name: "JobWithNoParallelism",
Fns: []testclient.ReactionFunc{
// GET job
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
// GET job
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
// LIST pods
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, &api.PodList{}, nil
},
// DELETE job
func(action testclient.Action) (handled bool, ret runtime.Object, err error) {
return true, reaped, nil
},
},
StopError: nil,
ExpectedActions: []testclient.Action{
testclient.NewGetAction("jobs", api.NamespaceDefault, name),
testclient.NewGetAction("jobs", api.NamespaceDefault, name),
testclient.NewListAction("pods", api.NamespaceDefault, api.ListOptions{}),
testclient.NewDeleteAction("jobs", api.NamespaceDefault, name),
},
}, },
} }
for _, test := range tests { for _, test := range tests {
*toBeReaped.Spec.Parallelism = one fake := testclient.NewSimpleFake(test.Objs...)
fake := &testclient.Fake{}
for i, reaction := range test.Fns {
fake.AddReactor(test.ExpectedActions[i].GetVerb(), test.ExpectedActions[i].GetResource(), reaction)
}
reaper := JobReaper{fake, time.Millisecond, time.Millisecond} reaper := JobReaper{fake, time.Millisecond, time.Millisecond}
err := reaper.Stop(api.NamespaceDefault, name, 0, nil) err := reaper.Stop(ns, name, 0, nil)
if !reflect.DeepEqual(err, test.StopError) { if !reflect.DeepEqual(err, test.StopError) {
t.Errorf("%s unexpected error: %v", test.Name, err) t.Errorf("%s unexpected error: %v", test.Name, err)
continue continue
} }
actions := fake.Actions() actions := fake.Actions()
if len(test.ExpectedActions) != len(actions) { if len(actions) != len(test.ExpectedActions) {
t.Errorf("%s: unexpected actions:\n%v\nexpected\n%v\n", test.Name, actions, test.ExpectedActions) t.Errorf("%s unexpected actions: %v, expected %d actions got %d", test.Name, actions, len(test.ExpectedActions), len(actions))
continue
} }
for i, action := range actions { for i, expAction := range test.ExpectedActions {
testAction := test.ExpectedActions[i] action := strings.Split(expAction, ":")
if !testAction.Matches(action.GetVerb(), action.GetResource()) { if actions[i].GetVerb() != action[0] {
t.Errorf("%s: unexpected action: %#v; expected %v", test.Name, action, testAction) t.Errorf("%s unexpected verb: %+v, expected %s", test.Name, actions[i], expAction)
}
if actions[i].GetResource() != action[1] {
t.Errorf("%s unexpected resource: %+v, expected %s", test.Name, actions[i], expAction)
} }
} }
} }