wip
This commit is contained in:
parent
200a600c02
commit
0f74925694
78
cmd/gowebdav/main_test.go
Normal file
78
cmd/gowebdav/main_test.go
Normal file
@ -0,0 +1,78 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"golang.org/x/net/webdav"
|
||||
)
|
||||
|
||||
func basicAuth(h http.Handler) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
if user, passwd, ok := r.BasicAuth(); ok {
|
||||
if user == "user" && passwd == "password" {
|
||||
h.ServeHTTP(w, r)
|
||||
return
|
||||
}
|
||||
|
||||
http.Error(w, "not authorized", 403)
|
||||
} else {
|
||||
w.Header().Set("WWW-Authenticate", `Basic realm="x"`)
|
||||
w.WriteHeader(401)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func newServer(t *testing.T) (*httptest.Server, webdav.FileSystem, context.Context) {
|
||||
mux := http.NewServeMux()
|
||||
fs := webdav.NewMemFS()
|
||||
ctx := fillFs(t, fs)
|
||||
mux.HandleFunc("/", basicAuth(&webdav.Handler{
|
||||
FileSystem: fs,
|
||||
LockSystem: webdav.NewMemLS(),
|
||||
}))
|
||||
srv := httptest.NewServer(mux)
|
||||
os.Setenv("ROOT", srv.URL)
|
||||
os.Setenv("USER", "user")
|
||||
os.Setenv("PASSWORD", "password")
|
||||
return srv, fs, ctx
|
||||
}
|
||||
|
||||
func fillFs(t *testing.T, fs webdav.FileSystem) context.Context {
|
||||
ctx := context.Background()
|
||||
f, err := fs.OpenFile(ctx, "hello.txt", os.O_CREATE, 0644)
|
||||
if err != nil {
|
||||
t.Errorf("fail to crate file: %v", err)
|
||||
}
|
||||
f.Write([]byte("hello gowebdav\n"))
|
||||
f.Close()
|
||||
err = fs.Mkdir(ctx, "/test", 0755)
|
||||
if err != nil {
|
||||
t.Errorf("fail to crate directory: %v", err)
|
||||
}
|
||||
f, err = fs.OpenFile(ctx, "/test/test.txt", os.O_CREATE, 0644)
|
||||
if err != nil {
|
||||
t.Errorf("fail to crate file: %v", err)
|
||||
}
|
||||
f.Write([]byte("test test gowebdav\n"))
|
||||
f.Close()
|
||||
return ctx
|
||||
}
|
||||
|
||||
func TestLs(t *testing.T) {
|
||||
oldArgs := os.Args
|
||||
defer func() { os.Args = oldArgs }()
|
||||
|
||||
srv, _, _ := newServer(t)
|
||||
|
||||
defer srv.Close()
|
||||
|
||||
flag.CommandLine = flag.NewFlagSet("ls", flag.ExitOnError)
|
||||
|
||||
os.Args = []string{"ls", "-X", "ls", "/"}
|
||||
main()
|
||||
}
|
Loading…
Reference in New Issue
Block a user