github.com/keltia/go-ipfs@v0.3.8-0.20150909044612-210793031c63/p2p/net/mock/mock.go (about)

     1  package mocknet
     2  
     3  import (
     4  	eventlog "github.com/ipfs/go-ipfs/thirdparty/eventlog"
     5  
     6  	context "github.com/ipfs/go-ipfs/Godeps/_workspace/src/golang.org/x/net/context"
     7  )
     8  
     9  var log = eventlog.Logger("mocknet")
    10  
    11  // WithNPeers constructs a Mocknet with N peers.
    12  func WithNPeers(ctx context.Context, n int) (Mocknet, error) {
    13  	m := New(ctx)
    14  	for i := 0; i < n; i++ {
    15  		if _, err := m.GenPeer(); err != nil {
    16  			return nil, err
    17  		}
    18  	}
    19  	return m, nil
    20  }
    21  
    22  // FullMeshLinked constructs a Mocknet with full mesh of Links.
    23  // This means that all the peers **can** connect to each other
    24  // (not that they already are connected. you can use m.ConnectAll())
    25  func FullMeshLinked(ctx context.Context, n int) (Mocknet, error) {
    26  	m, err := WithNPeers(ctx, n)
    27  	if err != nil {
    28  		return nil, err
    29  	}
    30  
    31  	if err := m.LinkAll(); err != nil {
    32  		return nil, err
    33  	}
    34  
    35  	return m, nil
    36  }
    37  
    38  // FullMeshConnected constructs a Mocknet with full mesh of Connections.
    39  // This means that all the peers have dialed and are ready to talk to
    40  // each other.
    41  func FullMeshConnected(ctx context.Context, n int) (Mocknet, error) {
    42  	m, err := FullMeshLinked(ctx, n)
    43  	if err != nil {
    44  		return nil, err
    45  	}
    46  
    47  	nets := m.Nets()
    48  	for _, n1 := range nets {
    49  		for _, n2 := range nets {
    50  			if _, err := m.ConnectNets(n1, n2); err != nil {
    51  				return nil, err
    52  			}
    53  		}
    54  	}
    55  
    56  	return m, nil
    57  }