Function TestHandleRecordingStop_WithThumbnail

agent/x/agentdesktop/api_test.go:1328–1376

(t *testing.T)

}

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

	logger := slogtest.Make(t, nil)
	// Create a fake JPEG header: 0xFF 0xD8 0xFF followed by 509 zero bytes.
	thumbnail := make([]byte, 512)
	thumbnail[0] = 0xff
	thumbnail[1] = 0xd8
	thumbnail[2] = 0xff

	fake := &fakeDesktop{
		startCfg:      agentdesktop.DisplayConfig{Width: 1920, Height: 1080},
		thumbnailData: thumbnail,
	}
	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.
	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, 2, "expected exactly 2 parts (video + thumbnail)")

	// The fake writes "fake-mp4-data-<id>-<counter>" as the MP4 content.
	expectedMP4 := []byte("fake-mp4-data-" + recID + "-1")
	assert.Equal(t, expectedMP4, parts["video/mp4"])
	assert.Equal(t, thumbnail, parts["image/jpeg"])
}

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