@@ -765,8 +765,9 @@ mod tests {
765765 use libp2p:: { Multiaddr , PeerId } ;
766766 use libp2p:: core:: upgrade;
767767 use libp2p:: core:: transport:: { Transport , MemoryTransport } ;
768- use libp2p:: core :: upgrade :: { InboundUpgradeExt , OutboundUpgradeExt } ;
768+ use libp2p:: noise ;
769769 use libp2p:: swarm:: Swarm ;
770+ use libp2p:: yamux;
770771 use std:: { collections:: HashSet , task:: Poll } ;
771772 use super :: { DiscoveryConfig , DiscoveryOut , protocol_name_from_protocol_id} ;
772773
@@ -779,25 +780,15 @@ mod tests {
779780 // the first swarm via `with_user_defined`.
780781 let mut swarms = ( 0 ..25 ) . map ( |i| {
781782 let keypair = Keypair :: generate_ed25519 ( ) ;
782- let keypair2 = keypair. clone ( ) ;
783+
784+ let noise_keys = noise:: Keypair :: < noise:: X25519Spec > :: new ( )
785+ . into_authentic ( & keypair)
786+ . unwrap ( ) ;
783787
784788 let transport = MemoryTransport
785- . and_then ( move |out, endpoint| {
786- let secio = libp2p:: secio:: SecioConfig :: new ( keypair2) ;
787- libp2p:: core:: upgrade:: apply (
788- out,
789- secio,
790- endpoint,
791- upgrade:: Version :: V1
792- )
793- } )
794- . and_then ( move |( peer_id, stream) , endpoint| {
795- let peer_id2 = peer_id. clone ( ) ;
796- let upgrade = libp2p:: yamux:: Config :: default ( )
797- . map_inbound ( move |muxer| ( peer_id, muxer) )
798- . map_outbound ( move |muxer| ( peer_id2, muxer) ) ;
799- upgrade:: apply ( stream, upgrade, endpoint, upgrade:: Version :: V1 )
800- } ) ;
789+ . upgrade ( upgrade:: Version :: V1 )
790+ . authenticate ( noise:: NoiseConfig :: xx ( noise_keys) . into_authenticated ( ) )
791+ . multiplex ( yamux:: Config :: default ( ) ) ;
801792
802793 let behaviour = {
803794 let mut config = DiscoveryConfig :: new ( keypair. public ( ) ) ;
0 commit comments