Merge pull request #17 from luchihoratiu/CT-190

(CT-190) Move output to stdout instead of stderr

Author: ciprianbadescu

app/puppet-code/deploy.go

@@ -3,7 +3,6 @@ package app
 import (
 	"context"
 	"fmt"
-	"os"
 
 	httptransport "github.com/go-openapi/runtime/client"
 	"github.com/puppetlabs/pe-sdk-go/app/puppet-code/api/client/operations"
@@ -64,8 +63,6 @@ type DeployArgs struct {
 }
 
 func writeDeployResult(args *DeployArgs, payload []*operations.DeployOKBodyItems0) []byte {
-	fmt.Fprintf(os.Stderr, "Found %d environments.\n", len(payload))
-
 	if args.DryRun {
 		separator := ""
 		environments := ""
@@ -79,5 +76,7 @@ func writeDeployResult(args *DeployArgs, payload []*operations.DeployOKBodyItems
 	if err != nil {
 		log.Error(err.Error())
 	}
-	return resultPayload
+
+	info := []byte(fmt.Sprintf("Found %d environments.\n", len(payload)))
+	return append(info, resultPayload...)
 }

app/puppet-code/deploy_test.go

@@ -3,6 +3,7 @@ package app
 import (
 	"context"
 	"errors"
+	"fmt"
 	"testing"
 
 	"github.com/golang/mock/gomock"
@@ -81,7 +82,8 @@ func TestDeployStatusSucces(t *testing.T) {
 	puppetCode.Client = api
 	res, err := puppetCode.DeployWithErrorDetails(&args)
 	expected, _ := json.MarshalIndent(result.Payload, "", "  ")
-	assert.Equal(expected, res)
+	info := []byte(fmt.Sprintf("Found %d environments.\n", 0))
+	assert.Equal(append(info, expected...), res)
 	assert.Nil(err)
 }