From 5f1d2524b8830105405826353f55099f26283c61 Mon Sep 17 00:00:00 2001
From: Wilmer Paulino <wilmer.paulino@gmail.com>
Date: Fri, 27 Apr 2018 16:52:41 -0400
Subject: [PATCH] channeldb: add address serialization tests

---
 channeldb/addr_test.go | 51 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 channeldb/addr_test.go

diff --git a/channeldb/addr_test.go b/channeldb/addr_test.go
new file mode 100644
index 00000000..2de179d6
--- /dev/null
+++ b/channeldb/addr_test.go
@@ -0,0 +1,51 @@
+package channeldb
+
+import (
+	"bytes"
+	"net"
+	"testing"
+
+	"github.com/lightningnetwork/lnd/tor"
+)
+
+// TestAddrSerialization tests that the serialization method used by channeldb
+// for net.Addr's works as intended.
+func TestAddrSerialization(t *testing.T) {
+	t.Parallel()
+
+	testAddrs := []net.Addr{
+		&net.TCPAddr{
+			IP:   net.ParseIP("192.168.1.1"),
+			Port: 12345,
+		},
+		&net.TCPAddr{
+			IP:   net.ParseIP("2001:0db8:0000:0000:0000:ff00:0042:8329"),
+			Port: 65535,
+		},
+		&tor.OnionAddr{
+			OnionService: "3g2upl4pq6kufc4m.onion",
+			Port:         9735,
+		},
+		&tor.OnionAddr{
+			OnionService: "vww6ybal4bd7szmgncyruucpgfkqahzddi37ktceo3ah7ngmcopnpyyd.onion",
+			Port:         80,
+		},
+	}
+
+	var b bytes.Buffer
+	for _, expectedAddr := range testAddrs {
+		if err := serializeAddr(&b, expectedAddr); err != nil {
+			t.Fatalf("unable to serialize address: %v", err)
+		}
+
+		addr, err := deserializeAddr(&b)
+		if err != nil {
+			t.Fatalf("unable to deserialize address: %v", err)
+		}
+
+		if addr.String() != expectedAddr.String() {
+			t.Fatalf("expected address %v after serialization, "+
+				"got %v", addr, expectedAddr)
+		}
+	}
+}