execrpc/message_test.go at main · bep/execrpc · GitHub

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
30
31
32
33
34
35
36
// Copyright 2025 Bjørn Erik Pedersen
// SPDX-License-Identifier: MIT

package execrpc

import (
	"bytes"
	"testing"

	qt "github.com/frankban/quicktest"
)

func TestMessage(t *testing.T) {
	c := qt.New(t)

	m1 := Message{
		Body: []byte("hello"),
		Header: Header{
			ID:      2,
			Version: 3,
			Status:  4,
			Size:    5,
		},
	}

	var b bytes.Buffer

	c.Assert(m1.Header.Write(&b), qt.IsNil)
	c.Assert(m1.Write(&b), qt.IsNil)

	var m2 Message
	c.Assert(m2.Header.Read(&b), qt.IsNil)
	c.Assert(m2.Read(&b), qt.IsNil)

	c.Assert(m2, qt.DeepEquals, m1)
}