basic_test.go

package bifrost

import (
	"context"
	"testing"

	"github.com/aperturerobotics/bifrost/sim/graph"
	"github.com/aperturerobotics/bifrost/sim/simulate"
	"github.com/sirupsen/logrus"
)

// TestBasic tests a simple connection between two peers on a LAN.
func TestBasic(t *testing.T) {
	ctx, ctxCancel := context.WithCancel(context.Background())
	defer ctxCancel()

	log := logrus.New()
	log.SetLevel(logrus.DebugLevel)
	le := logrus.NewEntry(log)

	g := graph.NewGraph()

	descrip := `p0 <-> [lan1] <-> p1`

	p0 := addPeer(t, g)
	p1 := addPeer(t, g)

	lan1 := graph.AddLAN(g)
	lan1.AddPeer(g, p0)
	lan1.AddPeer(g, p1)

	sim := initSimulator(
		t,
		ctx,
		le,
		g,
		simulate.WithVerbose(),
	)

	le.Infof("attempting to dial %v", descrip)
	if err := simulate.TestConnectivity(
		ctx,
		sim.GetPeerByID(p0.GetPeerID()),
		sim.GetPeerByID(p1.GetPeerID()),
	); err != nil {
		t.Fatal(err.Error())
	}

	le.Infof("successful connectivity test: %v", descrip)
	// <-ctx.Done()
}