refactor: flatten repository structure to root and archive research notes

internal/api/folder_download_test.go

@@ -0,0 +1,100 @@
+package api
+
+import (
+	"archive/zip"
+	"bytes"
+	"io"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"path/filepath"
+	"reflect"
+	"testing"
+
+	"chfs-ng/internal/auth"
+	"chfs-ng/internal/storage"
+)
+
+func TestFolderDownloadModes(t *testing.T) {
+	root := t.TempDir()
+	if err := os.MkdirAll(filepath.Join(root, "parent", "child"), 0o755); err != nil {
+		t.Fatal(err)
+	}
+	for _, p := range []string{
+		filepath.Join(root, "parent", "a.txt"),
+		filepath.Join(root, "parent", "child", "b.txt"),
+	} {
+		if err := os.WriteFile(p, []byte("x"), 0o644); err != nil {
+			t.Fatal(err)
+		}
+	}
+	mk := func(mode string) *Server {
+		return &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: mode,
+		}
+	}
+	for _, tc := range []struct {
+		mode, path string
+		want       int
+	}{
+		{"disable", "parent", http.StatusForbidden},
+		{"leaf", "parent", http.StatusForbidden},
+		{"leaf", "parent/child", http.StatusOK},
+		{"enable", "parent", http.StatusOK},
+	} {
+		t.Run(tc.mode+"-"+tc.path, func(t *testing.T) {
+			req := httptest.NewRequest(http.MethodGet, "/chfs/downloaddir/"+tc.path, nil)
+			w := httptest.NewRecorder()
+			mk(tc.mode).Handler().ServeHTTP(w, req)
+			if w.Code != tc.want {
+				t.Fatalf("status=%d want=%d body=%s", w.Code, tc.want, w.Body.String())
+			}
+		})
+	}
+}
+
+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)
+	}
+}