highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-12-12 05:42:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

tlv/truncated: add non-generic encoders for truncated types

Browse Source 
These encoders can be composed to create composite types without
incurring additional allocations that would be required to pass the
truncated types through the generic interface.
This commit is contained in:
Conner Fromknecht
2019-11-04 14:10:43 -08:00
parent 083b574fd8
commit b3b51923dc
2 changed files with 75 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
tlv/truncated_test.go
View File

@@ -60,6 +60,8 @@ func TestSizeTUint16(t *testing.T) {
func TestTUint16(t *testing.T) {
var buf [8]byte
for _, test := range tuint16Tests {
test := test
if len(test.bytes) != int(test.size) {
t.Fatalf("invalid test case, "+
"len(bytes)[%d] != size[%d]",
@@ -68,6 +70,7 @@ func TestTUint16(t *testing.T) {
name := fmt.Sprintf("0x%x", test.value)
t.Run(name, func(t *testing.T) {
// Test generic encoder.
var b bytes.Buffer
err := tlv.ETUint16(&b, &test.value, &buf)
if err != nil {
@@ -80,6 +83,19 @@ func TestTUint16(t *testing.T) {
test.bytes, b.Bytes())
}
// Test non-generic encoder.
var b2 bytes.Buffer
err = tlv.ETUint16T(&b2, test.value, &buf)
if err != nil {
t.Fatalf("unable to encode tuint16: %v", err)
}
if !bytes.Equal(b2.Bytes(), test.bytes) {
t.Fatalf("encoding mismatch, "+
"expected: %x, got: %x",
test.bytes, b2.Bytes())
}
var value uint16
r := bytes.NewReader(b.Bytes())
err = tlv.DTUint16(r, &value, &buf, test.size)
@@ -168,6 +184,8 @@ func TestSizeTUint32(t *testing.T) {
func TestTUint32(t *testing.T) {
var buf [8]byte
for _, test := range tuint32Tests {
test := test
if len(test.bytes) != int(test.size) {
t.Fatalf("invalid test case, "+
"len(bytes)[%d] != size[%d]",
@@ -176,6 +194,7 @@ func TestTUint32(t *testing.T) {
name := fmt.Sprintf("0x%x", test.value)
t.Run(name, func(t *testing.T) {
// Test generic encoder.
var b bytes.Buffer
err := tlv.ETUint32(&b, &test.value, &buf)
if err != nil {
@@ -188,6 +207,19 @@ func TestTUint32(t *testing.T) {
test.bytes, b.Bytes())
}
// Test non-generic encoder.
var b2 bytes.Buffer
err = tlv.ETUint32T(&b2, test.value, &buf)
if err != nil {
t.Fatalf("unable to encode tuint32: %v", err)
}
if !bytes.Equal(b2.Bytes(), test.bytes) {
t.Fatalf("encoding mismatch, "+
"expected: %x, got: %x",
test.bytes, b2.Bytes())
}
var value uint32
r := bytes.NewReader(b.Bytes())
err = tlv.DTUint32(r, &value, &buf, test.size)
@@ -322,6 +354,8 @@ func TestSizeTUint64(t *testing.T) {
func TestTUint64(t *testing.T) {
var buf [8]byte
for _, test := range tuint64Tests {
test := test
if len(test.bytes) != int(test.size) {
t.Fatalf("invalid test case, "+
"len(bytes)[%d] != size[%d]",
@@ -330,6 +364,7 @@ func TestTUint64(t *testing.T) {
name := fmt.Sprintf("0x%x", test.value)
t.Run(name, func(t *testing.T) {
// Test generic encoder.
var b bytes.Buffer
err := tlv.ETUint64(&b, &test.value, &buf)
if err != nil {
@@ -342,6 +377,19 @@ func TestTUint64(t *testing.T) {
test.bytes, b.Bytes())
}
// Test non-generic encoder.
var b2 bytes.Buffer
err = tlv.ETUint64T(&b2, test.value, &buf)
if err != nil {
t.Fatalf("unable to encode tuint64: %v", err)
}
if !bytes.Equal(b2.Bytes(), test.bytes) {
t.Fatalf("encoding mismatch, "+
"expected: %x, got: %x",
test.bytes, b2.Bytes())
}
var value uint64
r := bytes.NewReader(b.Bytes())
err = tlv.DTUint64(r, &value, &buf, test.size)