Function createLocalHostAndSyncClient

ethstorage/p2p/protocol/sync_test.go:332–371

(t *testing.T, chainID *big.Int, db ethdb.Database,
	storageManager StorageManager, metrics SyncClientMetrics, mux *event.Feed)

}

func createLocalHostAndSyncClient(t *testing.T, chainID *big.Int, db ethdb.Database,
	storageManager StorageManager, metrics SyncClientMetrics, mux *event.Feed) (host.Host, *SyncClient) {
	localHost := getNetHost(t)

	syncCl := NewSyncClient(lg, chainID, localHost.NewStream, storageManager, &params, db, metrics, mux)
	localHost.Network().Notify(&network.NotifyBundle{
		ConnectedF: func(nw network.Network, conn network.Conn) {
			shards := make(map[common.Address][]uint64)
			css, err := localHost.Peerstore().Get(conn.RemotePeer(), EthStorageENRKey)
			if err != nil {
				lg.Warn("Get shards from peer failed", "error", err.Error())
			} else {
				shards = ConvertToShardList(css.([]*ContractShards))
			}

			added := syncCl.AddPeer(conn.RemotePeer(), shards, conn.Stat().Direction)
			if !added {
				conn.Close()
			}
		},
		DisconnectedF: func(nw network.Network, conn network.Conn) {
			syncCl.RemovePeer(conn.RemotePeer())
		},
	})
	// the host may already be connected to peers, add them all to the sync client
	for _, conn := range localHost.Network().Conns() {
		shards := make(map[common.Address][]uint64)
		css, err := localHost.Peerstore().Get(conn.RemotePeer(), EthStorageENRKey)
		if err != nil {
			lg.Warn("Get shards from peer failed", "error", err.Error())
		} else {
			shards = ConvertToShardList(css.([]*ContractShards))
		}
		added := syncCl.AddPeer(conn.RemotePeer(), shards, conn.Stat().Direction)
		if !added {
			conn.Close()
		}
	}
	return localHost, syncCl
}

func createRemoteHost(t *testing.T, ctx context.Context, chainID *big.Int,
	storageManager *mockStorageManagerReader, db ethdb.Database, metrics SyncServerMetrics) host.Host {