@@ -424,84 +424,88 @@ describe(`${NodeConnectionManager.name} seed nodes test`, () => {
424424 await queue ?. stop ( ) ;
425425 }
426426 } ) ;
427- test ( 'should expand the network when nodes enter' , async ( ) => {
428- // Using a single seed node we need to check that each entering node adds itself to the seed node.
429- // Also need to check that the new nodes can be seen in the network.
430- let node1 : PolykeyAgent | undefined ;
431- let node2 : PolykeyAgent | undefined ;
432- const seedNodes : SeedNodes = { } ;
433- seedNodes [ nodesUtils . encodeNodeId ( remoteNodeId1 ) ] = {
434- host : remoteNode1 . proxy . getProxyHost ( ) ,
435- port : remoteNode1 . proxy . getProxyPort ( ) ,
436- } ;
437- seedNodes [ nodesUtils . encodeNodeId ( remoteNodeId2 ) ] = {
438- host : remoteNode2 . proxy . getProxyHost ( ) ,
439- port : remoteNode2 . proxy . getProxyPort ( ) ,
440- } ;
441- try {
442- logger . setLevel ( LogLevel . WARN ) ;
443- node1 = await PolykeyAgent . createPolykeyAgent ( {
444- nodePath : path . join ( dataDir , 'node1' ) ,
445- password : 'password' ,
446- networkConfig : {
447- proxyHost : localHost ,
448- agentHost : localHost ,
449- clientHost : localHost ,
450- forwardHost : localHost ,
451- } ,
452- seedNodes,
453- logger,
454- } ) ;
455- node2 = await PolykeyAgent . createPolykeyAgent ( {
456- nodePath : path . join ( dataDir , 'node2' ) ,
457- password : 'password' ,
458- networkConfig : {
459- proxyHost : localHost ,
460- agentHost : localHost ,
461- clientHost : localHost ,
462- forwardHost : localHost ,
463- } ,
464- seedNodes,
465- logger,
466- } ) ;
427+ test (
428+ 'should expand the network when nodes enter' ,
429+ async ( ) => {
430+ // Using a single seed node we need to check that each entering node adds itself to the seed node.
431+ // Also need to check that the new nodes can be seen in the network.
432+ let node1 : PolykeyAgent | undefined ;
433+ let node2 : PolykeyAgent | undefined ;
434+ const seedNodes : SeedNodes = { } ;
435+ seedNodes [ nodesUtils . encodeNodeId ( remoteNodeId1 ) ] = {
436+ host : remoteNode1 . proxy . getProxyHost ( ) ,
437+ port : remoteNode1 . proxy . getProxyPort ( ) ,
438+ } ;
439+ seedNodes [ nodesUtils . encodeNodeId ( remoteNodeId2 ) ] = {
440+ host : remoteNode2 . proxy . getProxyHost ( ) ,
441+ port : remoteNode2 . proxy . getProxyPort ( ) ,
442+ } ;
443+ try {
444+ logger . setLevel ( LogLevel . WARN ) ;
445+ node1 = await PolykeyAgent . createPolykeyAgent ( {
446+ nodePath : path . join ( dataDir , 'node1' ) ,
447+ password : 'password' ,
448+ networkConfig : {
449+ proxyHost : localHost ,
450+ agentHost : localHost ,
451+ clientHost : localHost ,
452+ forwardHost : localHost ,
453+ } ,
454+ seedNodes,
455+ logger,
456+ } ) ;
457+ node2 = await PolykeyAgent . createPolykeyAgent ( {
458+ nodePath : path . join ( dataDir , 'node2' ) ,
459+ password : 'password' ,
460+ networkConfig : {
461+ proxyHost : localHost ,
462+ agentHost : localHost ,
463+ clientHost : localHost ,
464+ forwardHost : localHost ,
465+ } ,
466+ seedNodes,
467+ logger,
468+ } ) ;
467469
468- await node1 . queue . drained ( ) ;
469- await node1 . nodeManager . refreshBucketQueueDrained ( ) ;
470- await node2 . queue . drained ( ) ;
471- await node2 . nodeManager . refreshBucketQueueDrained ( ) ;
470+ await node1 . queue . drained ( ) ;
471+ await node1 . nodeManager . refreshBucketQueueDrained ( ) ;
472+ await node2 . queue . drained ( ) ;
473+ await node2 . nodeManager . refreshBucketQueueDrained ( ) ;
472474
473- const getAllNodes = async ( node : PolykeyAgent ) => {
474- const nodes : Array < NodeIdEncoded > = [ ] ;
475- for await ( const [ nodeId ] of node . nodeGraph . getNodes ( ) ) {
476- nodes . push ( nodesUtils . encodeNodeId ( nodeId ) ) ;
477- }
478- return nodes ;
479- } ;
480- const rNode1Nodes = await getAllNodes ( remoteNode1 ) ;
481- const rNode2Nodes = await getAllNodes ( remoteNode2 ) ;
482- const node1Nodes = await getAllNodes ( node1 ) ;
483- const node2Nodes = await getAllNodes ( node2 ) ;
475+ const getAllNodes = async ( node : PolykeyAgent ) => {
476+ const nodes : Array < NodeIdEncoded > = [ ] ;
477+ for await ( const [ nodeId ] of node . nodeGraph . getNodes ( ) ) {
478+ nodes . push ( nodesUtils . encodeNodeId ( nodeId ) ) ;
479+ }
480+ return nodes ;
481+ } ;
482+ const rNode1Nodes = await getAllNodes ( remoteNode1 ) ;
483+ const rNode2Nodes = await getAllNodes ( remoteNode2 ) ;
484+ const node1Nodes = await getAllNodes ( node1 ) ;
485+ const node2Nodes = await getAllNodes ( node2 ) ;
484486
485- const nodeIdR1 = nodesUtils . encodeNodeId ( remoteNodeId1 ) ;
486- const nodeIdR2 = nodesUtils . encodeNodeId ( remoteNodeId2 ) ;
487- const nodeId1 = nodesUtils . encodeNodeId ( node1 . keyManager . getNodeId ( ) ) ;
488- const nodeId2 = nodesUtils . encodeNodeId ( node2 . keyManager . getNodeId ( ) ) ;
489- expect ( rNode1Nodes ) . toContain ( nodeId1 ) ;
490- expect ( rNode1Nodes ) . toContain ( nodeId2 ) ;
491- expect ( rNode2Nodes ) . toContain ( nodeId1 ) ;
492- expect ( rNode2Nodes ) . toContain ( nodeId2 ) ;
493- expect ( node1Nodes ) . toContain ( nodeIdR1 ) ;
494- expect ( node1Nodes ) . toContain ( nodeIdR2 ) ;
495- expect ( node1Nodes ) . toContain ( nodeId2 ) ;
496- expect ( node2Nodes ) . toContain ( nodeIdR1 ) ;
497- expect ( node2Nodes ) . toContain ( nodeIdR2 ) ;
498- expect ( node2Nodes ) . toContain ( nodeId1 ) ;
499- } finally {
500- logger . setLevel ( LogLevel . WARN ) ;
501- await node1 ?. stop ( ) ;
502- await node1 ?. destroy ( ) ;
503- await node2 ?. stop ( ) ;
504- await node2 ?. destroy ( ) ;
505- }
506- } , global . defaultTimeout * 2 ) ;
487+ const nodeIdR1 = nodesUtils . encodeNodeId ( remoteNodeId1 ) ;
488+ const nodeIdR2 = nodesUtils . encodeNodeId ( remoteNodeId2 ) ;
489+ const nodeId1 = nodesUtils . encodeNodeId ( node1 . keyManager . getNodeId ( ) ) ;
490+ const nodeId2 = nodesUtils . encodeNodeId ( node2 . keyManager . getNodeId ( ) ) ;
491+ expect ( rNode1Nodes ) . toContain ( nodeId1 ) ;
492+ expect ( rNode1Nodes ) . toContain ( nodeId2 ) ;
493+ expect ( rNode2Nodes ) . toContain ( nodeId1 ) ;
494+ expect ( rNode2Nodes ) . toContain ( nodeId2 ) ;
495+ expect ( node1Nodes ) . toContain ( nodeIdR1 ) ;
496+ expect ( node1Nodes ) . toContain ( nodeIdR2 ) ;
497+ expect ( node1Nodes ) . toContain ( nodeId2 ) ;
498+ expect ( node2Nodes ) . toContain ( nodeIdR1 ) ;
499+ expect ( node2Nodes ) . toContain ( nodeIdR2 ) ;
500+ expect ( node2Nodes ) . toContain ( nodeId1 ) ;
501+ } finally {
502+ logger . setLevel ( LogLevel . WARN ) ;
503+ await node1 ?. stop ( ) ;
504+ await node1 ?. destroy ( ) ;
505+ await node2 ?. stop ( ) ;
506+ await node2 ?. destroy ( ) ;
507+ }
508+ } ,
509+ global . defaultTimeout * 2 ,
510+ ) ;
507511} ) ;
0 commit comments