mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-21 07:01:41 +03:00
77 lines
2.0 KiB
Go
77 lines
2.0 KiB
Go
package cli_test
|
|
|
|
import (
|
|
"context"
|
|
"encoding/csv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/neilotoole/sq/cli/testrun"
|
|
"github.com/neilotoole/sq/libsq/source"
|
|
"github.com/neilotoole/sq/testh"
|
|
"github.com/neilotoole/sq/testh/sakila"
|
|
)
|
|
|
|
func TestCmdPing(t *testing.T) {
|
|
t.Parallel()
|
|
ctx := context.Background()
|
|
|
|
err := testrun.New(ctx, t, nil).Exec("ping")
|
|
require.Error(t, err, "no active data source")
|
|
|
|
err = testrun.New(ctx, t, nil).Exec("ping", "invalid_handle")
|
|
require.Error(t, err)
|
|
|
|
err = testrun.New(ctx, t, nil).Exec("ping", "@not_a_handle")
|
|
require.Error(t, err)
|
|
|
|
var tr *testrun.TestRun
|
|
|
|
th := testh.New(t)
|
|
src1, src2 := th.Source(sakila.CSVActor), th.Source(sakila.CSVActorNoHeader)
|
|
|
|
tr = testrun.New(ctx, t, nil).Add(*src1)
|
|
err = tr.Exec("ping", "--csv", src1.Handle)
|
|
require.NoError(t, err)
|
|
checkPingOutputCSV(t, tr, *src1)
|
|
|
|
tr = testrun.New(ctx, t, nil).Add(*src2)
|
|
err = tr.Exec("ping", "--csv", src2.Handle)
|
|
require.NoError(t, err)
|
|
checkPingOutputCSV(t, tr, *src2)
|
|
|
|
tr = testrun.New(ctx, t, nil).Add(*src1, *src2)
|
|
err = tr.Exec("ping", "--csv", src1.Handle, src2.Handle)
|
|
require.NoError(t, err)
|
|
checkPingOutputCSV(t, tr, *src1, *src2)
|
|
}
|
|
|
|
// checkPintOutputCSV reads CSV records from h.out, and verifies
|
|
// that there's an appropriate record for each of srcs.
|
|
func checkPingOutputCSV(t *testing.T, h *testrun.TestRun, srcs ...source.Source) {
|
|
recs, err := csv.NewReader(h.Out).ReadAll()
|
|
require.NoError(t, err)
|
|
require.Equal(t, len(srcs), len(recs))
|
|
|
|
if len(srcs) > 0 {
|
|
require.Equal(t, 3, len(recs[0]), "each ping record should have 3 fields, but got %d fields", len(recs[0]))
|
|
}
|
|
|
|
handles := make(map[string]bool)
|
|
for _, src := range srcs {
|
|
handles[src.Handle] = true
|
|
}
|
|
|
|
for i := 0; i < len(recs); i++ {
|
|
recHandle := recs[i][0]
|
|
require.True(t, handles[recHandle], "should have handle %s in map", recHandle)
|
|
|
|
_, err = time.ParseDuration(recs[i][1])
|
|
require.NoError(t, err, "should be a valid duration value")
|
|
|
|
require.Equal(t, "pong", recs[i][2], "error field should be empty")
|
|
}
|
|
}
|