mirror of
https://github.com/dunglas/frankenphp
synced 2024-11-22 07:40:16 +00:00
108 lines
3.5 KiB
Go
108 lines
3.5 KiB
Go
package frankenphp_test
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/dunglas/frankenphp"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestWorker(t *testing.T) {
|
|
runTest(t, func(handler func(http.ResponseWriter, *http.Request), _ *httptest.Server, i int) {
|
|
formData := url.Values{"baz": {"bat"}}
|
|
req := httptest.NewRequest("POST", "http://example.com/worker.php?foo=bar", strings.NewReader(formData.Encode()))
|
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
|
w := httptest.NewRecorder()
|
|
handler(w, req)
|
|
|
|
resp := w.Result()
|
|
body, _ := io.ReadAll(resp.Body)
|
|
|
|
assert.Contains(t, string(body), fmt.Sprintf("Requests handled: %d", i*2))
|
|
|
|
formData2 := url.Values{"baz2": {"bat2"}}
|
|
req2 := httptest.NewRequest("POST", "http://example.com/worker.php?foo2=bar2", strings.NewReader(formData2.Encode()))
|
|
req2.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
|
|
|
w2 := httptest.NewRecorder()
|
|
handler(w2, req2)
|
|
|
|
resp2 := w2.Result()
|
|
body2, _ := io.ReadAll(resp2.Body)
|
|
|
|
assert.Contains(t, string(body2), fmt.Sprintf("Requests handled: %d", i*2+1))
|
|
}, &testOptions{workerScript: "worker.php", nbWorkers: 1, nbParrallelRequests: 1})
|
|
}
|
|
|
|
func TestWorkerDie(t *testing.T) {
|
|
runTest(t, func(handler func(http.ResponseWriter, *http.Request), _ *httptest.Server, i int) {
|
|
req := httptest.NewRequest("GET", "http://example.com/die.php", nil)
|
|
w := httptest.NewRecorder()
|
|
handler(w, req)
|
|
}, &testOptions{workerScript: "die.php", nbWorkers: 1, nbParrallelRequests: 10})
|
|
}
|
|
|
|
func TestNonWorkerModeAlwaysWorks(t *testing.T) {
|
|
runTest(t, func(handler func(http.ResponseWriter, *http.Request), _ *httptest.Server, i int) {
|
|
req := httptest.NewRequest("GET", "http://example.com/index.php", nil)
|
|
w := httptest.NewRecorder()
|
|
handler(w, req)
|
|
|
|
resp := w.Result()
|
|
body, _ := io.ReadAll(resp.Body)
|
|
|
|
assert.Contains(t, string(body), "I am by birth a Genevese")
|
|
}, &testOptions{workerScript: "phpinfo.php"})
|
|
}
|
|
|
|
func TestCannotCallHandleRequestInNonWorkerMode(t *testing.T) {
|
|
runTest(t, func(handler func(http.ResponseWriter, *http.Request), _ *httptest.Server, i int) {
|
|
req := httptest.NewRequest("GET", "http://example.com/non-worker.php", nil)
|
|
w := httptest.NewRecorder()
|
|
handler(w, req)
|
|
|
|
resp := w.Result()
|
|
body, _ := io.ReadAll(resp.Body)
|
|
|
|
assert.Contains(t, string(body), "<b>Fatal error</b>: Uncaught RuntimeException: frankenphp_handle_request() called while not in worker mode")
|
|
}, nil)
|
|
}
|
|
|
|
func TestWorkerEnv(t *testing.T) {
|
|
runTest(t, func(handler func(http.ResponseWriter, *http.Request), _ *httptest.Server, i int) {
|
|
req := httptest.NewRequest("GET", fmt.Sprintf("http://example.com/env.php?i=%d", i), nil)
|
|
w := httptest.NewRecorder()
|
|
handler(w, req)
|
|
|
|
resp := w.Result()
|
|
body, _ := io.ReadAll(resp.Body)
|
|
|
|
assert.Equal(t, fmt.Sprintf("bar%d", i), string(body))
|
|
}, &testOptions{workerScript: "env.php", nbWorkers: 1, env: map[string]string{"FOO": "bar"}, nbParrallelRequests: 10})
|
|
}
|
|
|
|
func ExampleServeHTTP_workers() {
|
|
if err := frankenphp.Init(
|
|
frankenphp.WithWorkers("worker1.php", 4, map[string]string{"ENV1": "foo"}),
|
|
frankenphp.WithWorkers("worker2.php", 2, map[string]string{"ENV2": "bar"}),
|
|
); err != nil {
|
|
panic(err)
|
|
}
|
|
defer frankenphp.Shutdown()
|
|
|
|
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
req := frankenphp.NewRequestWithContext(r, "/path/to/document/root", nil)
|
|
if err := frankenphp.ServeHTTP(w, req); err != nil {
|
|
panic(err)
|
|
}
|
|
})
|
|
log.Fatal(http.ListenAndServe(":8080", nil))
|
|
}
|