package tusd import ( "io" "net/http" "net/http/httptest" "testing" ) type zeroStore struct{} func (store zeroStore) NewUpload(info FileInfo) (string, error) { return "", nil } func (store zeroStore) WriteChunk(id string, offset int64, src io.Reader) error { return nil } func (store zeroStore) GetInfo(id string) (FileInfo, error) { return FileInfo{}, nil } func (store zeroStore) GetReader(id string) (io.Reader, error) { return nil, ErrNotImplemented } type httpTest struct { Name string Method string URL string ReqBody io.Reader ReqHeader map[string]string Code int ResBody string ResHeader map[string]string } func (test *httpTest) Run(handler http.Handler, t *testing.T) { t.Log(test.Name) req, _ := http.NewRequest(test.Method, test.URL, test.ReqBody) // Add headers for key, value := range test.ReqHeader { req.Header.Set(key, value) } req.Host = "tus.io" w := httptest.NewRecorder() handler.ServeHTTP(w, req) if w.Code != test.Code { t.Errorf("Expected %v as status code (got %v)", test.Code, w.Code) } for key, value := range test.ResHeader { header := w.HeaderMap.Get(key) if value == "" && header == "" { t.Errorf("Expected '%s' in response", key) } if value != "" && value != header { t.Errorf("Expected '%s' as '%s' (got '%s')", value, key, header) } } if test.ResBody != "" && string(w.Body.Bytes()) != test.ResBody { t.Errorf("Expected '%s' as body (got '%s'", test.ResBody, string(w.Body.Bytes())) } }