mirror of
https://github.com/dunglas/frankenphp.git
synced 2025-12-24 13:38:11 +08:00
8341cc98c6
* refactor: rely on context.Context for log/slog and others * optimize * refactor * Apply suggestion from @Copilot Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> * fix watcher-skip * better globals handling * fix --------- Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
85 lines
2.4 KiB
Go
85 lines
2.4 KiB
Go
package frankenphp
|
|
|
|
import (
|
|
"io"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestWorkersExtension(t *testing.T) {
|
|
readyWorkers := 0
|
|
shutdownWorkers := 0
|
|
serverStarts := 0
|
|
serverShutDowns := 0
|
|
|
|
externalWorkers, o := WithExtensionWorkers(
|
|
"extensionWorkers",
|
|
"testdata/worker.php",
|
|
1,
|
|
WithWorkerOnReady(func(id int) {
|
|
readyWorkers++
|
|
}),
|
|
WithWorkerOnShutdown(func(id int) {
|
|
serverShutDowns++
|
|
}),
|
|
WithWorkerOnServerStartup(func() {
|
|
serverStarts++
|
|
}),
|
|
WithWorkerOnServerShutdown(func() {
|
|
shutdownWorkers++
|
|
}),
|
|
)
|
|
|
|
require.NoError(t, Init(o))
|
|
t.Cleanup(func() {
|
|
Shutdown()
|
|
assert.Equal(t, 1, shutdownWorkers, "Worker shutdown hook should have been called")
|
|
assert.Equal(t, 1, serverShutDowns, "Server shutdown hook should have been called")
|
|
})
|
|
|
|
assert.Equal(t, readyWorkers, 1, "Worker thread should have called onReady()")
|
|
assert.Equal(t, serverStarts, 1, "Server start hook should have been called")
|
|
assert.Equal(t, externalWorkers.NumThreads(), 1, "NumThreads() should report 1 thread")
|
|
|
|
// Create a test request
|
|
req := httptest.NewRequest("GET", "https://example.com/test/?foo=bar", nil)
|
|
req.Header.Set("X-Test-Header", "test-value")
|
|
w := httptest.NewRecorder()
|
|
|
|
// Inject the request into the worker through the extension
|
|
err := externalWorkers.SendRequest(w, req)
|
|
assert.NoError(t, err, "Sending request should not produce an error")
|
|
|
|
resp := w.Result()
|
|
body, _ := io.ReadAll(resp.Body)
|
|
|
|
// The worker.php script should output information about the request
|
|
// We're just checking that we got a response, not the specific content
|
|
assert.NotEmpty(t, body, "Response body should not be empty")
|
|
assert.Contains(t, string(body), "Requests handled: 0", "Response body should contain request information")
|
|
}
|
|
|
|
func TestWorkerExtensionSendMessage(t *testing.T) {
|
|
externalWorker, o := WithExtensionWorkers("extensionWorkers", "testdata/message-worker.php", 1)
|
|
|
|
err := Init(o)
|
|
require.NoError(t, err)
|
|
t.Cleanup(Shutdown)
|
|
|
|
ret, err := externalWorker.SendMessage(t.Context(), "Hello Workers", nil)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, "received message: Hello Workers", ret)
|
|
}
|
|
|
|
func TestErrorIf2WorkersHaveSameName(t *testing.T) {
|
|
_, o1 := WithExtensionWorkers("duplicateWorker", "testdata/worker.php", 1)
|
|
_, o2 := WithExtensionWorkers("duplicateWorker", "testdata/worker2.php", 1)
|
|
|
|
t.Cleanup(Shutdown)
|
|
require.Error(t, Init(o1, o2))
|
|
}
|