Function TestLogSender_SanitizeOutput

codersdk/agentsdk/logs_internal_test.go:232–275

(t *testing.T)

}

func TestLogSender_SanitizeOutput(t *testing.T) {
	t.Parallel()
	testCtx := testutil.Context(t, testutil.WaitShort)
	ctx, cancel := context.WithCancel(testCtx)
	logger := testutil.Logger(t)
	fDest := newFakeLogDest()
	uut := NewLogSender(logger)

	t0 := dbtime.Now()
	ls1 := uuid.UUID{0x11}

	uut.Enqueue(ls1,
		Log{
			CreatedAt: t0,
			Output:    "test log 0, src 1\x00\xc3\x28",
			Level:     codersdk.LogLevelInfo,
		},
		Log{
			CreatedAt: t0,
			Output:    "test log 1, src 1",
			Level:     codersdk.LogLevelInfo,
		})

	loopErr := make(chan error, 1)
	go func() {
		err := uut.SendLoop(ctx, fDest)
		loopErr <- err
	}()

	req := testutil.TryReceive(ctx, t, fDest.reqs)
	require.NotNil(t, req)
	require.Len(t, req.Logs, 2, "it should sanitize invalid output, but still send")
	// The sanitizer replaces the NUL byte and invalid UTF-8 with ❌ while
	// preserving the valid "(" byte that follows 0xc3.
	require.Equal(t, "test log 0, src 1❌❌(", req.Logs[0].GetOutput())
	require.Equal(t, proto.Log_INFO, req.Logs[0].GetLevel())
	require.Equal(t, "test log 1, src 1", req.Logs[1].GetOutput())
	require.Equal(t, proto.Log_INFO, req.Logs[1].GetLevel())
	testutil.RequireSend(ctx, t, fDest.resps, &proto.BatchCreateLogsResponse{})

	cancel()
	err := testutil.TryReceive(testCtx, t, loopErr)
	require.ErrorIs(t, err, context.Canceled)
}

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