Function TestPGCoordinator_MultiAgent

enterprise/tailnet/multiagent_test.go:24–56

TestPGCoordinator_MultiAgent tests a single coordinator with a MultiAgent connecting to one agent. +--------+ agent1 ---> | coord1 | <--- client +--------+

(t *testing.T)

//	agent1 ---> | coord1 | <--- client
//	            +--------+
func TestPGCoordinator_MultiAgent(t *testing.T) {
	t.Parallel()

	logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: true}).Leveled(slog.LevelDebug)
	store, ps := dbtestutil.NewDB(t)
	ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
	defer cancel()
	coord1, err := tailnet.NewPGCoord(ctx, logger.Named("coord1"), ps, store)
	require.NoError(t, err)
	defer coord1.Close()

	agent1 := agpltest.NewAgent(ctx, t, coord1, "agent1")
	defer agent1.Close(ctx)
	agent1.UpdateDERP(5)

	ma1 := agpltest.NewPeer(ctx, t, coord1, "client")
	defer ma1.Close(ctx)

	ma1.AddTunnel(agent1.ID)
	ma1.AssertEventuallyHasDERP(agent1.ID, 5)

	agent1.UpdateDERP(1)
	ma1.AssertEventuallyHasDERP(agent1.ID, 1)

	ma1.UpdateDERP(3)
	agent1.AssertEventuallyHasDERP(ma1.ID, 3)

	ma1.Disconnect()
	agent1.UngracefulDisconnect(ctx)

	assertEventuallyNoClientsForAgent(ctx, t, store, agent1.ID)
	assertEventuallyLost(ctx, t, store, agent1.ID)
}

func TestPGCoordinator_MultiAgent_CoordClose(t *testing.T) {
	t.Parallel()