@x47base/pocketbase-addon 0.1.0

package/web/static_test.go

@@ -0,0 +1,48 @@
+package web_test
+
+import (
+	"github.com/pocketbase/pocketbase/apis"
+	"github.com/pocketbase/pocketbase/core"
+	"github.com/spink-dev/pocketbase-extension/web"
+	"net/http/httptest"
+	"testing"
+	"testing/fstest"
+)
+
+func TestStaticWithOptions(t *testing.T) {
+	fsys := fstest.MapFS{"index.html": {Data: []byte("index")}, "200.html": {Data: []byte("fallback")}, "asset.txt": {Data: []byte("asset")}}
+	for _, tc := range []struct {
+		path, fallback, want string
+		fail                 bool
+	}{
+		{"missing/route", "200.html", "fallback", false},
+		{"asset.txt", "200.html", "asset", false},
+		{"missing", "", "", true},
+		{"missing", "absent.html", "", true},
+	} {
+		t.Run(tc.path+tc.fallback, func(t *testing.T) {
+			e := &core.RequestEvent{}
+			e.Request = httptest.NewRequest("GET", "/"+tc.path, nil)
+			e.Request.SetPathValue(apis.StaticWildcardParam, tc.path)
+			w := httptest.NewRecorder()
+			e.Response = w
+			err := web.StaticWithOptions(fsys, web.StaticOptions{Fallback: tc.fallback})(e)
+			if (err != nil) != tc.fail {
+				t.Fatalf("unexpected error: %v", err)
+			}
+			if !tc.fail && w.Body.String() != tc.want {
+				t.Fatalf("body %q", w.Body.String())
+			}
+		})
+	}
+	for _, path := range []string{"/absolute", "../outside", "a/../b", `a\b`} {
+		t.Run(path, func(t *testing.T) {
+			defer func() {
+				if recover() == nil {
+					t.Fatal("accepted invalid fallback")
+				}
+			}()
+			web.StaticWithOptions(fsys, web.StaticOptions{Fallback: path})
+		})
+	}
+}