ebpf_exporter

1
package decoder
2

3
import (
4
	"bytes"
5
	"testing"
6

7
	"github.com/cloudflare/ebpf_exporter/v2/config"
8
)
9

10
func TestInetIpDecoder(t *testing.T) {
11
	cases := []struct {
12
		in  []byte
13
		out []byte
14
	}{
15
		{
16
			in:  []byte{0x01, 0x02, 0x03, 0x04},
17
			out: []byte("1.2.3.4"),
18
		},
19
		{
20
			in: []byte{
21
				0x24, 0x00,
22
				0xcb, 0x00,
23
				0x00, 0x04,
24
				0x10, 0x24,
25
				0x00, 0x00,
26
				0x00, 0x00,
27
				0xa2, 0x9e,
28
				0xfd, 0x8f,
29
			},
30
			out: []byte("2400:cb00:4:1024::a29e:fd8f"),
31
		},
32
	}
33

34
	for _, c := range cases {
35
		d := &InetIP{}
36

37
		out, err := d.Decode(c.in, config.Decoder{})
38
		if err != nil {
39
			t.Errorf("Error decoding %#v to %#v: %s", c.in, c.out, err)
40
		}
41

42
		if !bytes.Equal(out, c.out) {
43
			t.Errorf("Expected %s, got %s", c.out, out)
44
		}
45
	}
46
}
47