frankenphp/embed_test.go at main · php/frankenphp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package frankenphp

import (
	"archive/tar"
	"bytes"
	"os"
	"path/filepath"
	"sync"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func setEmbeddedApp(t *testing.T, files map[string]string) {
	t.Helper()

	var buf bytes.Buffer
	tw := tar.NewWriter(&buf)
	require.NoError(t, tw.WriteHeader(&tar.Header{Name: "public/", Typeflag: tar.TypeDir, Mode: 0o755}))
	for name, content := range files {
		require.NoError(t, tw.WriteHeader(&tar.Header{Name: name, Typeflag: tar.TypeReg, Mode: 0o644, Size: int64(len(content))}))
		_, err := tw.Write([]byte(content))
		require.NoError(t, err)
	}
	require.NoError(t, tw.Close())

	previous := embeddedApp
	embeddedApp = buf.Bytes()
	t.Cleanup(func() { embeddedApp = previous })
}

func TestPrepareDefaultEmbeddedAppPathExtractsAndReuses(t *testing.T) {
	setEmbeddedApp(t, map[string]string{"public/index.php": "<?php echo 'ok';"})
	path := filepath.Join(t.TempDir(), "app")

	require.NoError(t, prepareDefaultEmbeddedAppPath(path))
	assert.FileExists(t, filepath.Join(path, "public", "index.php"))

	require.NoError(t, prepareDefaultEmbeddedAppPath(path))
	require.NoError(t, untar(path))
	content, err := os.ReadFile(filepath.Join(path, "public", "index.php"))
	require.NoError(t, err)
	assert.Equal(t, "<?php echo 'ok';", string(content))
}

func TestPrepareDefaultEmbeddedAppPathConcurrently(t *testing.T) {
	setEmbeddedApp(t, map[string]string{"public/index.php": "<?php echo 'ok';"})
	parent := t.TempDir()
	path := filepath.Join(parent, "app")

	var wg sync.WaitGroup
	errs := make([]error, 16)
	for i := range errs {
		wg.Go(func() { errs[i] = prepareDefaultEmbeddedAppPath(path) })
	}
	wg.Wait()

	for _, err := range errs {
		assert.NoError(t, err)
	}
	content, err := os.ReadFile(filepath.Join(path, "public", "index.php"))
	require.NoError(t, err)
	assert.Equal(t, "<?php echo 'ok';", string(content))

	entries, err := os.ReadDir(parent)
	require.NoError(t, err)
	assert.Len(t, entries, 1, "temporary extraction directories must be removed")
}

func TestPrepareDefaultEmbeddedAppPathRejectsSymlink(t *testing.T) {
	setEmbeddedApp(t, map[string]string{"public/index.php": "<?php echo 'ok';"})
	parent := t.TempDir()

	target := filepath.Join(parent, "target")
	require.NoError(t, os.Mkdir(target, 0o700))
	link := filepath.Join(parent, "link")
	if err := os.Symlink(target, link); err != nil {
		t.Skipf("cannot create symlinks: %v", err)
	}

	assert.Error(t, prepareDefaultEmbeddedAppPath(link))
}