restart_safety_test.go RawHistoryFolder

1package cef
2
3import (
4	"os"
5	"path/filepath"
6	"testing"
7
8	"github.com/stretchr/testify/require"
9)
10
11func TestConsumeNextStartSafetyMarkerDisablesVAAPIWithoutChangingVulkanStack(t *testing.T) {
12	root := t.TempDir()
13	t.Setenv(cefEnableVAAPIEnvVar, "1")
14	t.Setenv(cefChromiumFlagsEnvVar, "--ignore-gpu-blocklist --disable-gpu-driver-bug-workaround")
15	t.Setenv(dumberRenderStackEnvVar, renderStackVulkanDMABUF)
16
17	require.NoError(t, writeNextStartSafetyMarker(root))
18	require.True(t, consumeNextStartSafetyMarker(root))
19	applyNextStartSafetyEnvironment()
20
21	require.Equal(t, "0", os.Getenv(cefEnableVAAPIEnvVar))
22	require.Empty(t, os.Getenv(cefChromiumFlagsEnvVar))
23	require.Equal(t, renderStackVulkanDMABUF, os.Getenv(dumberRenderStackEnvVar))
24	_, err := os.Stat(filepath.Join(root, nextStartSafetyMarkerFile))
25	require.ErrorIs(t, err, os.ErrNotExist)
26}
27
28func TestGPUProcessRelaunchMarksNextStartSafeAfterRepeatedRelaunch(t *testing.T) {
29	root := t.TempDir()
30	engine := &Engine{stateRoot: root}
31	handler := &dumberBPH{engine: engine}
32	commandLine := relaunchCommandLineStub{commandLineString: "dumber --type=gpu-process"}
33
34	require.Zero(t, handler.OnAlreadyRunningAppRelaunch(commandLine, ""))
35	require.NoFileExists(t, filepath.Join(root, nextStartSafetyMarkerFile))
36	require.Zero(t, handler.OnAlreadyRunningAppRelaunch(commandLine, ""))
37	require.FileExists(t, filepath.Join(root, nextStartSafetyMarkerFile))
38}