pgweb/pkg/client/result_test.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)
}