Function TestHandleRecordingStop_OversizedThumbnail

agent/x/agentdesktop/api_test.go:1419–1465

(t *testing.T)

}

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

	logger := slogtest.Make(t, nil)
	// Create thumbnail data that exceeds MaxThumbnailSize.
	oversizedThumb := make([]byte, workspacesdk.MaxThumbnailSize+1)
	oversizedThumb[0] = 0xff
	oversizedThumb[1] = 0xd8
	oversizedThumb[2] = 0xff

	fake := &fakeDesktop{
		startCfg:      agentdesktop.DisplayConfig{Width: 1920, Height: 1080},
		thumbnailData: oversizedThumb,
	}
	api := agentdesktop.NewAPI(logger, fake, nil)
	defer api.Close()

	handler := api.Routes()

	// Start recording.
	recID := uuid.New().String()
	startBody, err := json.Marshal(map[string]string{"recording_id": recID})
	require.NoError(t, err)
	rr := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodPost, "/recording/start", bytes.NewReader(startBody))
	handler.ServeHTTP(rr, req)
	require.Equal(t, http.StatusOK, rr.Code)

	// Stop recording.
	stopBody, err := json.Marshal(map[string]string{"recording_id": recID})
	require.NoError(t, err)
	rr = httptest.NewRecorder()
	req = httptest.NewRequest(http.MethodPost, "/recording/stop", bytes.NewReader(stopBody))
	handler.ServeHTTP(rr, req)
	require.Equal(t, http.StatusOK, rr.Code)

	// Verify multipart response contains only the video part.
	ct := rr.Header().Get("Content-Type")
	assert.True(t, strings.HasPrefix(ct, "multipart/mixed"),
		"expected multipart/mixed Content-Type, got %s", ct)

	parts := parseMultipartParts(t, ct, rr.Body.Bytes())
	assert.Len(t, parts, 1, "expected exactly 1 part (video only, oversized thumbnail discarded)")

	expectedMP4 := []byte("fake-mp4-data-" + recID + "-1")
	assert.Equal(t, expectedMP4, parts["video/mp4"])
}