Function TestRecordingStartAfterCompleted

agent/x/agentdesktop/api_test.go:1153–1204

(t *testing.T)

}

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

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

	handler := api.Routes()

	// Step 1: Start recording.
	startBody, err := json.Marshal(map[string]string{"recording_id": testRecIDRestart})
	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)

	// Step 2: Stop recording (gets first MP4 data).
	stopBody, err := json.Marshal(map[string]string{"recording_id": testRecIDRestart})
	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)
	firstParts := parseMultipartParts(t, rr.Header().Get("Content-Type"), rr.Body.Bytes())
	firstData := firstParts["video/mp4"]
	require.NotEmpty(t, firstData)

	// Step 3: Start again with the same ID - should succeed
	// (old file discarded, new recording started).
	rr = httptest.NewRecorder()
	req = httptest.NewRequest(http.MethodPost, "/recording/start", bytes.NewReader(startBody))
	handler.ServeHTTP(rr, req)
	require.Equal(t, http.StatusOK, rr.Code)

	// Step 4: Stop again - should return NEW MP4 data.
	rr = httptest.NewRecorder()
	req = httptest.NewRequest(http.MethodPost, "/recording/stop", bytes.NewReader(stopBody))
	handler.ServeHTTP(rr, req)
	require.Equal(t, http.StatusOK, rr.Code)
	secondParts := parseMultipartParts(t, rr.Header().Get("Content-Type"), rr.Body.Bytes())
	secondData := secondParts["video/mp4"]
	require.NotEmpty(t, secondData)

	// The two recordings should have different data because the
	// fake increments a counter on each fresh start.
	assert.NotEqual(t, firstData, secondData,
		"restarted recording should produce different data")
}

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