modify for supporting added ip parameter of BuildResponsePeerInfo fun…

…ction
DE-labtory · Nov 4, 2018 · 7880078 · 7880078
1 parent 85927f2
commit 7880078
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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