118 lines
3.2 KiB
Go
118 lines
3.2 KiB
Go
package api
|
|
|
|
import (
|
|
"archive/zip"
|
|
"bytes"
|
|
"encoding/json"
|
|
"io"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"rawmark/internal/vault"
|
|
)
|
|
|
|
func TestImportNotesAPI(t *testing.T) {
|
|
f := newFixture(t)
|
|
send := func(client *http.Client, token, dir, folder, name string, data []byte) *http.Response {
|
|
t.Helper()
|
|
var body bytes.Buffer
|
|
form := multipart.NewWriter(&body)
|
|
if err := form.WriteField("dir", dir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := form.WriteField("folder", folder); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
file, err := form.CreateFormFile("files", name)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := file.Write(data); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := form.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
req, err := http.NewRequest("POST", f.url+"/api/v1/import", &body)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
req.Header.Set("Content-Type", form.FormDataContentType())
|
|
if token != "" {
|
|
req.Header.Set("Authorization", "Bearer "+token)
|
|
}
|
|
res, err := client.Do(req)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return res
|
|
}
|
|
readToken := decode[struct{ Token string }](t, f.do(f.admin, "POST", "/api/v1/tokens", `{"name":"只读","scope":"read"}`, nil)).Token
|
|
for _, tc := range []struct {
|
|
client *http.Client
|
|
token string
|
|
status int
|
|
}{{newClient(), "", 401}, {newClient(), readToken, 403}} {
|
|
res := send(tc.client, tc.token, "", "", "n.md", []byte("正文"))
|
|
res.Body.Close()
|
|
if res.StatusCode != tc.status {
|
|
t.Fatalf("权限状态不符: %d", res.StatusCode)
|
|
}
|
|
}
|
|
res := send(f.admin, "", "", "", "n.md", []byte("正文"))
|
|
if res.StatusCode != 201 {
|
|
t.Fatalf("导入失败: %d", res.StatusCode)
|
|
}
|
|
var result vault.ImportResult
|
|
if err := json.NewDecoder(res.Body).Decode(&result); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
res.Body.Close()
|
|
if result.Notes != 1 || len(result.Paths) != 1 || result.Paths[0] != "n.md" || result.Warnings == nil {
|
|
t.Fatalf("响应不符: %+v", result)
|
|
}
|
|
res = send(f.admin, "", "", "", "n.md", []byte("覆盖"))
|
|
res.Body.Close()
|
|
if res.StatusCode != 409 {
|
|
t.Fatalf("应拒绝冲突: %d", res.StatusCode)
|
|
}
|
|
data, err := os.ReadFile(filepath.Join(f.root, "n.md"))
|
|
if err != nil || string(data) != "正文" {
|
|
t.Fatal("原文被覆盖", err)
|
|
}
|
|
writeToken := decode[struct{ Token string }](t, f.do(f.admin, "POST", "/api/v1/tokens", `{"name":"写入","scope":"write"}`, nil)).Token
|
|
var b bytes.Buffer
|
|
zw := zip.NewWriter(&b)
|
|
zf, err := zw.Create("子目录/笔记.md")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := io.WriteString(zf, "# ZIP"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := zw.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
res = send(newClient(), writeToken, "", "知识库", "export.md.zip", b.Bytes())
|
|
res.Body.Close()
|
|
if res.StatusCode != 201 {
|
|
t.Fatalf("写入令牌导入失败: %d", res.StatusCode)
|
|
}
|
|
for _, tc := range []struct {
|
|
dir, folder, name, data string
|
|
status int
|
|
}{
|
|
{"", "", "x.zip", "invalid", 400}, {"不存在", "", "x.md", "x", 404}, {"n.md", "", "x.md", "x", 400}, {"", "", "x.md", strings.Repeat("a", (20<<20)+1), 413},
|
|
} {
|
|
res := send(f.admin, "", tc.dir, tc.folder, tc.name, []byte(tc.data))
|
|
res.Body.Close()
|
|
if res.StatusCode != tc.status {
|
|
t.Fatalf("%s 状态不符: %d", tc.name, res.StatusCode)
|
|
}
|
|
}
|
|
}
|