server/federation.go

@@ -374,7 +374,7 @@ func (s *Server) addPeer(newPeer *Peer, ping bool, subscribe bool) error {
 		// This is weird because we're doing grpc and jsonrpc here.
 		// Do we still want to custom grpc?
 		log.Warn("Sending peer to NotifierChan")
-		s.NotifierChan <- newPeer
+		s.NotifierChan <- peerNotification{newPeer.Address, newPeer.Port}
 
 		// Subscribe to all our peers for now
 		if subscribe {

server/federation_test.go

@@ -93,7 +93,6 @@ func TestAddPeer(t *testing.T) {
 					log.Println(err)
 				}
 			}
-			hubServer.Stop()
 			var m = &dto.Metric{}
 			if err := metrics.PeersKnown.Write(m); err != nil {
 				t.Errorf("Error getting metrics %+v\n", err)
@@ -102,6 +101,7 @@ func TestAddPeer(t *testing.T) {
 			if got != tt.want {
 				t.Errorf("len(server.PeerServers) = %d, want %d\n", got, tt.want)
 			}
+			hubServer.Stop()
 		})
 	}