mirror of
https://github.com/rclone/rclone.git
synced 2024-12-12 12:19:27 +08:00
5d6b8141ec
As of Go 1.16, the same functionality is now provided by package io or package os, and those implementations should be preferred in new code.
77 lines
2.6 KiB
Go
77 lines
2.6 KiB
Go
package serve
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/rclone/rclone/fstest/mockobject"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestObjectBadMethod(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
r := httptest.NewRequest("BADMETHOD", "http://example.com/aFile", nil)
|
|
o := mockobject.New("aFile")
|
|
Object(w, r, o)
|
|
resp := w.Result()
|
|
assert.Equal(t, http.StatusMethodNotAllowed, resp.StatusCode)
|
|
body, _ := io.ReadAll(resp.Body)
|
|
assert.Equal(t, "Method Not Allowed\n", string(body))
|
|
}
|
|
|
|
func TestObjectHEAD(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
r := httptest.NewRequest("HEAD", "http://example.com/aFile", nil)
|
|
o := mockobject.New("aFile").WithContent([]byte("hello"), mockobject.SeekModeNone)
|
|
Object(w, r, o)
|
|
resp := w.Result()
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
assert.Equal(t, "5", resp.Header.Get("Content-Length"))
|
|
assert.Equal(t, "bytes", resp.Header.Get("Accept-Ranges"))
|
|
body, _ := io.ReadAll(resp.Body)
|
|
assert.Equal(t, "", string(body))
|
|
}
|
|
|
|
func TestObjectGET(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
r := httptest.NewRequest("GET", "http://example.com/aFile", nil)
|
|
o := mockobject.New("aFile").WithContent([]byte("hello"), mockobject.SeekModeNone)
|
|
Object(w, r, o)
|
|
resp := w.Result()
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
assert.Equal(t, "5", resp.Header.Get("Content-Length"))
|
|
assert.Equal(t, "bytes", resp.Header.Get("Accept-Ranges"))
|
|
body, _ := io.ReadAll(resp.Body)
|
|
assert.Equal(t, "hello", string(body))
|
|
}
|
|
|
|
func TestObjectRange(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
r := httptest.NewRequest("GET", "http://example.com/aFile", nil)
|
|
r.Header.Add("Range", "bytes=3-5")
|
|
o := mockobject.New("aFile").WithContent([]byte("0123456789"), mockobject.SeekModeNone)
|
|
Object(w, r, o)
|
|
resp := w.Result()
|
|
assert.Equal(t, http.StatusPartialContent, resp.StatusCode)
|
|
assert.Equal(t, "3", resp.Header.Get("Content-Length"))
|
|
assert.Equal(t, "bytes", resp.Header.Get("Accept-Ranges"))
|
|
assert.Equal(t, "bytes 3-5/10", resp.Header.Get("Content-Range"))
|
|
body, _ := io.ReadAll(resp.Body)
|
|
assert.Equal(t, "345", string(body))
|
|
}
|
|
|
|
func TestObjectBadRange(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
r := httptest.NewRequest("GET", "http://example.com/aFile", nil)
|
|
r.Header.Add("Range", "xxxbytes=3-5")
|
|
o := mockobject.New("aFile").WithContent([]byte("0123456789"), mockobject.SeekModeNone)
|
|
Object(w, r, o)
|
|
resp := w.Result()
|
|
assert.Equal(t, http.StatusBadRequest, resp.StatusCode)
|
|
assert.Equal(t, "10", resp.Header.Get("Content-Length"))
|
|
body, _ := io.ReadAll(resp.Body)
|
|
assert.Equal(t, "Bad Request\n", string(body))
|
|
}
|