sq/cli/output/xmlw/xmlw_test.go
Neil O'Toole 27ff71516e
Added go test to github action; fixed Windows test issues (#52)
* initial test of testing workflow

* added .gitattributes for crlf issue on windows

* testh was attempting to remove the copied sqlite3 file before it was closed by sqlite3.database.Close()

* fiddling with .gitattributes to get line endings to work
2020-08-07 23:10:41 -06:00

108 lines
2.9 KiB
Go

package xmlw_test
import (
"bytes"
"io/ioutil"
"testing"
"github.com/neilotoole/sq/testh/testsrc"
"github.com/neilotoole/sq/cli/output"
"github.com/neilotoole/sq/cli/output/xmlw"
"github.com/neilotoole/sq/testh"
"github.com/neilotoole/sq/testh/sakila"
"github.com/stretchr/testify/require"
)
func TestRecordWriter_Actor(t *testing.T) {
const (
want0 = `<?xml version="1.0"?>
<records />
`
want3Pretty = `<?xml version="1.0"?>
<records>
<record>
<actor_id>1</actor_id>
<first_name>PENELOPE</first_name>
<last_name>GUINESS</last_name>
<last_update>2020-06-11T02:50:54Z</last_update>
</record>
<record>
<actor_id>2</actor_id>
<first_name>NICK</first_name>
<last_name>WAHLBERG</last_name>
<last_update>2020-06-11T02:50:54Z</last_update>
</record>
<record>
<actor_id>3</actor_id>
<first_name>ED</first_name>
<last_name>CHASE</last_name>
<last_update>2020-06-11T02:50:54Z</last_update>
</record>
</records>
`
want3NoPretty = `<?xml version="1.0"?>
<records>
<record><actor_id>1</actor_id><first_name>PENELOPE</first_name><last_name>GUINESS</last_name><last_update>2020-06-11T02:50:54Z</last_update></record>
<record><actor_id>2</actor_id><first_name>NICK</first_name><last_name>WAHLBERG</last_name><last_update>2020-06-11T02:50:54Z</last_update></record>
<record><actor_id>3</actor_id><first_name>ED</first_name><last_name>CHASE</last_name><last_update>2020-06-11T02:50:54Z</last_update></record>
</records>
`
)
testCases := []struct {
name string
color bool
pretty bool
numRecs int
want string
}{
{name: "actor_0_pretty", color: false, pretty: true, numRecs: 0, want: want0},
{name: "actor_0_no_pretty", color: false, pretty: false, numRecs: 0, want: want0},
{name: "actor_3_pretty", color: false, pretty: true, numRecs: 3, want: want3Pretty},
{name: "actor_3_no_pretty", color: false, pretty: false, numRecs: 3, want: want3NoPretty},
}
for _, tc := range testCases {
tc := tc
t.Run(tc.name, func(t *testing.T) {
fm := output.NewFormatting()
fm.EnableColor(tc.color)
fm.Pretty = tc.pretty
recMeta, recs := testh.RecordsFromTbl(t, sakila.SL3, sakila.TblActor)
recs = recs[0:tc.numRecs]
buf := &bytes.Buffer{}
w := xmlw.NewRecordWriter(buf, fm)
require.NoError(t, w.Open(recMeta))
require.NoError(t, w.WriteRecords(recs))
require.NoError(t, w.Close())
require.Equal(t, tc.want, buf.String())
})
}
}
func TestRecordWriter_TblTypes(t *testing.T) {
fm := output.NewFormatting()
fm.EnableColor(false)
recMeta, recs := testh.RecordsFromTbl(t, testsrc.MiscDB, testsrc.TblTypes)
buf := &bytes.Buffer{}
w := xmlw.NewRecordWriter(buf, fm)
require.NoError(t, w.Open(recMeta))
require.NoError(t, w.WriteRecords(recs))
require.NoError(t, w.Close())
want, err := ioutil.ReadFile("testdata/tbl_types.xml")
require.NoError(t, err)
require.Equal(t, string(want), buf.String())
}