From 7880078fe94ac36ee85be15621024df4d91db7f9 Mon Sep 17 00:00:00 2001
From: dongin <needshield@gmail.com>
Date: Sat, 3 Nov 2018 17:37:49 +0900
Subject: [PATCH] modify for supporting added ip parameter of
 BuildResponsePeerInfo function

---
 client/client.go | 2 +-
 server/server.go | 2 +-
 util_test.go     | 3 ++-
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/client/client.go b/client/client.go
index 0bda290e..0778767e 100644
--- a/client/client.go
+++ b/client/client.go
@@ -122,7 +122,7 @@ func waitServer(streamWrapper bifrost.StreamWrapper) error {
 
 // handShake 두번째 과정 함수. client 의 peer info 메세지를 server 에게 전달한다.
 func sendInfo(streamWrapper bifrost.StreamWrapper, clientOpts ClientOpts, metaData map[string]string) error {
-	env, err := bifrost.BuildResponsePeerInfo(clientOpts.PubKey, metaData)
+	env, err := bifrost.BuildResponsePeerInfo(clientOpts.Ip, clientOpts.PubKey, metaData)
 
 	if err != nil {
 		return err
diff --git a/server/server.go b/server/server.go
index c71f1ac6..eac8ba91 100644
--- a/server/server.go
+++ b/server/server.go
@@ -92,7 +92,7 @@ func requestInfo(streamWrapper bifrost.StreamWrapper) error {
 
 func (s Server) sendInfo(streamWrapper bifrost.StreamWrapper, metaData map[string]string) error {
 
-	envelope, err := bifrost.BuildResponsePeerInfo(s.pubKey, metaData)
+	envelope, err := bifrost.BuildResponsePeerInfo(s.ip, s.pubKey, metaData)
 
 	if err != nil {
 		return errors.New("fail to build info")
diff --git a/util_test.go b/util_test.go
index 2a3967f9..8de4008f 100644
--- a/util_test.go
+++ b/util_test.go
@@ -11,10 +11,11 @@ import (
 
 func TestBuildResponsePeerInfo(t *testing.T) {
 	//given
+	ip := "127.0.0.1:2323"
 	keyOpt := mocks.NewMockKeyOpts()
 
 	//when
-	envelope, err := bifrost.BuildResponsePeerInfo(keyOpt.PubKey, nil)
+	envelope, err := bifrost.BuildResponsePeerInfo(ip, keyOpt.PubKey, nil)
 	assert.NoError(t, err)
 
 	//then