706caa44bf
- Adds binary serialization into hex/base64 - Default codec is base64 - Codec can be changed via `--binary-codec` CLI option
93 lines
2.0 KiB
Go
93 lines
2.0 KiB
Go
package client
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestPostProcess(t *testing.T) {
|
|
t.Run("large numbers", func(t *testing.T) {
|
|
result := Result{
|
|
Columns: []string{"value"},
|
|
Rows: []Row{
|
|
{int(1234)},
|
|
{int64(9223372036854775807)},
|
|
{int64(-9223372036854775808)},
|
|
{float64(9223372036854775808.9223372036854775808)},
|
|
{float64(999999999999999.9)},
|
|
},
|
|
}
|
|
|
|
result.PostProcess()
|
|
|
|
assert.Equal(t, 1234, result.Rows[0][0])
|
|
assert.Equal(t, "9223372036854775807", result.Rows[1][0])
|
|
assert.Equal(t, "-9223372036854775808", result.Rows[2][0])
|
|
assert.Equal(t, "9.223372036854776e+18", result.Rows[3][0])
|
|
assert.Equal(t, "9.999999999999999e+14", result.Rows[4][0])
|
|
})
|
|
|
|
t.Run("binary encoding", func(t *testing.T) {
|
|
result := Result{
|
|
Columns: []string{"data"},
|
|
Rows: []Row{
|
|
{"text value"},
|
|
{"text with symbols !@#$%"},
|
|
{string([]byte{10, 11, 12, 13})},
|
|
},
|
|
}
|
|
|
|
result.PostProcess()
|
|
|
|
assert.Equal(t, "text value", result.Rows[0][0])
|
|
assert.Equal(t, "text with symbols !@#$%", result.Rows[1][0])
|
|
assert.Equal(t, "CgsMDQ==", result.Rows[2][0])
|
|
})
|
|
}
|
|
|
|
func TestCSV(t *testing.T) {
|
|
result := Result{
|
|
Columns: []string{"id", "name", "email"},
|
|
Rows: []Row{
|
|
{1, "John", "john@example.com"},
|
|
{2, "Bob", "bob@example.com"},
|
|
},
|
|
}
|
|
|
|
expected := "id,name,email\n1,John,john@example.com\n2,Bob,bob@example.com\n"
|
|
output := string(result.CSV())
|
|
|
|
assert.Equal(t, expected, output)
|
|
}
|
|
|
|
func TestJSON(t *testing.T) {
|
|
result := Result{
|
|
Columns: []string{"id", "name", "email"},
|
|
Rows: []Row{
|
|
{1, "John", "john@example.com"},
|
|
{2, "Bob", "bob@example.com"},
|
|
},
|
|
}
|
|
|
|
obj := []struct {
|
|
Id int
|
|
Name string
|
|
Email string
|
|
}{}
|
|
|
|
expected := []struct {
|
|
Id int
|
|
Name string
|
|
Email string
|
|
}{
|
|
{1, "John", "john@example.com"},
|
|
{2, "Bob", "bob@example.com"},
|
|
}
|
|
|
|
assert.NoError(t, json.Unmarshal(result.JSON(), &obj))
|
|
assert.Equal(t, 2, len(obj))
|
|
assert.Equal(t, expected, obj)
|
|
}
|