mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-24 08:36:43 +03:00
111 lines
3.0 KiB
Go
111 lines
3.0 KiB
Go
//nolint:lll
|
|
package xmlw_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"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/neilotoole/sq/testh/testsrc"
|
|
)
|
|
|
|
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) {
|
|
ctx := context.Background()
|
|
|
|
pr := output.NewPrinting()
|
|
pr.EnableColor(tc.color)
|
|
pr.Compact = !tc.pretty
|
|
|
|
recMeta, recs := testh.RecordsFromTbl(t, sakila.SL3, sakila.TblActor)
|
|
recs = recs[0:tc.numRecs]
|
|
|
|
buf := &bytes.Buffer{}
|
|
|
|
w := xmlw.NewRecordWriter(buf, pr)
|
|
require.NoError(t, w.Open(ctx, recMeta))
|
|
require.NoError(t, w.WriteRecords(ctx, recs))
|
|
require.NoError(t, w.Close(ctx))
|
|
|
|
require.Equal(t, tc.want, buf.String())
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRecordWriter_TblTypes(t *testing.T) {
|
|
ctx := context.Background()
|
|
pr := output.NewPrinting()
|
|
pr.EnableColor(false)
|
|
|
|
recMeta, recs := testh.RecordsFromTbl(t, testsrc.MiscDB, testsrc.TblTypes)
|
|
buf := &bytes.Buffer{}
|
|
|
|
w := xmlw.NewRecordWriter(buf, pr)
|
|
require.NoError(t, w.Open(ctx, recMeta))
|
|
require.NoError(t, w.WriteRecords(ctx, recs))
|
|
require.NoError(t, w.Close(ctx))
|
|
|
|
want, err := os.ReadFile("testdata/tbl_types.xml")
|
|
require.NoError(t, err)
|
|
require.Equal(t, string(want), buf.String())
|
|
}
|