test: verify folder download archive contents

This commit is contained in:
OpenClaw Agent
2026-09-22 00:18:22 +08:00
parent 0bdc206d6c
commit 58ba7c716e
@@ -1,10 +1,14 @@
package api package api
import ( import (
"archive/zip"
"bytes"
"io"
"net/http" "net/http"
"net/http/httptest" "net/http/httptest"
"os" "os"
"path/filepath" "path/filepath"
"reflect"
"testing" "testing"
"chfs-ng/internal/auth" "chfs-ng/internal/auth"
@@ -53,3 +57,44 @@ func TestFolderDownloadModes(t *testing.T) {
}) })
} }
} }
func TestFolderDownloadZipContainsOnlyReadableFiles(t *testing.T) {
root := t.TempDir()
if err := os.MkdirAll(filepath.Join(root, "parent", "child"), 0o755); err != nil {
t.Fatal(err)
}
if err := os.WriteFile(filepath.Join(root, "parent", "a.txt"), []byte("A"), 0o644); err != nil {
t.Fatal(err)
}
if err := os.WriteFile(filepath.Join(root, "parent", "child", "b.txt"), []byte("B"), 0o644); err != nil {
t.Fatal(err)
}
s := &Server{Store: storage.Local{Root: root}, Auth: &auth.Manager{Accounts: map[string]*auth.Account{}, Guest: &auth.Account{Mask: auth.Read | auth.Write | auth.Delete}}, FolderDownload: "enable"}
r := httptest.NewRequest(http.MethodGet, "/chfs/downloaddir/parent", nil)
w := httptest.NewRecorder()
s.Handler().ServeHTTP(w, r)
if w.Code != http.StatusOK {
t.Fatalf("status=%d", w.Code)
}
zr, err := zip.NewReader(bytes.NewReader(w.Body.Bytes()), int64(w.Body.Len()))
if err != nil {
t.Fatalf("invalid zip: %v", err)
}
got := map[string]string{}
for _, f := range zr.File {
body, err := f.Open()
if err != nil {
t.Fatal(err)
}
b, err := io.ReadAll(body)
body.Close()
if err != nil {
t.Fatal(err)
}
got[f.Name] = string(b)
}
want := map[string]string{"a.txt": "A", "child/b.txt": "B"}
if !reflect.DeepEqual(got, want) {
t.Fatalf("zip entries=%#v want=%#v", got, want)
}
}