Skip to content

Commit

Permalink
fix(test): fix test with rdv point
Browse files Browse the repository at this point in the history
  • Loading branch information
gfanton committed Apr 16, 2020
1 parent f70fb2f commit fe4d273
Show file tree
Hide file tree
Showing 6 changed files with 41 additions and 170 deletions.
28 changes: 21 additions & 7 deletions go/internal/handshake/handshake_util_test.go
Expand Up @@ -30,6 +30,8 @@ type mockedPeer struct {
type mockedHandshake struct {
requester *mockedPeer
responder *mockedPeer

cleanup func()
}

type requesterTestFunc func(
Expand Down Expand Up @@ -64,26 +66,37 @@ func newMockedPeer(t *testing.T, ctx context.Context, ipfsOpts *ipfsutil.Testing
func newMockedHandshake(t *testing.T, ctx context.Context) *mockedHandshake {
t.Helper()

mn := p2pmocknet.New(ctx)
rdvp, err := mn.GenPeer()
require.NoError(t, err, "failed to generate mocked peer")

_, rdv_cleanup := ipfsutil.TestingRDVP(ctx, t, rdvp)

opts := &ipfsutil.TestingAPIOpts{
Mocknet: p2pmocknet.New(ctx),
Mocknet: mn,
RDVPeer: rdvp.ID(),
}
requester, cleanup := newMockedPeer(t, ctx, opts)
defer cleanup()

responder, cleanup := newMockedPeer(t, ctx, opts)
defer cleanup()
requester, req_cleanup := newMockedPeer(t, ctx, opts)
responder, res_cleanup := newMockedPeer(t, ctx, opts)

// link responder & requester
err := opts.Mocknet.LinkAll()
err = opts.Mocknet.LinkAll()
require.NoError(t, err, "can't link peers")

// connect responder & requester
err = opts.Mocknet.ConnectAllButSelf()
require.NoError(t, err, "can't connect peers")

cleanup := func() {
res_cleanup()
req_cleanup()
rdv_cleanup()
}

return &mockedHandshake{
requester: requester,
responder: responder,
cleanup: cleanup,
}
}

Expand All @@ -92,6 +105,7 @@ func (mh *mockedHandshake) close(t *testing.T) {

mh.requester.coreAPI.Close()
mh.responder.coreAPI.Close()
mh.cleanup()
}

func newTestHandshakeContext(stream p2pnetwork.Stream, ownAccountID p2pcrypto.PrivKey, peerAccountID p2pcrypto.PubKey) *handshakeContext {
Expand Down
6 changes: 2 additions & 4 deletions go/internal/testutil/logging.go
Expand Up @@ -6,7 +6,7 @@ import (
"strconv"
"testing"

"berty.tech/berty/v2/go/internal/ipfsutil"
ipfs_log "github.com/ipfs/go-log"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
Expand Down Expand Up @@ -41,9 +41,7 @@ func Logger(t *testing.T) *zap.Logger {
}

if libp2pDebug {
if err := ipfsutil.ConfigureLogger("*", logger, "debug"); err != nil {
logger.Error("setup libp2p logger", zap.Error(err))
}
ipfs_log.SetDebugLogging()
}

if orbitdbDebug {
Expand Down
66 changes: 0 additions & 66 deletions go/internal/tinder/driver_mocked_test.go

This file was deleted.

10 changes: 7 additions & 3 deletions go/internal/tinder/driver_multi_test.go
Expand Up @@ -5,6 +5,7 @@ import (
"testing"
"time"

"berty.tech/berty/v2/go/internal/testutil"
p2p_discovery "github.com/libp2p/go-libp2p-core/discovery"
p2p_host "github.com/libp2p/go-libp2p-core/host"
p2p_disc "github.com/libp2p/go-libp2p-discovery"
Expand Down Expand Up @@ -38,6 +39,7 @@ func TestMultiDriver_Advertise(t *testing.T) {
},
}

logger := testutil.Logger(t)
ctx, cancel := context.WithCancel(context.Background())
defer cancel()

Expand All @@ -54,7 +56,7 @@ func TestMultiDriver_Advertise(t *testing.T) {

peers := testingPeers(t, mn, tc.NMock)
drivers := testingMockedDriverClients(t, ms, peers...)
md := NewMultiDriver(drivers...)
md := NewMultiDriver(logger, drivers...)

const testKey = "testkey"

Expand Down Expand Up @@ -96,6 +98,7 @@ func TestMultiDriver_FindPeers(t *testing.T) {
},
}

logger := testutil.Logger(t)
ctx, cancel := context.WithCancel(context.Background())
defer cancel()

Expand All @@ -112,7 +115,7 @@ func TestMultiDriver_FindPeers(t *testing.T) {

peers := testingPeers(t, mn, tc.NMock)
drivers := testingMockedDriverClients(t, ms, peers...)
md := NewMultiDriver(drivers...)
md := NewMultiDriver(logger, drivers...)

const testKey = "testkey"
_, err = md.Advertise(ctx, testKey, tc.Opts...)
Expand Down Expand Up @@ -147,6 +150,7 @@ func TestMultiDriver_Unregister(t *testing.T) {
},
}

logger := testutil.Logger(t)
ctx, cancel := context.WithCancel(context.Background())
defer cancel()

Expand All @@ -159,7 +163,7 @@ func TestMultiDriver_Unregister(t *testing.T) {

peers := testingPeers(t, mn, tc.NMock)
drivers := testingMockedDriverClients(t, ms, peers...)
md := NewMultiDriver(drivers...)
md := NewMultiDriver(logger, drivers...)

const testKey = "testkey"
_, err := md.Advertise(ctx, testKey, tc.Opts...)
Expand Down
87 changes: 0 additions & 87 deletions go/internal/tinder/driver_rdv_test.go

This file was deleted.

14 changes: 11 additions & 3 deletions go/pkg/bertyprotocol/orbitdb_utils_test.go
Expand Up @@ -74,8 +74,15 @@ func createPeersWithGroup(ctx context.Context, t testing.TB, pathBase string, me
t.Fatal(err)
}

mn := mocknet.New(ctx)
rdvp, err := mn.GenPeer()
require.NoError(t, err, "failed to generate mocked peer")

_, _ = ipfsutil.TestingRDVP(ctx, t, rdvp)

ipfsopts := ipfsutil.TestingAPIOpts{
Mocknet: mocknet.New(ctx),
Mocknet: mn,
RDVPeer: rdvp.ID(),
}
deviceIndex := 0

Expand Down Expand Up @@ -117,7 +124,8 @@ func createPeersWithGroup(ctx context.Context, t testing.TB, pathBase string, me
DevKS: devKS,
}

cls = append(cls, func() {
// setup cleanup
cls[i] = func() {
if ms := mp.GC.MetadataStore(); ms != nil {
err := ms.Drop()
assert.NoError(t, err)
Expand All @@ -129,7 +137,7 @@ func createPeersWithGroup(ctx context.Context, t testing.TB, pathBase string, me
}

cleanupNode()
})
}

mockedPeers[deviceIndex] = mp
deviceIndex++
Expand Down

0 comments on commit fe4d273

Please sign in to comment.