diff options
Diffstat (limited to 'go/internal/command/discover/discover_test.go')
-rw-r--r-- | go/internal/command/discover/discover_test.go | 16 |
1 files changed, 12 insertions, 4 deletions
diff --git a/go/internal/command/discover/discover_test.go b/go/internal/command/discover/discover_test.go index f85add8..40c0d4e 100644 --- a/go/internal/command/discover/discover_test.go +++ b/go/internal/command/discover/discover_test.go @@ -78,10 +78,14 @@ func TestExecute(t *testing.T) { for _, tc := range testCases { t.Run(tc.desc, func(t *testing.T) { - cmd := &Command{Config: &config.Config{GitlabUrl: url}, Args: tc.arguments} buffer := &bytes.Buffer{} + cmd := &Command{ + Config: &config.Config{GitlabUrl: url}, + Args: tc.arguments, + ReadWriter: &readwriter.ReadWriter{Out: buffer}, + } - err := cmd.Execute(&readwriter.ReadWriter{Out: buffer}) + err := cmd.Execute() assert.NoError(t, err) assert.Equal(t, tc.expectedOutput, buffer.String()) @@ -118,10 +122,14 @@ func TestFailingExecute(t *testing.T) { for _, tc := range testCases { t.Run(tc.desc, func(t *testing.T) { - cmd := &Command{Config: &config.Config{GitlabUrl: url}, Args: tc.arguments} buffer := &bytes.Buffer{} + cmd := &Command{ + Config: &config.Config{GitlabUrl: url}, + Args: tc.arguments, + ReadWriter: &readwriter.ReadWriter{Out: buffer}, + } - err := cmd.Execute(&readwriter.ReadWriter{Out: buffer}) + err := cmd.Execute() assert.Empty(t, buffer.String()) assert.EqualError(t, err, tc.expectedError) |