From dca3db0a509565f291e182472be7ade5d94d64c0 Mon Sep 17 00:00:00 2001
From: Wojciech Tyczynski <wojtekt@google.com>
Date: Thu, 12 Mar 2015 17:10:28 +0100
Subject: [PATCH] Speedup pkg/kubelet/runonce_test.go

---
 pkg/kubelet/runonce.go      | 10 +++++-----
 pkg/kubelet/runonce_test.go |  3 ++-
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/pkg/kubelet/runonce.go b/pkg/kubelet/runonce.go
index 6389b36f4a2..68b284d52bd 100644
--- a/pkg/kubelet/runonce.go
+++ b/pkg/kubelet/runonce.go
@@ -42,7 +42,7 @@ func (kl *Kubelet) RunOnce(updates <-chan PodUpdate) ([]RunPodResult, error) {
 	select {
 	case u := <-updates:
 		glog.Infof("processing manifest with %d pods", len(u.Pods))
-		result, err := kl.runOnce(u.Pods)
+		result, err := kl.runOnce(u.Pods, RunOnceRetryDelay)
 		glog.Infof("finished processing %d pods", len(u.Pods))
 		return result, err
 	case <-time.After(RunOnceManifestDelay):
@@ -51,7 +51,7 @@ func (kl *Kubelet) RunOnce(updates <-chan PodUpdate) ([]RunPodResult, error) {
 }
 
 // runOnce runs a given set of pods and returns their status.
-func (kl *Kubelet) runOnce(pods []api.BoundPod) (results []RunPodResult, err error) {
+func (kl *Kubelet) runOnce(pods []api.BoundPod, retryDelay time.Duration) (results []RunPodResult, err error) {
 	if kl.dockerPuller == nil {
 		kl.dockerPuller = dockertools.NewDockerPuller(kl.dockerClient, kl.pullQPS, kl.pullBurst)
 	}
@@ -61,7 +61,7 @@ func (kl *Kubelet) runOnce(pods []api.BoundPod) (results []RunPodResult, err err
 	for i := range pods {
 		pod := pods[i] // Make a copy
 		go func() {
-			err := kl.runPod(pod)
+			err := kl.runPod(pod, retryDelay)
 			ch <- RunPodResult{&pod, err}
 		}()
 	}
@@ -87,8 +87,8 @@ func (kl *Kubelet) runOnce(pods []api.BoundPod) (results []RunPodResult, err err
 }
 
 // runPod runs a single pod and wait until all containers are running.
-func (kl *Kubelet) runPod(pod api.BoundPod) error {
-	delay := RunOnceRetryDelay
+func (kl *Kubelet) runPod(pod api.BoundPod, retryDelay time.Duration) error {
+	delay := retryDelay
 	retry := 0
 	for {
 		dockerContainers, err := dockertools.GetKubeletDockerContainers(kl.dockerClient, false)
diff --git a/pkg/kubelet/runonce_test.go b/pkg/kubelet/runonce_test.go
index f20a6073096..4669a9e1764 100644
--- a/pkg/kubelet/runonce_test.go
+++ b/pkg/kubelet/runonce_test.go
@@ -20,6 +20,7 @@ import (
 	"fmt"
 	"strconv"
 	"testing"
+	"time"
 
 	"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
 	"github.com/GoogleCloudPlatform/kubernetes/pkg/client/record"
@@ -140,7 +141,7 @@ func TestRunOnce(t *testing.T) {
 				},
 			},
 		},
-	})
+	}, time.Millisecond)
 	if err != nil {
 		t.Errorf("unexpected error: %v", err)
 	}