From 94db23fd41e74e2c03fc9ee9d77e81201bdc57ce Mon Sep 17 00:00:00 2001 From: vyzo Date: Thu, 23 Apr 2020 12:04:56 +0300 Subject: [PATCH] add signed peer records only in the center of the star for signed peer record test --- gossipsub_test.go | 25 +++++++++---------------- 1 file changed, 9 insertions(+), 16 deletions(-) diff --git a/gossipsub_test.go b/gossipsub_test.go index 0b1451a..7ca243a 100644 --- a/gossipsub_test.go +++ b/gossipsub_test.go @@ -1007,10 +1007,8 @@ func TestGossipsubStarTopologyWithSignedPeerRecords(t *testing.T) { psubs := getGossipsubs(ctx, hosts, WithPeerExchange(true)) // manually create signed peer records for each host and add them to the - // peerstores of the other hosts. - // this is necessary until the identify service is updated to exchange - // signed records - for i := range hosts { + // peerstore of the center of the star, which is doing the bootstrapping + for i := range hosts[1:] { privKey := hosts[i].Peerstore().PrivKey(hosts[i].ID()) if privKey == nil { t.Fatalf("unable to get private key for host %s", hosts[i].ID().Pretty()) @@ -1022,18 +1020,13 @@ func TestGossipsubStarTopologyWithSignedPeerRecords(t *testing.T) { t.Fatalf("error creating signed peer record: %s", err) } - for j := range hosts { - if i == j { - continue - } - cab, ok := peerstore.GetCertifiedAddrBook(hosts[j].Peerstore()) - if !ok { - t.Fatal("peerstore does not implement CertifiedAddrBook") - } - _, err := cab.ConsumePeerRecord(signedRec, peerstore.PermanentAddrTTL) - if err != nil { - t.Fatalf("error adding signed peer record: %s", err) - } + cab, ok := peerstore.GetCertifiedAddrBook(hosts[0].Peerstore()) + if !ok { + t.Fatal("peerstore does not implement CertifiedAddrBook") + } + _, err = cab.ConsumePeerRecord(signedRec, peerstore.PermanentAddrTTL) + if err != nil { + t.Fatalf("error adding signed peer record: %s", err) } }