mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-29 19:24:26 +03:00
123 lines
3.9 KiB
Go
123 lines
3.9 KiB
Go
package xmlud_test
|
||
|
||
import (
|
||
"bytes"
|
||
"github.com/neilotoole/sq/testh/tutil"
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/assert"
|
||
"github.com/stretchr/testify/require"
|
||
"gopkg.in/yaml.v2"
|
||
|
||
"github.com/neilotoole/sq/cli/config"
|
||
"github.com/neilotoole/sq/drivers/userdriver/xmlud"
|
||
"github.com/neilotoole/sq/testh"
|
||
"github.com/neilotoole/sq/testh/proj"
|
||
"github.com/neilotoole/sq/testh/testsrc"
|
||
)
|
||
|
||
const (
|
||
driverRSS = "rss"
|
||
driverPpl = "ppl"
|
||
)
|
||
|
||
func TestImport_Ppl(t *testing.T) {
|
||
th := testh.New(t)
|
||
|
||
ext := &config.Ext{}
|
||
require.NoError(t, yaml.Unmarshal(proj.ReadFile(testsrc.PathDriverDefPpl), ext))
|
||
require.Equal(t, 1, len(ext.UserDrivers))
|
||
udDef := ext.UserDrivers[0]
|
||
require.Equal(t, driverPpl, udDef.Name)
|
||
require.Equal(t, xmlud.Genre, udDef.Genre)
|
||
|
||
scratchDB, err := th.Databases().OpenScratch(th.Context, "ppl")
|
||
require.NoError(t, err)
|
||
t.Cleanup(func() {
|
||
assert.NoError(t, scratchDB.Close())
|
||
})
|
||
|
||
data := proj.ReadFile("drivers/userdriver/xmlud/testdata/people.xml")
|
||
err = xmlud.Import(th.Context, th.Log, udDef, bytes.NewReader(data), scratchDB)
|
||
require.NoError(t, err)
|
||
|
||
srcMeta, err := scratchDB.SourceMetadata(th.Context)
|
||
require.NoError(t, err)
|
||
require.Equal(t, 2, len(srcMeta.Tables))
|
||
require.Equal(t, "person", srcMeta.Tables[0].Name)
|
||
require.Equal(t, "skill", srcMeta.Tables[1].Name)
|
||
|
||
sink, err := th.QuerySQL(scratchDB.Source(), "SELECT * FROM person")
|
||
require.NoError(t, err)
|
||
require.Equal(t, 3, len(sink.Recs))
|
||
require.Equal(t, "Nikola", tutil.Val(sink.Recs[0][1]))
|
||
for i, rec := range sink.Recs {
|
||
// Verify that the primary id cols are sequential
|
||
require.Equal(t, int64(i+1), tutil.Val(rec[0]))
|
||
}
|
||
|
||
sink, err = th.QuerySQL(scratchDB.Source(), "SELECT * FROM skill")
|
||
require.NoError(t, err)
|
||
require.Equal(t, 6, len(sink.Recs))
|
||
require.Equal(t, "Electrifying", tutil.Val(sink.Recs[0][2]))
|
||
for i, rec := range sink.Recs {
|
||
// Verify that the primary id cols are sequential
|
||
require.Equal(t, int64(i+1), tutil.Val(rec[0]))
|
||
}
|
||
}
|
||
|
||
func TestImport_RSS(t *testing.T) {
|
||
th := testh.New(t)
|
||
|
||
ext := &config.Ext{}
|
||
require.NoError(t, yaml.Unmarshal(proj.ReadFile(testsrc.PathDriverDefRSS), ext))
|
||
require.Equal(t, 1, len(ext.UserDrivers))
|
||
udDef := ext.UserDrivers[0]
|
||
require.Equal(t, driverRSS, udDef.Name)
|
||
require.Equal(t, xmlud.Genre, udDef.Genre)
|
||
|
||
scratchDB, err := th.Databases().OpenScratch(th.Context, "rss")
|
||
require.NoError(t, err)
|
||
t.Cleanup(func() {
|
||
assert.NoError(t, scratchDB.Close())
|
||
})
|
||
|
||
data := proj.ReadFile("drivers/userdriver/xmlud/testdata/nytimes_local.rss.xml")
|
||
err = xmlud.Import(th.Context, th.Log, udDef, bytes.NewReader(data), scratchDB)
|
||
require.NoError(t, err)
|
||
|
||
srcMeta, err := scratchDB.SourceMetadata(th.Context)
|
||
require.NoError(t, err)
|
||
require.Equal(t, 3, len(srcMeta.Tables))
|
||
require.Equal(t, "category", srcMeta.Tables[0].Name)
|
||
require.Equal(t, "channel", srcMeta.Tables[1].Name)
|
||
require.Equal(t, "item", srcMeta.Tables[2].Name)
|
||
|
||
sink, err := th.QuerySQL(scratchDB.Source(), "SELECT * FROM channel")
|
||
require.NoError(t, err)
|
||
require.Equal(t, 1, len(sink.Recs))
|
||
require.Equal(t, "NYT > World", tutil.Val(sink.Recs[0][1]))
|
||
for i, rec := range sink.Recs {
|
||
// Verify that the primary id cols are sequential
|
||
require.Equal(t, int64(i+1), tutil.Val(rec[0]))
|
||
}
|
||
|
||
sink, err = th.QuerySQL(scratchDB.Source(), "SELECT * FROM category")
|
||
require.NoError(t, err)
|
||
require.Equal(t, 251, len(sink.Recs))
|
||
require.EqualValues(t, "Extradition", tutil.Val(sink.Recs[0][2]))
|
||
for i, rec := range sink.Recs {
|
||
// Verify that the primary id cols are sequential
|
||
require.Equal(t, int64(i+1), tutil.Val(rec[0]))
|
||
}
|
||
|
||
sink, err = th.QuerySQL(scratchDB.Source(), "SELECT * FROM item")
|
||
require.NoError(t, err)
|
||
require.Equal(t, 45, len(sink.Recs))
|
||
require.EqualValues(t, "Trilobites: Fishing for Clues to Solve Namibia’s Fairy Circle Mystery", tutil.Val(sink.Recs[17][4]))
|
||
for i, rec := range sink.Recs {
|
||
// Verify that the primary id cols are sequential
|
||
require.Equal(t, int64(i+1), tutil.Val(rec[0]))
|
||
}
|
||
}
|