Adds PrintCsv() function and unit tests to bluemix/terminal/table (#352)

* added PrintCsv() function and unit tests to bluemix/terminal/table

* Added handled errors to function PrintCsv

* Added unit test for empty table in function PrintCsv

* Added translations for error message and returned it

---------

Co-authored-by: Aerex <[email protected]>

Author: edsonarios

bluemix/terminal/table.go

@@ -1,10 +1,12 @@
 package terminal
 
 import (
+	"encoding/csv"
 	"fmt"
 	"io"
 	"strings"
 
+	. "github.com/IBM-Cloud/ibm-cloud-cli-sdk/i18n"
 	"github.com/mattn/go-runewidth"
 )
 
@@ -18,6 +20,7 @@ type Table interface {
 	Add(row ...string)
 	Print()
 	PrintJson()
+	PrintCsv() error
 }
 
 type PrintableTable struct {
@@ -157,3 +160,16 @@ func (t *PrintableTable) PrintJson() {
 	// mimic behavior of Print()
 	t.rows = [][]string{}
 }
+
+func (t *PrintableTable) PrintCsv() error {
+	csvwriter := csv.NewWriter(t.writer)
+	err := csvwriter.Write(t.headers)
+	if err != nil {
+		return fmt.Errorf(T("Failed, header could not convert to csv format"), err.Error())
+	}
+	err = csvwriter.WriteAll(t.rows)
+	if err != nil {
+		return fmt.Errorf(T("Failed, rows could not convert to csv format"), err.Error())
+	}
+	return nil
+}

bluemix/terminal/table_test.go

@@ -2,6 +2,7 @@ package terminal_test
 
 import (
 	"bytes"
+	"strings"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -92,3 +93,47 @@ func TestNotEnoughRowEntiresJson(t *testing.T) {
 	assert.Contains(t, buf.String(), "\"column_1\": \"row1\"")
 	assert.Contains(t, buf.String(), "\"column_1\": \"\"")
 }
+
+func TestPrintCsvSimple(t *testing.T) {
+	buf := bytes.Buffer{}
+	testTable := NewTable(&buf, []string{"col1", "col2"})
+	testTable.Add("row1-col1", "row1-col2")
+	testTable.Add("row2-col1", "row2-col2")
+	err := testTable.PrintCsv()
+	assert.Equal(t, err, nil)
+	assert.Contains(t, buf.String(), "col1,col2")
+	assert.Contains(t, buf.String(), "row1-col1,row1-col2")
+	assert.Contains(t, buf.String(), "row2-col1,row2-col2")
+}
+
+func TestNotEnoughColPrintCsv(t *testing.T) {
+	buf := bytes.Buffer{}
+	testTable := NewTable(&buf, []string{"", "col2"})
+	testTable.Add("row1-col1", "row1-col2")
+	testTable.Add("row2-col1", "row2-col2")
+	err := testTable.PrintCsv()
+	assert.Equal(t, err, nil)
+	assert.Contains(t, buf.String(), ",col2")
+	assert.Contains(t, buf.String(), "row1-col1,row1-col2")
+	assert.Contains(t, buf.String(), "row2-col1,row2-col2")
+}
+
+func TestNotEnoughRowPrintCsv(t *testing.T) {
+	buf := bytes.Buffer{}
+	testTable := NewTable(&buf, []string{"col1", "col2"})
+	testTable.Add("row1-col1", "row1-col2")
+	testTable.Add("row2-col1", "")
+	err := testTable.PrintCsv()
+	assert.Equal(t, err, nil)
+	assert.Contains(t, buf.String(), "col1,col2")
+	assert.Contains(t, buf.String(), "row1-col1,row1-col2")
+	assert.Contains(t, buf.String(), "row2-col1,")
+}
+
+func TestEmptyTable(t *testing.T) {
+	buf := bytes.Buffer{}
+	testTable := NewTable(&buf, []string{})
+	err := testTable.PrintCsv()
+	assert.Equal(t, err, nil)
+	assert.Equal(t, len(strings.TrimSpace(buf.String())), 0)
+}