75 lines
1.5 KiB
Go
75 lines
1.5 KiB
Go
package tusd_test
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
"testing"
|
|
|
|
. "github.com/tus/tusd"
|
|
)
|
|
|
|
type headStore struct {
|
|
zeroStore
|
|
}
|
|
|
|
func (s headStore) GetInfo(id string) (FileInfo, error) {
|
|
if id != "yes" {
|
|
return FileInfo{}, os.ErrNotExist
|
|
}
|
|
|
|
return FileInfo{
|
|
Offset: 11,
|
|
Size: 44,
|
|
MetaData: map[string]string{
|
|
"name": "lunrjs.png",
|
|
"type": "image/png",
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func TestHead(t *testing.T) {
|
|
handler, _ := NewHandler(Config{
|
|
BasePath: "https://buy.art/",
|
|
DataStore: headStore{},
|
|
})
|
|
|
|
res := (&httpTest{
|
|
Name: "Successful request",
|
|
Method: "HEAD",
|
|
URL: "yes",
|
|
ReqHeader: map[string]string{
|
|
"Tus-Resumable": "1.0.0",
|
|
},
|
|
Code: http.StatusOK,
|
|
ResHeader: map[string]string{
|
|
"Upload-Offset": "11",
|
|
"Upload-Length": "44",
|
|
"Cache-Control": "no-store",
|
|
},
|
|
}).Run(handler, t)
|
|
|
|
// Since the order of a map is not guaranteed in Go, we need to be prepared
|
|
// for the case, that the order of the metadata may have been changed
|
|
if v := res.Header().Get("Upload-Metadata"); v != "name bHVucmpzLnBuZw==,type aW1hZ2UvcG5n" &&
|
|
v != "type aW1hZ2UvcG5n,name bHVucmpzLnBuZw==" {
|
|
t.Errorf("Expected valid metadata (got '%s')", v)
|
|
}
|
|
|
|
res = (&httpTest{
|
|
Name: "Non-existing file",
|
|
Method: "HEAD",
|
|
URL: "no",
|
|
ReqHeader: map[string]string{
|
|
"Tus-Resumable": "1.0.0",
|
|
},
|
|
Code: http.StatusNotFound,
|
|
ResHeader: map[string]string{
|
|
"Content-Length": "0",
|
|
},
|
|
}).Run(handler, t)
|
|
|
|
if string(res.Body.Bytes()) != "" {
|
|
t.Errorf("Expected empty body for failed HEAD request")
|
|
}
|
|
}
|