shoopea/gowebdav
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: add unit test for upload file to server acquire content length

Browse Source 
Signed-off-by: Ben Tam <master@murasakiakari.moe>
This commit is contained in:
Ben Tam 2023-10-31 23:24:53 +08:00
parent df102c9cec
commit 828132ab79
No known key found for this signature in database
GPG Key ID: FD345C984C044DC0
1 changed files with 69 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
client_test.go
View File

@ -39,6 +39,25 @@ func basicAuth(h http.Handler) http.HandlerFunc {
}
}
func basicAuthWithPostHandlerFunc(h http.Handler, postHandlerFunc http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
user, passwd, ok := r.BasicAuth()
if !ok {
w.Header().Set("WWW-Authenticate", `Basic realm="x"`)
w.WriteHeader(401)
return
}
if user != "user" || passwd != "password" {
http.Error(w, "not authorized", 403)
return
}
h.ServeHTTP(w, r)
postHandlerFunc(w, r)
}
}
func multipleAuth(h http.Handler) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
notAuthed := false
@ -130,6 +149,38 @@ func newAuthSrv(t *testing.T, auth func(h http.Handler) http.HandlerFunc) (*http
return srv, fs, ctx
}
func newAuthServerAcquireContentLength(t *testing.T) (*Client, *httptest.Server, webdav.FileSystem, context.Context) {
srv, fs, ctx := newAuthSrvAcquireContentLength(t, basicAuthWithPostHandlerFunc)
cli := NewClient(srv.URL, "user", "password")
return cli, srv, fs, ctx
}
func newAuthSrvAcquireContentLength(t *testing.T, authWithPostHandlerFunc func(h http.Handler, postHandlerFunc http.HandlerFunc) http.HandlerFunc) (*httptest.Server, webdav.FileSystem, context.Context) {
mux := http.NewServeMux()
fs := webdav.NewMemFS()
ctx := fillFs(t, fs)
mux.HandleFunc("/", authWithPostHandlerFunc(&webdav.Handler{
FileSystem: fs,
LockSystem: webdav.NewMemLS(),
}, func(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodPut {
return
}
fileName := strings.TrimPrefix(r.URL.Path, "/")
stat, err := fs.Stat(ctx, fileName)
if err != nil {
t.Fatalf("got: %v, want nil", err)
}
if r.ContentLength != stat.Size() {
t.Fatalf("acquire content length got: %v, want %v", r.ContentLength, stat.Size())
}
}))
srv := httptest.NewServer(mux)
return srv, fs, ctx
}
func TestConnect(t *testing.T) {
cli, srv, _, _ := newServer(t)
defer srv.Close()
@ -572,3 +623,21 @@ func TestWriteStreamFromPipe(t *testing.T) {
t.Fatalf("got: %v, want file size: %d bytes", info.Size(), 8)
}
}
func TestWriteToServerAcquireContentLength(t *testing.T) {
cli, srv, _, _ := newAuthServerAcquireContentLength(t)
defer srv.Close()
if err := cli.Write("/newfile.txt", []byte("foo bar\n"), 0660); err != nil {
t.Fatalf("got: %v, want nil", err)
}
}
func TestWriteStreamToServerAcquireContentLength(t *testing.T) {
cli, srv, _, _ := newAuthServerAcquireContentLength(t)
defer srv.Close()
if err := cli.WriteStream("/newfile.txt", strings.NewReader("foo bar\n"), 0660); err != nil {
t.Fatalf("got: %v, want nil", err)
}
}