watcher: handle closed channel

logs/pod_watcher_test.go

@@ -89,8 +89,8 @@ func TestPodWatcherClosedWatcher(t *testing.T) {
 
 	// WatchPods should wait for the logs and return cleanly.
 	err := pw.WatchPods(ctx)
-	require.EqualError(t, err, context.DeadlineExceeded.Error())
+	require.Equal(t, context.DeadlineExceeded, err)
-	require.Equal(t, "[foo] it worked\n", buf.String())
+	assert.Equal(t, "[foo] it worked\n", buf.String())
 }
 
 func TestPodWatcher(t *testing.T) {

logs/watcher.go

@@ -96,10 +96,14 @@ func (w *Watcher) Watch(ctx context.Context) error {
 	ticker := time.NewTicker(time.Second)
 	defer ticker.Stop()
 
+	resultChan := deploymentsWatcher.ResultChan()
 	for {
 		select {
-		// TODO: check if closed.
+		case evt, ok := <-resultChan:
-		case evt := <-deploymentsWatcher.ResultChan():
+			if !ok {
+				resultChan = nil
+				continue
+			}
 			switch evt.Type {
 			case watch.Added, watch.Modified:
 				deployment := evt.Object.(*appsv1.Deployment)

logs/watcher_test.go

@@ -64,6 +64,26 @@ func TestWatcherPodWatcherError(t *testing.T) {
 	assert.EqualError(t, err, wantErr.Error())
 }
 
+func TestWatcherClosedChannel(t *testing.T) {
+	deploymentsWatcher := watch.NewFake()
+	clientset := testclient.NewSimpleClientset(
+		&appsv1.Deployment{ObjectMeta: metav1.ObjectMeta{Name: "mydeployment", Namespace: "default"}},
+	)
+	clientset.PrependWatchReactor("deployments", k8stest.DefaultWatchReactor(deploymentsWatcher, nil))
+
+	var buf bytes.Buffer
+	client := logs.KubernetesClient{Interface: clientset}
+	watcher := logs.NewWatcher("mydeployment", "mycontainer", true, client, nil, &buf)
+	go deploymentsWatcher.Stop()
+
+	ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*500)
+	defer cancel()
+
+	err := watcher.Watch(ctx)
+	require.Equal(t, context.DeadlineExceeded, err)
+	assert.Equal(t, "", buf.String())
+}
+
 func TestWatcherWithPodWatcher(t *testing.T) {
 	ctx, cancel := context.WithCancel(context.Background())