added multipart range test

This commit is contained in:
Gani Georgiev 2022-11-29 18:12:40 +02:00
parent bd65125744
commit 2deca759fa
1 changed files with 31 additions and 1 deletions

View File

@ -9,6 +9,7 @@ import (
"net/http/httptest"
"os"
"path/filepath"
"strings"
"testing"
"github.com/pocketbase/pocketbase/tools/filesystem"
@ -299,7 +300,7 @@ func TestFileSystemServe(t *testing.T) {
}
}
func TestFileSystemServeRange(t *testing.T) {
func TestFileSystemServeSingleRange(t *testing.T) {
dir := createTestDir(t)
defer os.RemoveAll(dir)
@ -333,6 +334,35 @@ func TestFileSystemServeRange(t *testing.T) {
}
}
func TestFileSystemServeMultiRange(t *testing.T) {
dir := createTestDir(t)
defer os.RemoveAll(dir)
fs, err := filesystem.NewLocal(dir)
if err != nil {
t.Fatal(err)
}
defer fs.Close()
res := httptest.NewRecorder()
req := httptest.NewRequest("GET", "/", nil)
req.Header.Add("Range", "bytes=0-20, 25-30")
if err := fs.Serve(res, req, "image.png", "image.png"); err != nil {
t.Fatal(err)
}
result := res.Result()
if result.StatusCode != http.StatusPartialContent {
t.Fatalf("Expected StatusCode %d, got %d", http.StatusPartialContent, result.StatusCode)
}
if ct := result.Header.Get("Content-Type"); !strings.HasPrefix(ct, "multipart/byteranges; boundary=") {
t.Fatalf("Expected Content-Type to be multipart/byteranges, got %v", ct)
}
}
func TestFileSystemCreateThumb(t *testing.T) {
dir := createTestDir(t)
defer os.RemoveAll(dir)