250 lines
6.4 KiB
Go
250 lines
6.4 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"path"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestServerRoutes(t *testing.T) {
|
|
server := NewServer(t.TempDir())
|
|
if err := server.Routes(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := ensureAndWrite(path.Join(server.root, "ui", "files.ctmpl"), []byte(`{{ define "files" }}{{ template "_import" }}HI FROM FILES{{ end }}`)); err != nil {
|
|
t.Fatal(err)
|
|
} else if err := ensureAndWrite(path.Join(server.root, "ui", "search.ctmpl"), []byte(`{{ define "search" }}{{ template "_import" }}HI FROM SEARCH{{ end }}`)); err != nil {
|
|
t.Fatal(err)
|
|
} else if err := ensureAndWrite(path.Join(server.root, "ui", "templates", "_import.ctmpl"), []byte(`{{ define "_import" }}HI FROM IMPORT{{ end }}`)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := ensureAndWrite(server.diskMediaPath("id"), []byte("hi")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
ensureAndWrite(server.diskMediaPath("delid"), []byte("hi"))
|
|
|
|
tree := server.tree()
|
|
if err := tree.Put(NewID("getfid"), Leaf{Title: "", Content: "getfid body"}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tree.Put(NewID("putfid"), Leaf{Title: "putfid title", Content: "initial putfid body"})
|
|
tree.Put(NewID("delfid"), Leaf{Title: "delfid title", Content: "delfid body"})
|
|
t.Log(tree.GetRoot())
|
|
|
|
ensureAndWrite(path.Join(server.root, "index.html"), []byte("mom"))
|
|
|
|
cases := map[string]struct {
|
|
path string
|
|
method string
|
|
body string
|
|
want string
|
|
}{
|
|
"v0: /: get": {
|
|
path: "/",
|
|
method: http.MethodGet,
|
|
want: "/ui/files/",
|
|
},
|
|
"v0: search: get": {
|
|
path: "/api/v0/search?q=getf%20bod",
|
|
method: http.MethodGet,
|
|
want: `["getfid"]`,
|
|
},
|
|
"v0: tree: get": {
|
|
path: "/api/v0/tree",
|
|
method: http.MethodGet,
|
|
want: `"Title":"Untitled",`,
|
|
},
|
|
"v0: media: post": {
|
|
path: "/api/v0/media",
|
|
method: http.MethodPost,
|
|
want: `{"data":{"filePath":"/api/v0/media/`,
|
|
},
|
|
"v0: media id: del": {
|
|
path: "/api/v0/media/delid",
|
|
method: http.MethodDelete,
|
|
},
|
|
"v0: media id: get": {
|
|
path: "/api/v0/media/id",
|
|
method: http.MethodGet,
|
|
want: "hi",
|
|
},
|
|
"v0: files: post": {
|
|
path: "/api/v0/files",
|
|
method: http.MethodPost,
|
|
},
|
|
"v0: files id: get": {
|
|
path: "/api/v0/files/getfid",
|
|
method: http.MethodGet,
|
|
want: "getfid body",
|
|
},
|
|
"v0: files id: put": {
|
|
path: "/api/v0/files/putfid",
|
|
method: http.MethodPut,
|
|
body: "putfid body",
|
|
},
|
|
"v0: files id: del": {
|
|
path: "/api/v0/files/delfid",
|
|
method: http.MethodDelete,
|
|
},
|
|
"v0: /: redir": {
|
|
path: "/",
|
|
method: http.MethodGet,
|
|
want: "/ui/files/",
|
|
},
|
|
"v0: /ui/: redir": {
|
|
path: "/ui/",
|
|
method: http.MethodGet,
|
|
want: "/ui/files/",
|
|
},
|
|
"v0: /ui: redir": {
|
|
path: "/ui",
|
|
method: http.MethodGet,
|
|
want: "/ui/files/",
|
|
},
|
|
"v0: /ui/search": {
|
|
path: "/ui/search",
|
|
method: http.MethodGet,
|
|
},
|
|
"v0: /ui/search?q=abc": {
|
|
path: "/ui/search?q=abc",
|
|
method: http.MethodGet,
|
|
},
|
|
"v0: /ui/files/getfid": {
|
|
path: "/ui/files/getfid",
|
|
method: http.MethodGet,
|
|
},
|
|
"v0: /ui/files": {
|
|
path: "/ui/files",
|
|
method: http.MethodGet,
|
|
},
|
|
}
|
|
|
|
for name, d := range cases {
|
|
c := d
|
|
t.Run(name, func(t *testing.T) {
|
|
r := httptest.NewRequest(c.method, c.path, strings.NewReader(c.body))
|
|
w := httptest.NewRecorder()
|
|
server.ServeHTTP(w, r)
|
|
if w.Code == http.StatusNotFound {
|
|
t.Fatal(w)
|
|
}
|
|
if w.Code >= 400 {
|
|
t.Fatal(w)
|
|
}
|
|
if len(c.want) > 0 && !strings.Contains(string(w.Body.Bytes()), c.want) {
|
|
t.Fatal(w)
|
|
}
|
|
t.Logf("%s %s (%+v) => %s", c.method, c.path, w.Header(), w.Body.Bytes())
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestServerPutTreeGetFile(t *testing.T) {
|
|
server := NewServer(t.TempDir())
|
|
if err := server.Routes(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
server.tree().Put(NewID("my pid"), Leaf{})
|
|
var id string
|
|
t.Run("put to create an id", func(t *testing.T) {
|
|
r := httptest.NewRequest(http.MethodPut, "/my%20pid/my-put-id", strings.NewReader("body"))
|
|
r.Header.Set("Title", "my put title")
|
|
w := httptest.NewRecorder()
|
|
if err := server.apiV0FilesIDPutHandler(w, r); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Logf("%s", w.Body.Bytes())
|
|
if w.Code != http.StatusOK {
|
|
t.Fatal(w)
|
|
}
|
|
var resp struct {
|
|
Data struct {
|
|
FilePath string `json:"filePath"`
|
|
} `json:"data"`
|
|
}
|
|
if err := json.NewDecoder(w.Body).Decode(&resp); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if path.Base(resp.Data.FilePath) != "my-put-id" {
|
|
t.Fatal(resp.Data.FilePath)
|
|
}
|
|
})
|
|
t.Run("post", func(t *testing.T) {
|
|
r := httptest.NewRequest(http.MethodPost, "/my%20pid", strings.NewReader("body"))
|
|
r.Header.Set("Title", "my title")
|
|
w := httptest.NewRecorder()
|
|
if err := server.apiV0FilesHandler(w, r); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Logf("%s", w.Body.Bytes())
|
|
if w.Code != http.StatusOK {
|
|
t.Fatal(w)
|
|
}
|
|
var resp struct {
|
|
Data struct {
|
|
FilePath string `json:"filePath"`
|
|
} `json:"data"`
|
|
}
|
|
if err := json.NewDecoder(w.Body).Decode(&resp); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
id = strings.TrimPrefix(resp.Data.FilePath, "/api/v0/files/")
|
|
})
|
|
t.Run("tree", func(t *testing.T) {
|
|
r := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
w := httptest.NewRecorder()
|
|
if err := server.apiV0TreeHandler(w, r); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Logf("%s", w.Body.Bytes())
|
|
if w.Code != http.StatusOK {
|
|
t.Fatal(w)
|
|
}
|
|
if !bytes.Contains(w.Body.Bytes(), []byte(`{"Title":"my title","Deleted":false,"Content":"`)) {
|
|
t.Fatal(w)
|
|
}
|
|
var branch Branch
|
|
if err := json.NewDecoder(w.Body).Decode(&branch); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Logf("TODO: %+v", branch)
|
|
if branch.Leaf != (Leaf{}) {
|
|
t.Error(branch.Leaf)
|
|
}
|
|
if parent, ok := branch.Branches["my pid"]; !ok {
|
|
t.Error(ok, branch)
|
|
} else if parent.Leaf.Title != "Untitled" {
|
|
t.Error(parent.Leaf)
|
|
} else if child, ok := parent.Branches[NewID(id)]; !ok {
|
|
t.Error(ok, NewID("my pid").Push(id), parent)
|
|
} else if child.Leaf.Title != "my title" {
|
|
t.Error(child.Leaf)
|
|
}
|
|
})
|
|
t.Run("get", func(t *testing.T) {
|
|
r := httptest.NewRequest(http.MethodGet, "/"+url.PathEscape(id), nil)
|
|
t.Logf("%s", r.URL.String())
|
|
w := httptest.NewRecorder()
|
|
if err := server.apiV0FilesIDGetHandler(w, r); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Logf("%s", w.Body.Bytes())
|
|
if w.Code != http.StatusOK {
|
|
t.Fatal(w)
|
|
}
|
|
if !bytes.Contains(w.Body.Bytes(), []byte(`body`)) {
|
|
t.Fatal(w)
|
|
}
|
|
if title := w.Header().Get("Title"); title != "my title" {
|
|
t.Fatal(title)
|
|
}
|
|
})
|
|
}
|