handlers_test.go

package main

import (
    "testing"
    "io"
    "os"
    "os/user"
    "strings"
    "bytes"
    "path/filepath"
    "net/http"
    "net/http/httptest"
    "net/url"
    "encoding/json"
    "sort"
    "time"
    "strconv"
)

func TestDumpJsonResponse(t *testing.T) {
    rr := httptest.NewRecorder()
    dumpJsonResponse(rr, http.StatusBadRequest, map[string]string{ "foo": "bar" })
    if rr.Code != http.StatusBadRequest {
        t.Fatal("failed to respect the request")
    }

    var contents map[string]string
    dec := json.NewDecoder(rr.Body)
    dec.Decode(&contents)
    if contents["foo"] != "bar" {
        t.Fatalf("failed to dump JSON correctly")
    }

    headers := rr.Header()
    ct := headers["Content-Type"]
    if len(ct) != 1 || ct[0] != "application/json" {
        t.Fatalf("failed to set the content type correctly")
    }
}

func TestValidatePath(t *testing.T) {
    _, err := validatePath("")
    if err == nil || !strings.Contains(err.Error(), "empty string") {
        t.Fatalf("expected an empty string error")
    }

    _, err = validatePath("foobar")
    if err == nil || !strings.Contains(err.Error(), "absolute path") {
        t.Fatalf("expected an absolute path error")
    }

    path, err := validatePath("/whee/foobar/")
    if err != nil {
        t.Fatal(err)
    }
    if path != "/whee/foobar" {
        t.Fatalf("expected elimination of trailing slashes, got %q instead", path)
    }

    path, err = validatePath("/whee/a/../foobar/")
    if err != nil {
        t.Fatal(err)
    }
    if path != "/whee/foobar" {
        t.Fatalf("expected cleaning of the path, got %q instead", path)
    }
}

func TestCheckVerificationCode(t *testing.T) {
    v := newVerificationRegistry(time.Minute)

    t.Run("success", func(t * testing.T) {
        target, err := os.MkdirTemp("", "")
        if err != nil {
            t.Fatal(err)
        }

        code, err := v.Provision(target)
        if err != nil {
            t.Fatal(err)
        }

        err = os.WriteFile(filepath.Join(target, code), []byte{}, 0644)
        if err != nil {
            t.Fatal(err)
        }

        info, err := checkVerificationCode(target, v, 1)
        if err != nil {
            t.Fatal(err)
        }

        if info == nil || !strings.HasPrefix(info.Name(), ".sewer_") {
            t.Fatalf("unexpected file %v", info)
        }
    })

    t.Run("no code", func(t * testing.T) {
        target, err := os.MkdirTemp("", "")
        if err != nil {
            t.Fatal(err)
        }

        _, err = checkVerificationCode(target, v, 1)
        if err == nil || !strings.Contains(err.Error(), "no verification code") {
            t.Fatal("should have failed")
        }
    })

    t.Run("no file", func(t * testing.T) {
        target, err := os.MkdirTemp("", "")
        if err != nil {
            t.Fatal(err)
        }

        _, err = v.Provision(target)
        if err != nil {
            t.Fatal(err)
        }

        _, err = checkVerificationCode(target, v, 1)
        if err == nil || !strings.Contains(err.Error(), "verification failed") {
            t.Fatal("should have failed")
        }
    })

    t.Run("hard links", func(t * testing.T) {
        target, err := os.MkdirTemp("", "")
        if err != nil {
            t.Fatal(err)
        }

        code, err := v.Provision(target)
        if err != nil {
            t.Fatal(err)
        }

        err = os.WriteFile(filepath.Join(target, code), []byte{}, 0644)
        if err != nil {
            t.Fatal(err)
        }

        err = os.Link(filepath.Join(target, code), filepath.Join(target, "foo"))
        if err != nil {
            t.Fatal(err)
        }

        _, err = checkVerificationCode(target, v, 1)
        if err == nil || !strings.Contains(err.Error(), "hard link") {
            t.Fatal("should have failed")
        }
    })
}

func decodeStringyResponse(input io.Reader, t *testing.T) map[string]string {
    var output map[string]string
    dec := json.NewDecoder(input)
    err := dec.Decode(&output)
    if err != nil {
        t.Fatal(err)
    }
    return output
}

func createJsonRequest(method, endpoint string, body map[string]interface{}, t *testing.T) *http.Request {
    contents, err := json.Marshal(body)
    if err != nil {
        t.Fatal(err)
    }

    req, err := http.NewRequest(method, endpoint, bytes.NewReader(contents))
    if err != nil {
        t.Fatal(err)
    }

    return req
}

func TestRegisterHandlers(t *testing.T) {
    tmp, err := os.MkdirTemp("", "")
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer os.RemoveAll(tmp)

    dbpath := filepath.Join(tmp, "db.sqlite3")
    dbconn, err := initializeDatabase(dbpath)
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer dbconn.Close()

    verifier := newVerificationRegistry(time.Minute)

    to_add := filepath.Join(tmp, "to_add")
    err = mockDirectory(to_add)
    if err != nil {
        t.Fatalf(err.Error())
    }

    tokr, err := newUnicodeTokenizer(false)
    if err != nil {
        t.Fatalf(err.Error())
    }

    var code string
    t.Run("register start", func(t *testing.T) {
        handler := http.HandlerFunc(newRegisterStartHandler(verifier))

        {
            req := createJsonRequest("POST", "/register/start", map[string]interface{}{ "path": "foo" }, t)
            rr := httptest.NewRecorder()
            handler.ServeHTTP(rr, req)
            if rr.Code != http.StatusBadRequest {
                t.Fatalf("should have failed on a non-absolute path; %v", rr.Code)
            }

            output := decodeStringyResponse(rr.Body, t)
            if output["status"] != "ERROR" || !strings.Contains(output["reason"], "absolute") {
                t.Fatalf("unexpected body")
            }
        }

        {
            req := createJsonRequest("POST", "/register/start", map[string]interface{}{ "path": to_add }, t)
            rr := httptest.NewRecorder()
            handler.ServeHTTP(rr, req)
            if rr.Code != http.StatusAccepted {
                t.Fatalf("should have succeeded")
            }

            output := decodeStringyResponse(rr.Body, t)
            code = output["code"]
            if output["status"] != "PENDING" || !strings.HasPrefix(code, ".sewer_") {
                t.Fatalf("unexpected body")
            }
        }
    })

    t.Run("register start symlink", func(t *testing.T) {
        handler := http.HandlerFunc(newRegisterStartHandler(verifier))

        tmp, err := os.MkdirTemp("", "")
        if err != nil {
            t.Fatal(err)
        }

        to_add2 := filepath.Join(tmp, "symlink")
        err = os.Symlink(to_add, to_add2)
        if err != nil {
            t.Fatal(err)
        }

        req := createJsonRequest("POST", "/register/start", map[string]interface{}{ "path": to_add2 }, t)
        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusAccepted {
            t.Fatalf("should have succeeded")
        }
    })

    quickRegisterStart := func() string {
        handler := http.HandlerFunc(newRegisterStartHandler(verifier))
        req := createJsonRequest("POST", "/register/start", map[string]interface{}{ "path": to_add }, t)
        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusAccepted {
            t.Fatalf("should have succeeded")
        }
        output := decodeStringyResponse(rr.Body, t)
        return output["code"]
    }

    t.Run("register finish without verification", func(t *testing.T) {
        quickRegisterStart()
        handler := http.HandlerFunc(newRegisterFinishHandler(dbconn, verifier, tokr, 1))
        req := createJsonRequest("POST", "/register/finish", map[string]interface{}{ "path": to_add }, t)
        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusUnauthorized {
            t.Fatalf("should have failed due to lack of code")
        }
    })

    t.Run("register finish ok", func(t *testing.T) {
        code := quickRegisterStart()
        err := os.WriteFile(filepath.Join(to_add, code), []byte(""), 0644)
        if err != nil {
            t.Fatal(err)
        }

        handler := http.HandlerFunc(newRegisterFinishHandler(dbconn, verifier, tokr, 1))
        req := createJsonRequest("POST", "/register/finish", map[string]interface{}{ "path": to_add }, t)
        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        output := struct {
            Status string
            Comments []string
        }{}
        dec := json.NewDecoder(rr.Body)
        err = dec.Decode(&output)
        if err != nil {
            t.Fatal(err)
        }
        if output.Status != "SUCCESS" || len(output.Comments) > 0 {
            t.Fatalf("unexpected body")
        }

        all_paths, err := listPaths(dbconn, to_add)
        if err != nil {
            t.Fatal(err)
        }
        if !equalStringArrays(all_paths, []string{ "metadata.json", "stuff/metadata.json" }) {
            t.Fatalf("unexpected paths in the database %v", all_paths)
        }
    })

    t.Run("register finish with empty names", func(t *testing.T) {
        quickRegisterStart()
        handler := http.HandlerFunc(newRegisterFinishHandler(dbconn, verifier, tokr, 1))
        req := createJsonRequest("POST", "/register/finish", map[string]interface{}{ "path": to_add, "base": []string{ "", "metadata.json" } }, t)
        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusBadRequest {
            t.Fatalf("should have failed with empty names")
        }
    })

    t.Run("register finish with proper names", func(t *testing.T) {
        code := quickRegisterStart()
        err := os.WriteFile(filepath.Join(to_add, code), []byte(""), 0644)
        if err != nil {
            t.Fatal(err)
        }

        handler := http.HandlerFunc(newRegisterFinishHandler(dbconn, verifier, tokr, 1))
        req := createJsonRequest("POST", "/register/finish", map[string]interface{}{ "path": to_add, "base": []string{ "metadata.json", "other.json" } }, t)
        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        output := struct {
            Status string
            Comments []string
        }{}
        dec := json.NewDecoder(rr.Body)
        err = dec.Decode(&output)
        if err != nil {
            t.Fatal(err)
        }
        if output.Status != "SUCCESS" || len(output.Comments) > 0 {
            t.Fatalf("unexpected body")
        }

        all_paths, err := listPaths(dbconn, to_add)
        if err != nil {
            t.Fatal(err)
        }
        if !equalStringArrays(all_paths, []string{ "metadata.json", "stuff/metadata.json", "stuff/other.json", "whee/other.json" }) {
            t.Fatalf("unexpected paths in the database %v", all_paths)
        }
    })

    t.Run("register finish reuse names", func(t *testing.T) {
        {
            code := quickRegisterStart()
            err := os.WriteFile(filepath.Join(to_add, code), []byte(""), 0644)
            if err != nil {
                t.Fatal(err)
            }

            handler := http.HandlerFunc(newRegisterFinishHandler(dbconn, verifier, tokr, 1))
            rr := httptest.NewRecorder()
            req := createJsonRequest("POST", "/register/finish", map[string]interface{}{ "path": to_add, "base": []string{ "alpha.json", "bravo.json" } }, t)
            handler.ServeHTTP(rr, req)
            if rr.Code != http.StatusOK {
                t.Fatalf("should have succeeded")
            }
        }

        // Re-registering with the same names.
        {
            code := quickRegisterStart()
            err := os.WriteFile(filepath.Join(to_add, code), []byte(""), 0644)
            if err != nil {
                t.Fatal(err)
            }

            handler := http.HandlerFunc(newRegisterFinishHandler(dbconn, verifier, tokr, 1))
            rr := httptest.NewRecorder()
            req := createJsonRequest("POST", "/register/finish", map[string]interface{}{ "path": to_add }, t)
            handler.ServeHTTP(rr, req)
            if rr.Code != http.StatusOK {
                t.Fatalf("should have succeeded")
            }

            all_paths, err := listDirs(dbconn)
            if err != nil {
                t.Fatal(err)
            }
            my_names, ok := all_paths[to_add]
            if !ok || !equalStringArrays(my_names, []string{ "alpha.json", "bravo.json" }) {
                t.Fatalf("unexpected paths in the database %v", my_names)
            }
        }

        // Re-registering with different names, to check that it indeed gets overridden.
        {
            code := quickRegisterStart()
            err := os.WriteFile(filepath.Join(to_add, code), []byte(""), 0644)
            if err != nil {
                t.Fatal(err)
            }

            handler := http.HandlerFunc(newRegisterFinishHandler(dbconn, verifier, tokr, 1))
            rr := httptest.NewRecorder()
            req := createJsonRequest("POST", "/register/finish", map[string]interface{}{ "path": to_add, "base": []string{ "metadata.json" } }, t)
            handler.ServeHTTP(rr, req)
            if rr.Code != http.StatusOK {
                t.Fatalf("should have succeeded")
            }

            all_paths, err := listDirs(dbconn)
            if err != nil {
                t.Fatal(err)
            }
            my_names, ok := all_paths[to_add]
            if !ok || !equalStringArrays(my_names, []string{ "metadata.json" }) {
                t.Fatalf("unexpected paths in the database %v", my_names)
            }
        }
    })
}

func TestDeregisterHandlers(t *testing.T) {
    tmp, err := os.MkdirTemp("", "")
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer os.RemoveAll(tmp)

    dbpath := filepath.Join(tmp, "db.sqlite3")
    dbconn, err := initializeDatabase(dbpath)
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer dbconn.Close()

    verifier := newVerificationRegistry(time.Minute)

    tokr, err := newUnicodeTokenizer(false)
    if err != nil {
        t.Fatalf(err.Error())
    }

    to_add := filepath.Join(tmp, "to_add")
    err = mockDirectory(to_add)
    if err != nil {
        t.Fatalf(err.Error())
    }

    comments, err := addNewDirectory(dbconn, to_add, []string{ "metadata.json", "other.json" }, "myself", tokr)
    if err != nil {
        t.Fatal(err)
    }
    if len(comments) != 0 {
        t.Fatal("no comments should be present")
    }

    var code string
    t.Run("deregister start", func(t *testing.T) {
        handler := http.HandlerFunc(newDeregisterStartHandler(dbconn, verifier))

        {
            req := createJsonRequest("POST", "/deregister/start", map[string]interface{}{ "path": "foo" }, t)
            rr := httptest.NewRecorder()
            handler.ServeHTTP(rr, req)
            if rr.Code != http.StatusBadRequest {
                t.Fatalf("should have failed on a non-absolute path; %v", rr.Code)
            }
        }

        {
            req := createJsonRequest("POST", "/deregister/start", map[string]interface{}{ "path": to_add }, t)
            if err != nil {
                t.Fatal(err)
            }

            rr := httptest.NewRecorder()
            handler.ServeHTTP(rr, req)
            if rr.Code != http.StatusAccepted {
                t.Fatalf("should have succeeded")
            }

            output := decodeStringyResponse(rr.Body, t)
            code = output["code"]
            if output["status"] != "PENDING" || !strings.HasPrefix(code, ".sewer_") {
                t.Fatalf("unexpected body")
            }
        }
    })

    quickDeregisterStart := func() string {
        handler := http.HandlerFunc(newDeregisterStartHandler(dbconn, verifier))
        req := createJsonRequest("POST", "/deregister/start", map[string]interface{}{ "path": to_add }, t)
        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusAccepted {
            t.Fatalf("should have succeeded")
        }
        output := decodeStringyResponse(rr.Body, t)
        return output["code"]
    }

    t.Run("deregister fail", func(t *testing.T) {
        quickDeregisterStart()
        handler := http.HandlerFunc(newDeregisterFinishHandler(dbconn, verifier, 1))

        // First attempt fails, because we didn't add the registration code.
        req := createJsonRequest("POST", "/deregister/finish", map[string]interface{}{ "path": to_add }, t)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusUnauthorized {
            t.Fatalf("should have failed with a 401 (got %d instead)", rr.Code)
        }
    })

    t.Run("deregister ok", func(t *testing.T) {
        code := quickDeregisterStart()
        err := os.WriteFile(filepath.Join(to_add, code), []byte(""), 0644)
        if err != nil {
            t.Fatal(err)
        }

        handler := http.HandlerFunc(newDeregisterFinishHandler(dbconn, verifier, 1))
        req := createJsonRequest("POST", "/deregister/finish", map[string]interface{}{ "path": to_add }, t)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        output := decodeStringyResponse(rr.Body, t)
        if output["status"] != "SUCCESS" {
            t.Fatalf("unexpected body")
        }

        all_paths, err := listPaths(dbconn, to_add)
        if err != nil {
            t.Fatal(err)
        }
        if len(all_paths) != 0 {
            t.Fatalf("unexpected paths in the database %v", all_paths)
        }
    })

    // Readding the directory, and then removing it from the file system.
    comments, err = addNewDirectory(dbconn, to_add, []string{ "metadata.json", "other.json" }, "myself", tokr)
    if err != nil {
        t.Fatal(err)
    }
    if len(comments) != 0 {
        t.Fatal("no comments should be present")
    }

    err = os.RemoveAll(to_add)
    if err != nil {
        t.Fatal(err)
    }

    t.Run("deregister immediate", func(t *testing.T) {
        handler := http.HandlerFunc(newDeregisterStartHandler(dbconn, verifier))
        req := createJsonRequest("POST", "/deregister/start", map[string]interface{}{ "path": to_add }, t)
        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        output := decodeStringyResponse(rr.Body, t)
        if output["status"] != "SUCCESS" {
            t.Fatalf("unexpected body")
        }

        all_paths, err := listPaths(dbconn, to_add)
        if err != nil {
            t.Fatal(err)
        }
        if len(all_paths) != 0 {
            t.Fatalf("unexpected paths in the database %v", all_paths)
        }
    })
}

func TestQueryHandler(t *testing.T) {
    tmp, err := os.MkdirTemp("", "")
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer os.RemoveAll(tmp)

    dbpath := filepath.Join(tmp, "db.sqlite3")
    dbconn, err := initializeDatabase(dbpath)
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer dbconn.Close()

    tokr, err := newUnicodeTokenizer(false)
    if err != nil {
        t.Fatalf(err.Error())
    }

    wtokr, err := newUnicodeTokenizer(true)
    if err != nil {
        t.Fatalf(err.Error())
    }

    to_add := filepath.Join(tmp, "to_add")
    err = mockDirectory(to_add)
    if err != nil {
        t.Fatalf(err.Error())
    }

    comments, err := addNewDirectory(dbconn, to_add, []string{ "metadata.json", "other.json" }, "myself", tokr)
    if err != nil {
        t.Fatal(err)
    }
    if len(comments) != 0 {
        t.Fatal("no comments should be present")
    }

    handler := http.HandlerFunc(newQueryHandler(dbconn, tokr, wtokr, "/query"))

    self, err := user.Current()
    if err != nil {
        t.Fatal(err)
    }
    selfname := self.Username

    validateSearchResults := func(input io.Reader) ([]string, string) {
        output := struct {
            Results []map[string]interface{}
            Next string
        }{}

        dec := json.NewDecoder(input)
        err = dec.Decode(&output)
        if err != nil {
            t.Fatal(err)
        }

        all_paths := []string{}
        for _, r := range output.Results {
            path_i, ok := r["path"]
            if !ok {
                t.Fatalf("expected a path property in %v", r)
            }
            path, ok := path_i.(string)
            if !ok {
                t.Fatalf("expected a path string property in %v", r)
            }
            all_paths = append(all_paths, path)

            user_i, ok := r["user"]
            if !ok {
                t.Fatalf("expected a user property in %v", r)
            }
            user, ok := user_i.(string)
            if !ok {
                t.Fatalf("expected a user string property in %v", r)
            }
            if user != selfname {
                t.Fatalf("unexpected username %v", user)
            }

            time_i, ok := r["time"]
            if !ok {
                t.Fatalf("expected a time property in %v", r)
            }
            time, ok := time_i.(float64)
            if !ok {
                t.Fatalf("expected a time integer property in %v", r)
            }
            if time <= 0 {
                t.Fatalf("time should be positive %v", time)
            }

            meta_i, ok := r["metadata"]
            if !ok {
                t.Fatalf("expected a metadata property in %v", r)
            }
            _, ok = meta_i.(map[string]interface{})
            if !ok {
                t.Fatalf("expected a metadata object property in %v", r)
            }
        }

        sort.Strings(all_paths)
        return all_paths, output.Next
    }

    t.Run("basic", func (t *testing.T) {
        req, err := http.NewRequest("POST", "/query", strings.NewReader(`{ "type": "text", "text": "aaron" }`))
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        all_paths, scroll := validateSearchResults(rr.Body)
        if scroll != "" {
            t.Fatalf("unexpected scroll %v", scroll)
        }
        if len(all_paths) != 1 || all_paths[0] != filepath.Join(to_add, "metadata.json") {
            t.Fatalf("unexpected paths %v", all_paths)
        }
    })

    t.Run("multiple", func (t *testing.T) {
        // Check that we report multiple hits correctly.
        req, err := http.NewRequest("POST", "/query", strings.NewReader(`{ "type": "text", "text": "yuru" }`))
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        all_paths, scroll := validateSearchResults(rr.Body)
        if scroll != "" {
            t.Fatalf("unexpected scroll %v", scroll)
        }
        if !equalPathArrays(all_paths, []string{ "stuff/metadata.json", "whee/other.json" }, to_add) {
            t.Fatalf("unexpected paths %v", all_paths)
        }
    })

    t.Run("sanitized", func (t *testing.T) {
        // Check that the query is indeed sanitized.
        req, err := http.NewRequest("POST", "/query", strings.NewReader(`{ "type": "text", "text": "lamb chicken" }`))
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        all_paths, scroll := validateSearchResults(rr.Body)
        if scroll != "" {
            t.Fatalf("unexpected scroll %v", scroll)
        }
        if !equalPathArrays(all_paths, []string{ "stuff/other.json" }, to_add) {
            t.Fatalf("unexpected paths %v", all_paths)
        }
    })

    t.Run("translated", func (t *testing.T) {
        req, err := http.NewRequest("POST", "/query?translate=true", strings.NewReader(`{ "type": "text", "text": "lamb OR chicken" }`))
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        all_paths, scroll := validateSearchResults(rr.Body)
        if scroll != "" {
            t.Fatalf("unexpected scroll %v", scroll)
        }
        if !equalPathArrays(all_paths, []string{ "metadata.json", "stuff/other.json" }, to_add) {
            t.Fatalf("unexpected paths %v", all_paths)
        }
    })

    t.Run("wildcards", func (t *testing.T) {
        req, err := http.NewRequest("POST", "/query?translate=true", strings.NewReader(`{ "type": "text", "text": "l?mb OR chick*" }`))
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        all_paths, scroll := validateSearchResults(rr.Body)
        if scroll != "" {
            t.Fatalf("unexpected scroll %v", scroll)
        }
        if !equalPathArrays(all_paths, []string{ "metadata.json", "stuff/other.json" }, to_add) {
            t.Fatalf("unexpected paths %v", all_paths)
        }
    })

    t.Run("scroll", func (t *testing.T) {
        dummy_query := `{ "type": "text", "text": "   " }`

        req, err := http.NewRequest("POST", "/query?limit=2", strings.NewReader(dummy_query))
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        all_paths, scroll := validateSearchResults(rr.Body)
        if !strings.HasPrefix(scroll, "/query?scroll=") {
            t.Fatalf("unexpected scroll %v", scroll)
        }

        // Next scroll.
        req, err = http.NewRequest("POST", scroll, strings.NewReader(dummy_query))
        if err != nil {
            t.Fatal(err)
        }

        rr = httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        all_paths2, scroll := validateSearchResults(rr.Body)
        if scroll != "" { // fully exhausted the scroll now.
            t.Fatalf("unexpected scroll %v", scroll)
        }

        all_paths = append(all_paths, all_paths2...)
        sort.Strings(all_paths)
        if !equalPathArrays(all_paths, []string{ "metadata.json", "stuff/metadata.json", "stuff/other.json", "whee/other.json" }, to_add) {
            t.Fatalf("unexpected paths %v", all_paths)
        }
    })
}

func TestRetrieveMetadataHandler(t *testing.T) {
    tmp, err := os.MkdirTemp("", "")
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer os.RemoveAll(tmp)

    dbpath := filepath.Join(tmp, "db.sqlite3")
    dbconn, err := initializeDatabase(dbpath)
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer dbconn.Close()

    tokr, err := newUnicodeTokenizer(false)
    if err != nil {
        t.Fatalf(err.Error())
    }

    to_add := filepath.Join(tmp, "to_add")
    err = mockDirectory(to_add)
    if err != nil {
        t.Fatalf(err.Error())
    }

    comments, err := addNewDirectory(dbconn, to_add, []string{ "metadata.json", "other.json" }, "myself", tokr)
    if err != nil {
        t.Fatal(err)
    }
    if len(comments) != 0 {
        t.Fatal("no comments should be present")
    }

    handler := http.HandlerFunc(newRetrieveMetadataHandler(dbconn))

    self, err := user.Current()
    if err != nil {
        t.Fatal(err)
    }
    selfname := self.Username

    validateResult := func(input io.Reader, expected_path string, has_metadata bool) {
        r := map[string]interface{}{}
        dec := json.NewDecoder(input)
        err = dec.Decode(&r)
        if err != nil {
            t.Fatal(err)
        }

        path_i, ok := r["path"]
        if !ok {
            t.Fatalf("expected a path property in %v", r)
        }
        path, ok := path_i.(string)
        if !ok {
            t.Fatalf("expected a path string property in %v", r)
        }
        if path != expected_path {
            t.Fatalf("unexpected value for the path %s", path)
        }

        user_i, ok := r["user"]
        if !ok {
            t.Fatalf("expected a user property in %v", r)
        }
        user, ok := user_i.(string)
        if !ok {
            t.Fatalf("expected a user string property in %v", r)
        }
        if user != selfname {
            t.Fatalf("unexpected username %v", user)
        }

        time_i, ok := r["time"]
        if !ok {
            t.Fatalf("expected a time property in %v", r)
        }
        time, ok := time_i.(float64)
        if !ok {
            t.Fatalf("expected a time integer property in %v", r)
        }
        if time <= 0 {
            t.Fatalf("time should be positive %v", time)
        }

        meta_i, ok := r["metadata"]
        if has_metadata {
            if !ok {
                t.Fatalf("expected a metadata property in %v", r)
            }
            _, ok = meta_i.(map[string]interface{})
            if !ok {
                t.Fatalf("expected a metadata object property in %v", r)
            }
        } else {
            if ok {
                t.Fatalf("unexpected metadata property in %v", r)
            }
        }
    }

    t.Run("simple", func (t *testing.T) {
        candidate := filepath.Join(to_add, "metadata.json")
        req, err := http.NewRequest("GET", "/retrieve/metadata?path=" + url.QueryEscape(candidate), nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        validateResult(rr.Body, candidate, true)
    })

    t.Run("no metadata", func (t *testing.T) {
        candidate := filepath.Join(to_add, "metadata.json")
        req, err := http.NewRequest("GET", "/retrieve/metadata?path=" + url.QueryEscape(candidate) + "&metadata=false", nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        validateResult(rr.Body, candidate, false)
    })

    t.Run("missing", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/retrieve/metadata?path=missing.json&metadata=false", nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusNotFound {
            t.Fatalf("should have failed without being found")
        }
    })
}

func TestRetrieveFileHandler(t *testing.T) {
    tmp, err := os.MkdirTemp("", "")
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer os.RemoveAll(tmp)

    dbpath := filepath.Join(tmp, "db.sqlite3")
    dbconn, err := initializeDatabase(dbpath)
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer dbconn.Close()

    tokr, err := newUnicodeTokenizer(false)
    if err != nil {
        t.Fatalf(err.Error())
    }

    to_add := filepath.Join(tmp, "to_add")
    err = mockDirectory(to_add)
    if err != nil {
        t.Fatalf(err.Error())
    }

    // Here, nothing is actually indexed! So we can't get confused with the metadata retrievals.
    comments, err := addNewDirectory(dbconn, to_add, []string{}, "myself", tokr)
    if err != nil {
        t.Fatal(err)
    }
    if len(comments) != 0 {
        t.Fatal("no comments should be present")
    }

    handler := http.HandlerFunc(newRetrieveFileHandler(dbconn))

    t.Run("success", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/retrieve/file?path=" + url.QueryEscape(filepath.Join(to_add, "metadata.json")), nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        r := map[string]interface{}{}
        dec := json.NewDecoder(rr.Body)
        err = dec.Decode(&r)
        if err != nil {
            t.Fatal(err)
        }

        foo, ok := r["foo"]
        if !ok || foo != "Aaron had a little lamb" {
            t.Fatal("unexpected result from file retrieval")
        }
    })

    t.Run("head", func (t *testing.T) {
        req, err := http.NewRequest("HEAD", "/retrieve/file?path=" + url.QueryEscape(filepath.Join(to_add, "metadata.json")), nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded")
        }

        headers := rr.Header()
        cl, err := strconv.Atoi(headers.Get("content-length"))
        if err != nil || cl == 0 {
            t.Fatal("expected a non-zero content-length header");
        }

        ct := headers.Get("content-type")
        if ct != "application/json" {
            t.Fatal("expected a JSON content type header");
        }

        lm := headers.Get("last-modified")
        _, err = time.Parse(time.RFC1123, lm)
        if err != nil {
            t.Fatalf("failed to parse the last-modified header; %v", err);
        }
    })

    t.Run("not found", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/retrieve/file?path=" + url.QueryEscape(filepath.Join(to_add, "other.json")), nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusNotFound {
            t.Fatalf("should have failed with a 404")
        }
    })

    t.Run("unregistered", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/retrieve/file?path=" + url.QueryEscape(filepath.Join(tmp, "metadata.json")), nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusForbidden {
            t.Fatalf("should have failed with a 403")
        }
    })

    t.Run("is directory", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/retrieve/file?path=" + url.QueryEscape(filepath.Join(to_add, "stuff")), nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusBadRequest {
            t.Fatalf("should have failed with a 400")
        }
    })
}

func TestListFilesHandler(t *testing.T) {
    tmp, err := os.MkdirTemp("", "")
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer os.RemoveAll(tmp)

    dbpath := filepath.Join(tmp, "db.sqlite3")
    dbconn, err := initializeDatabase(dbpath)
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer dbconn.Close()

    tokr, err := newUnicodeTokenizer(false)
    if err != nil {
        t.Fatalf(err.Error())
    }

    to_add := filepath.Join(tmp, "to_add")
    err = mockDirectory(to_add)
    if err != nil {
        t.Fatalf(err.Error())
    }

    comments, err := addNewDirectory(dbconn, to_add, []string{}, "myself", tokr)
    if err != nil {
        t.Fatal(err)
    }
    if len(comments) != 0 {
        t.Fatal("no comments should be present")
    }

    handler := http.HandlerFunc(newListFilesHandler(dbconn))

    t.Run("non-recursive", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/list?path=" + url.QueryEscape(to_add), nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded (got %d)", rr.Code)
        }

        r := []string{}
        dec := json.NewDecoder(rr.Body)
        err = dec.Decode(&r)
        if err != nil {
            t.Fatal(err)
        }

        sort.Strings(r)
        if len(r) != 3 || r[0] != "metadata.json" || r[1] != "stuff/" || r[2] != "whee/" {
            t.Fatalf("unexpected listing results %q", r)
        }
    })

    t.Run("recursive", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/list?path=" + url.QueryEscape(to_add) + "&recursive=true", nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded (got %d)", rr.Code)
        }

        r := []string{}
        dec := json.NewDecoder(rr.Body)
        err = dec.Decode(&r)
        if err != nil {
            t.Fatal(err)
        }

        sort.Strings(r)
        if len(r) != 4 || r[0] != "metadata.json" || r[1] != "stuff/metadata.json" || r[2] != "stuff/other.json" || r[3] != "whee/other.json" {
            t.Fatalf("unexpected listing results %q", r)
        }
    })

    t.Run("nested", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/list?path=" + url.QueryEscape(filepath.Join(to_add, "stuff")), nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded (got %d)", rr.Code)
        }

        r := []string{}
        dec := json.NewDecoder(rr.Body)
        err = dec.Decode(&r)
        if err != nil {
            t.Fatal(err)
        }

        sort.Strings(r)
        if len(r) != 2 || r[0] != "metadata.json" || r[1] != "other.json" {
            t.Fatalf("unexpected listing results %q", r)
        }
    })

    t.Run("not found", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/list?path=" + url.QueryEscape(filepath.Join(to_add, "missing")), nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusNotFound {
            t.Fatalf("should have failed with a 404 (got %d instead)", rr.Code)
        }
    })

    t.Run("unregistered", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/list?path=" + url.QueryEscape(tmp), nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusForbidden {
            t.Fatalf("should have failed with a 403")
        }
    })

    t.Run("is file", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/list?path=" + url.QueryEscape(filepath.Join(to_add, "metadata.json")), nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusBadRequest {
            t.Fatalf("should have failed with a 400 (got %d instead)", rr.Code)
        }
    })
}

func TestListRegisteredDirectoriesHandler(t *testing.T) {
    tmp, err := os.MkdirTemp("", "")
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer os.RemoveAll(tmp)

    dbpath := filepath.Join(tmp, "db.sqlite3")
    dbconn, err := initializeDatabase(dbpath)
    if err != nil {
        t.Fatalf(err.Error())
    }
    defer dbconn.Close()

    tokr, err := newUnicodeTokenizer(false)
    if err != nil {
        t.Fatalf(err.Error())
    }

    for _, name := range []string{ "akari", "ai", "alice" } {
        to_add := filepath.Join(tmp, "to_add_" + name)
        err = mockDirectory(to_add)
        if err != nil {
            t.Fatalf(err.Error())
        }

        comments, err := addNewDirectory(dbconn, to_add, []string{"metadata.json"}, name, tokr)
        if err != nil {
            t.Fatal(err)
        }
        if len(comments) != 0 {
            t.Fatal("no comments should be present")
        }
    }

    handler := http.HandlerFunc(newListRegisteredDirectoriesHandler(dbconn))

    t.Run("basic", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/registered", nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded (got %d)", rr.Code)
        }

        type lrdResult struct {
            Path string
            User string
            Time int64
            Names []string
        }

        r := []lrdResult{}
        dec := json.NewDecoder(rr.Body)
        err = dec.Decode(&r)
        if err != nil {
            t.Fatal(err)
        }

        if len(r) != 3 || r[0].User != "akari" || r[1].User != "ai" || r[2].User != "alice" {
            t.Fatalf("unexpected listing results for the users %q", r)
        }

        if filepath.Base(r[0].Path) != "to_add_akari" || r[0].Time == 0 || r[0].Names[0] != "metadata.json" {
            t.Fatalf("unexpected listing results for the first entry %q", r)
        }
    })

    t.Run("filtered by user", func (t *testing.T) {
        req, err := http.NewRequest("GET", "/registered?user=alice", nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded (got %d)", rr.Code)
        }

        type lrdResult struct {
            Path string
            User string
            Time int64
            Names []string
        }

        r := []lrdResult{}
        dec := json.NewDecoder(rr.Body)
        err = dec.Decode(&r)
        if err != nil {
            t.Fatal(err)
        }

        if len(r) != 1 || r[0].User != "alice" || filepath.Base(r[0].Path) != "to_add_alice" || r[0].Time == 0 || r[0].Names[0] != "metadata.json" {
            t.Fatalf("unexpected listing results %q", r)
        }
    })

    t.Run("filtered by contains_path", func (t *testing.T) {
        inside := filepath.Join(tmp, "to_add_akari", "stuff")
        encoded := url.QueryEscape(inside)
        req, err := http.NewRequest("GET", "/registered?contains_path=" + encoded, nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded (got %d)", rr.Code)
        }

        type lrdResult struct {
            Path string
            User string
            Time int64
            Names []string
        }

        r := []lrdResult{}
        dec := json.NewDecoder(rr.Body)
        err = dec.Decode(&r)
        if err != nil {
            t.Fatal(err)
        }

        if len(r) != 1 || r[0].User != "akari" {
            t.Fatalf("unexpected listing results %q", r)
        }
    })

    t.Run("filtered by has_prefix", func (t *testing.T) {
        encoded := url.QueryEscape(tmp)
        req, err := http.NewRequest("GET", "/registered?has_prefix=" + encoded, nil)
        if err != nil {
            t.Fatal(err)
        }

        rr := httptest.NewRecorder()
        handler.ServeHTTP(rr, req)
        if rr.Code != http.StatusOK {
            t.Fatalf("should have succeeded (got %d)", rr.Code)
        }

        type lrdResult struct {
            Path string
            User string
            Time int64
            Names []string
        }

        r := []lrdResult{}
        dec := json.NewDecoder(rr.Body)
        err = dec.Decode(&r)
        if err != nil {
            t.Fatal(err)
        }

        if len(r) != 3 {
            t.Fatalf("unexpected listing results %q", r)
        }
    })

}