Function TestJSUPWriter

api/queryio/jsup_test.go:16–37

(t *testing.T)

)

func TestJSUPWriter(t *testing.T) {
	const record = `{x:1}`
	const expected = `
{"type":"QueryChannelSet","value":{"channel":"main"}}
{"type":{"kind":"record","id":31,"fields":[{"name":"x","type":{"kind":"primitive","name":"int64"},"opt":false}]},"value":["1"]}
{"type":"QueryChannelEnd","value":{"channel":"main"}}
{"type":"QueryError","value":{"error":"test.err"}}
`
	var buf bytes.Buffer
	w := queryio.NewJSUPWriter(&buf)
	err := w.WriteControl(api.QueryChannelSet{Channel: "main"})
	require.NoError(t, err)
	sctx := super.NewContext()
	vals := sbuf.NewArray([]super.Value{sup.MustParseValue(sctx, record)})
	err = w.Push(sbuf.Dematerialize(sctx, vals))
	require.NoError(t, err)
	err = w.WriteControl(api.QueryChannelEnd{Channel: "main"})
	require.NoError(t, err)
	err = w.WriteControl(api.QueryError{Error: "test.err"})
	require.NoError(t, err)
	assert.Equal(t, expected, "\n"+buf.String())
}