52 lines
918 B
Go
52 lines
918 B
Go
|
package tusd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
type getStore struct {
|
||
|
|
zeroStore
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s getStore) GetInfo(id string) (FileInfo, error) {
|
||
|
|
if id != "yes" {
|
||
|
|
return FileInfo{}, os.ErrNotExist
|
||
|
|
}
|
||
|
|
|
||
|
|
return FileInfo{
|
||
|
|
Offset: 5,
|
||
|
|
Size: 20,
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s getStore) GetReader(id string) (io.Reader, error) {
|
||
|
|
return strings.NewReader("hello"), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetFile(t *testing.T) {
|
||
|
|
handler, _ := NewHandler(Config{
|
||
|
|
DataStore: getStore{},
|
||
|
|
})
|
||
|
|
|
||
|
|
// Test successfull download
|
||
|
|
req, _ := http.NewRequest("GET", "yes", nil)
|
||
|
|
w := httptest.NewRecorder()
|
||
|
|
handler.ServeHTTP(w, req)
|
||
|
|
if w.Code != http.StatusOK {
|
||
|
|
t.Errorf("Expected %v (got %v)", http.StatusOK, w.Code)
|
||
|
|
}
|
||
|
|
|
||
|
|
if string(w.Body.Bytes()) != "hello" {
|
||
|
|
t.Errorf("Expected response body to be 'hello'")
|
||
|
|
}
|
||
|
|
|
||
|
|
if w.HeaderMap.Get("Content-Length") != "5" {
|
||
|
|
t.Errorf("Expected Content-Length to be 5")
|
||
|
|
}
|
||
|
|
}
|