From e2f7387906ded9b495ac7be8e4d1fcca7d645609 Mon Sep 17 00:00:00 2001
From: Daniel Nephin <dnephin@docker.com>
Date: Thu, 8 Sep 2016 15:11:38 -0400
Subject: [PATCH] Fix a test that expects whitespace at the end of the line.

Signed-off-by: Daniel Nephin <dnephin@docker.com>
---
 command/container/hijack.go         |  2 +-
 command/container/tty.go            |  2 +-
 command/formatter/container_test.go | 29 ++++++++++++++---------------
 3 files changed, 16 insertions(+), 17 deletions(-)

diff --git a/command/container/hijack.go b/command/container/hijack.go
index 855a152904..ea429245cf 100644
--- a/command/container/hijack.go
+++ b/command/container/hijack.go
@@ -6,8 +6,8 @@ import (
 	"sync"
 
 	"github.com/Sirupsen/logrus"
-	"github.com/docker/docker/cli/command"
 	"github.com/docker/docker/api/types"
+	"github.com/docker/docker/cli/command"
 	"github.com/docker/docker/pkg/stdcopy"
 	"golang.org/x/net/context"
 )
diff --git a/command/container/tty.go b/command/container/tty.go
index 5360c6b040..edb11592d3 100644
--- a/command/container/tty.go
+++ b/command/container/tty.go
@@ -8,8 +8,8 @@ import (
 	"time"
 
 	"github.com/Sirupsen/logrus"
-	"github.com/docker/docker/cli/command"
 	"github.com/docker/docker/api/types"
+	"github.com/docker/docker/cli/command"
 	"github.com/docker/docker/client"
 	"github.com/docker/docker/pkg/signal"
 	"golang.org/x/net/context"
diff --git a/command/formatter/container_test.go b/command/formatter/container_test.go
index deaa915a89..29b8450db9 100644
--- a/command/formatter/container_test.go
+++ b/command/formatter/container_test.go
@@ -9,6 +9,7 @@ import (
 
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/pkg/stringid"
+	"github.com/docker/docker/pkg/testutil/assert"
 )
 
 func TestContainerPsContext(t *testing.T) {
@@ -232,19 +233,19 @@ containerID2        ubuntu              ""                  24 hours ago
 image: ubuntu
 command: ""
 created_at: %s
-status:
+status: 
 names: foobar_baz
-labels:
-ports:
+labels: 
+ports: 
 
 container_id: containerID2
 image: ubuntu
 command: ""
 created_at: %s
-status:
+status: 
 names: foobar_bar
-labels:
-ports:
+labels: 
+ports: 
 
 `, expectedTime, expectedTime),
 		},
@@ -259,20 +260,20 @@ ports:
 image: ubuntu
 command: ""
 created_at: %s
-status:
+status: 
 names: foobar_baz
-labels:
-ports:
+labels: 
+ports: 
 size: 0 B
 
 container_id: containerID2
 image: ubuntu
 command: ""
 created_at: %s
-status:
+status: 
 names: foobar_bar
-labels:
-ports:
+labels: 
+ports: 
 size: 0 B
 
 `, expectedTime, expectedTime),
@@ -316,9 +317,7 @@ size: 0 B
 		context.context.Containers = containers
 		context.context.Write()
 		actual := out.String()
-		if actual != context.expected {
-			t.Fatalf("Expected \n%s, got \n%s", context.expected, actual)
-		}
+		assert.Equal(t, actual, context.expected)
 		// Clean buffer
 		out.Reset()
 	}