From c686c3c34129fb2456b8c4fcf69339c1acebfbd7 Mon Sep 17 00:00:00 2001
From: Jeff Sutherland <jeffdsuth@gmail.com>
Date: Tue, 11 Dec 2018 11:40:30 +0700
Subject: [PATCH] Ensure submit prints on success

---
 cmd/submit_test.go | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/cmd/submit_test.go b/cmd/submit_test.go
index 06e7caf55..2dae04117 100644
--- a/cmd/submit_test.go
+++ b/cmd/submit_test.go
@@ -145,6 +145,7 @@ func TestSubmitFiles(t *testing.T) {
 	co := newCapturedOutput()
 	co.override()
 	defer co.reset()
+	Err = &bytes.Buffer{}
 
 	// The fake endpoint will populate this when it receives the call from the command.
 	submittedFiles := map[string]string{}
@@ -186,14 +187,16 @@ func TestSubmitFiles(t *testing.T) {
 	files := []string{
 		file1, file2, readme,
 	}
-	err = runSubmit(cfg, pflag.NewFlagSet("fake", pflag.PanicOnError), files)
-	assert.NoError(t, err)
 
-	assert.Equal(t, 3, len(submittedFiles))
+	err = runSubmit(cfg, pflag.NewFlagSet("fake", pflag.PanicOnError), files)
 
-	assert.Equal(t, "This is file 1.", submittedFiles["file-1.txt"])
-	assert.Equal(t, "This is file 2.", submittedFiles["subdir/file-2.txt"])
-	assert.Equal(t, "This is the readme.", submittedFiles["README.md"])
+	if assert.NoError(t, err) {
+		assert.Equal(t, 3, len(submittedFiles))
+		assert.Equal(t, "This is file 1.", submittedFiles["file-1.txt"])
+		assert.Equal(t, "This is file 2.", submittedFiles["subdir/file-2.txt"])
+		assert.Equal(t, "This is the readme.", submittedFiles["README.md"])
+		assert.Regexp(t, "submitted successfully", Err)
+	}
 }
 
 func TestLegacyMetadataMigration(t *testing.T) {