Function TestBasicTelemetryController_Unimplemented

tailnet/controllers_test.go:679–734

(t *testing.T)

}

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

	ft := newFakeTelemetryClient()

	uut := tailnet.NewBasicTelemetryController(logger)
	uut.New(ft)

	// bad code, doesn't count
	telemetryError := drpcerr.WithCode(xerrors.New("Unimplemented"), 0)

	sendDone := make(chan struct{})
	go func() {
		defer close(sendDone)
		uut.SendTelemetryEvent(&proto.TelemetryEvent{})
	}()

	call := testutil.TryReceive(ctx, t, ft.calls)
	testutil.RequireSend(ctx, t, call.errCh, telemetryError)
	testutil.TryReceive(ctx, t, sendDone)

	sendDone = make(chan struct{})
	go func() {
		defer close(sendDone)
		uut.SendTelemetryEvent(&proto.TelemetryEvent{})
	}()

	// we get another call since it wasn't really the Unimplemented error
	call = testutil.TryReceive(ctx, t, ft.calls)

	// for real this time
	telemetryError = errUnimplemented
	testutil.RequireSend(ctx, t, call.errCh, telemetryError)
	testutil.TryReceive(ctx, t, sendDone)

	// now this returns immediately without a call, because unimplemented error disables calling
	sendDone = make(chan struct{})
	go func() {
		defer close(sendDone)
		uut.SendTelemetryEvent(&proto.TelemetryEvent{})
	}()
	testutil.TryReceive(ctx, t, sendDone)

	// getting a "new" client resets
	uut.New(ft)
	sendDone = make(chan struct{})
	go func() {
		defer close(sendDone)
		uut.SendTelemetryEvent(&proto.TelemetryEvent{})
	}()
	call = testutil.TryReceive(ctx, t, ft.calls)
	testutil.RequireSend(ctx, t, call.errCh, nil)
	testutil.TryReceive(ctx, t, sendDone)
}

func TestBasicTelemetryController_NotRecognised(t *testing.T) {