From c3504c111e409f9c7783a8ca6bddee218efa4ba1 Mon Sep 17 00:00:00 2001
From: Ethan Buchman <ethan@coinculture.info>
Date: Mon, 2 Jul 2018 13:04:34 -0400
Subject: [PATCH] add test

---
 p2p/listener_test.go | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/p2p/listener_test.go b/p2p/listener_test.go
index c82ae3909..f87b5d6f5 100644
--- a/p2p/listener_test.go
+++ b/p2p/listener_test.go
@@ -2,8 +2,11 @@ package p2p
 
 import (
 	"bytes"
+	"net"
+	"strings"
 	"testing"
 
+	"github.com/stretchr/testify/require"
 	"github.com/tendermint/tendermint/libs/log"
 )
 
@@ -45,3 +48,32 @@ func TestListener(t *testing.T) {
 	// Close the server, no longer needed.
 	l.Stop()
 }
+
+func TestExternalAddress(t *testing.T) {
+	{
+		// Create a listener with no external addr. Should default
+		// to local ipv4.
+		l := NewDefaultListener("tcp://:8001", "", false, log.TestingLogger())
+		lAddr := l.ExternalAddress().String()
+		_, _, err := net.SplitHostPort(lAddr)
+		require.Nil(t, err)
+		spl := strings.Split(lAddr, ".")
+		require.Equal(t, len(spl), 4)
+		l.Stop()
+	}
+
+	{
+		// Create a listener with set external ipv4 addr.
+		setExAddr := "8.8.8.8:8080"
+		l := NewDefaultListener("tcp://:8001", setExAddr, false, log.TestingLogger())
+		lAddr := l.ExternalAddress().String()
+		require.Equal(t, lAddr, setExAddr)
+		l.Stop()
+	}
+
+	{
+		// Invalid external addr causes panic
+		setExAddr := "awrlsckjnal:8080"
+		require.Panics(t, func() { NewDefaultListener("tcp://:8001", setExAddr, false, log.TestingLogger()) })
+	}
+}