Function TestTunnelSrcCoordController_RemoveDestination

tailnet/controllers_test.go:194–252

(t *testing.T)

}

func TestTunnelSrcCoordController_RemoveDestination(t *testing.T) {
	t.Parallel()
	ctx := testutil.Context(t, testutil.WaitShort)
	logger := testutil.Logger(t)

	fConn := &fakeCoordinatee{}
	uut := tailnet.NewTunnelSrcCoordController(logger, fConn)

	// GIVEN: 1 destination
	dest1 := uuid.UUID{1}
	uut.AddDestination(dest1)

	// GIVEN: client already connected
	client1 := newFakeCoordinatorClient(ctx, t)
	cws := make(chan tailnet.CloserWaiter)
	go func() {
		cws <- uut.New(client1)
	}()
	call := testutil.TryReceive(ctx, t, client1.reqs)
	testutil.RequireSend(ctx, t, call.err, nil)
	cw1 := testutil.TryReceive(ctx, t, cws)

	// WHEN: we remove one destination
	removeDone := make(chan struct{})
	go func() {
		defer close(removeDone)
		uut.RemoveDestination(dest1)
	}()

	// THEN: Controller sends RemoveTunnel for the destination
	call = testutil.TryReceive(ctx, t, client1.reqs)
	require.Equal(t, dest1[:], call.req.GetRemoveTunnel().GetId())
	testutil.RequireSend(ctx, t, call.err, nil)
	_ = testutil.TryReceive(ctx, t, removeDone)

	// WHEN: Closed from server side and reconnect
	respCall := testutil.TryReceive(ctx, t, client1.resps)
	testutil.RequireSend(ctx, t, respCall.err, io.EOF)
	closeCall := testutil.TryReceive(ctx, t, client1.close)
	testutil.RequireSend(ctx, t, closeCall, nil)
	err := testutil.TryReceive(ctx, t, cw1.Wait())
	require.ErrorIs(t, err, io.EOF)

	client2 := newFakeCoordinatorClient(ctx, t)
	go func() {
		cws <- uut.New(client2)
	}()

	// THEN: should immediately resolve without sending anything
	cw2 := testutil.TryReceive(ctx, t, cws)

	// close client2
	respCall = testutil.TryReceive(ctx, t, client2.resps)
	testutil.RequireSend(ctx, t, respCall.err, io.EOF)
	closeCall = testutil.TryReceive(ctx, t, client2.close)
	testutil.RequireSend(ctx, t, closeCall, nil)
	err = testutil.TryReceive(ctx, t, cw2.Wait())
	require.ErrorIs(t, err, io.EOF)