mirror of
https://github.com/rclone/rclone.git
synced 2024-11-30 04:23:48 +08:00
89 lines
2.2 KiB
Go
89 lines
2.2 KiB
Go
|
// +build go1.8
|
||
|
|
|
||
|
|
package dlna
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/ncw/rclone/vfs"
|
||
|
|
|
||
|
|
_ "github.com/ncw/rclone/backend/local"
|
||
|
|
"github.com/ncw/rclone/cmd/serve/dlna/dlnaflags"
|
||
|
|
"github.com/ncw/rclone/fs"
|
||
|
|
"github.com/ncw/rclone/fs/config"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
dlnaServer *server
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
testBindAddress = "localhost:51777"
|
||
|
|
testURL = "http://" + testBindAddress + "/"
|
||
|
|
)
|
||
|
|
|
||
|
|
func startServer(t *testing.T, f fs.Fs) {
|
||
|
|
opt := dlnaflags.DefaultOpt
|
||
|
|
opt.ListenAddr = testBindAddress
|
||
|
|
dlnaServer = newServer(f, &opt)
|
||
|
|
assert.NoError(t, dlnaServer.Serve())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestInit(t *testing.T) {
|
||
|
|
config.LoadConfig()
|
||
|
|
|
||
|
|
f, err := fs.NewFs("testdata/files")
|
||
|
|
l, _ := f.List("")
|
||
|
|
fmt.Println(l)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
startServer(t, f)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Make sure that it serves rootDesc.xml (SCPD in uPnP parlance).
|
||
|
|
func TestRootSCPD(t *testing.T) {
|
||
|
|
req, err := http.NewRequest("GET", testURL+"rootDesc.xml", nil)
|
||
|
|
require.NoError(t, err)
|
||
|
|
resp, err := http.DefaultClient.Do(req)
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
||
|
|
body, err := ioutil.ReadAll(resp.Body)
|
||
|
|
require.NoError(t, err)
|
||
|
|
// Make sure that the SCPD contains a CDS service.
|
||
|
|
require.Contains(t, string(body),
|
||
|
|
"<serviceType>urn:schemas-upnp-org:service:ContentDirectory:1</serviceType>")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Make sure that it serves content from the remote.
|
||
|
|
func TestServeContent(t *testing.T) {
|
||
|
|
itemPath := "/small_jpeg.jpg"
|
||
|
|
pathQuery := url.QueryEscape(itemPath)
|
||
|
|
req, err := http.NewRequest("GET", testURL+"res?path="+pathQuery, nil)
|
||
|
|
require.NoError(t, err)
|
||
|
|
resp, err := http.DefaultClient.Do(req)
|
||
|
|
require.NoError(t, err)
|
||
|
|
defer fs.CheckClose(resp.Body, &err)
|
||
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
||
|
|
actualContents, err := ioutil.ReadAll(resp.Body)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
// Now compare the contents with the golden file.
|
||
|
|
node, err := dlnaServer.vfs.Stat(itemPath)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
goldenFile := node.(*vfs.File)
|
||
|
|
goldenReader, err := goldenFile.Open(os.O_RDONLY)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
defer fs.CheckClose(goldenReader, &err)
|
||
|
|
goldenContents, err := ioutil.ReadAll(goldenReader)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
require.Equal(t, goldenContents, actualContents)
|
||
|
|
}
|