aboutsummaryrefslogtreecommitdiffstats
path: root/vendor/github.com/ClickHouse/ch-go/proto/profile_test.go
blob: 441277eaf1c65a621752aeb2cce1a61f9fc31b53 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
package proto

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestProfile_EncodeAware(t *testing.T) {
	p := Profile{
		Rows:                      1234,
		Blocks:                    235123,
		Bytes:                     424,
		AppliedLimit:              true,
		RowsBeforeLimit:           2341,
		CalculatedRowsBeforeLimit: false,
	}
	var b Buffer
	p.EncodeAware(&b, Version)
	Gold(t, &p)

	t.Run("Decode", func(t *testing.T) {
		buf := skipCode(t, b.Buf, int(ServerCodeProfile))
		var dec Profile
		requireDecode(t, buf, aware(&dec))
		require.Equal(t, p, dec)
		requireNoShortRead(t, buf, aware(&dec))
	})
}