-
Notifications
You must be signed in to change notification settings - Fork 0
/
ping_test.go
55 lines (41 loc) · 1.53 KB
/
ping_test.go
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
package mqtt
import (
"bytes"
"testing"
"github.com/srishina/mqtt.go/internal/packettype"
"github.com/stretchr/testify/require"
)
func TestCodecPingReqPacket(t *testing.T) {
encoded := []byte{0xC0, 0x00}
reader := bytes.NewBuffer(encoded)
byte0, remainingLength, err := readFixedHeader(reader)
require.NoError(t, err, "Decoding PINGREQ fixed header returned error")
require.Equal(t, packettype.PINGREQ, packettype.PacketType(byte0>>4))
require.Equal(t, uint32(0x00), remainingLength)
p := pingReq{}
err = p.decode(reader, 0)
require.NoError(t, err, "PingReq.decode returned an error")
var buf bytes.Buffer
err = p.encode(&buf)
require.NoError(t, err, "PingReq.encode returned an error")
if !bytes.Equal(encoded, buf.Bytes()) {
t.Errorf("PingReq.encode did not return expected bytes %v but %v ", encoded, buf.Bytes())
}
}
func TestCodecPingRespPacket(t *testing.T) {
encoded := []byte{0xD0, 0x00}
reader := bytes.NewBuffer(encoded)
byte0, remainingLength, err := readFixedHeader(reader)
require.NoError(t, err, "Decoding PINGRESP fixed header returned error")
require.Equal(t, packettype.PINGRESP, packettype.PacketType(byte0>>4))
require.Equal(t, uint32(0x00), remainingLength)
p := pingResp{}
err = p.decode(reader, 0)
require.NoError(t, err, "PingResp.decode returned an error")
var buf bytes.Buffer
err = p.encode(&buf)
require.NoError(t, err, "PingResp.encode returned an error")
if !bytes.Equal(encoded, buf.Bytes()) {
t.Errorf("PingReq.encode did not return expected bytes %v but %v ", encoded, buf.Bytes())
}
}