ProjectHentai/alist
6
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: add aria2 rpc package

Browse Source
This commit is contained in:
Noah Hsu
2022-06-15 17:15:22 +08:00
parent c65a9b3001
commit 2cddd3cf2b
13 changed files with 1723 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

125
pkg/aria2/rpc/client_test.go Normal file
View File

@ -0,0 +1,125 @@
package rpc
import (
"context"
"testing"
"time"
)
func TestHTTPAll(t *testing.T) {
const targetURL = "https://nodejs.org/dist/index.json"
rpc, err := New(context.Background(), "http://localhost:6800/jsonrpc", "", time.Second, &DummyNotifier{})
if err != nil {
t.Fatal(err)
}
defer rpc.Close()
g, err := rpc.AddURI([]string{targetURL})
if err != nil {
t.Fatal(err)
}
println(g)
if _, err = rpc.TellActive(); err != nil {
t.Error(err)
}
if _, err = rpc.PauseAll(); err != nil {
t.Error(err)
}
if _, err = rpc.TellStatus(g); err != nil {
t.Error(err)
}
if _, err = rpc.GetURIs(g); err != nil {
t.Error(err)
}
if _, err = rpc.GetFiles(g); err != nil {
t.Error(err)
}
if _, err = rpc.GetPeers(g); err != nil {
t.Error(err)
}
if _, err = rpc.TellActive(); err != nil {
t.Error(err)
}
if _, err = rpc.TellWaiting(0, 1); err != nil {
t.Error(err)
}
if _, err = rpc.TellStopped(0, 1); err != nil {
t.Error(err)
}
if _, err = rpc.GetOption(g); err != nil {
t.Error(err)
}
if _, err = rpc.GetGlobalOption(); err != nil {
t.Error(err)
}
if _, err = rpc.GetGlobalStat(); err != nil {
t.Error(err)
}
if _, err = rpc.GetSessionInfo(); err != nil {
t.Error(err)
}
if _, err = rpc.Remove(g); err != nil {
t.Error(err)
}
if _, err = rpc.TellActive(); err != nil {
t.Error(err)
}
}
func TestWebsocketAll(t *testing.T) {
const targetURL = "https://nodejs.org/dist/index.json"
rpc, err := New(context.Background(), "ws://localhost:6800/jsonrpc", "", time.Second, &DummyNotifier{})
if err != nil {
t.Fatal(err)
}
defer rpc.Close()
g, err := rpc.AddURI([]string{targetURL})
if err != nil {
t.Fatal(err)
}
println(g)
if _, err = rpc.TellActive(); err != nil {
t.Error(err)
}
if _, err = rpc.PauseAll(); err != nil {
t.Error(err)
}
if _, err = rpc.TellStatus(g); err != nil {
t.Error(err)
}
if _, err = rpc.GetURIs(g); err != nil {
t.Error(err)
}
if _, err = rpc.GetFiles(g); err != nil {
t.Error(err)
}
if _, err = rpc.GetPeers(g); err != nil {
t.Error(err)
}
if _, err = rpc.TellActive(); err != nil {
t.Error(err)
}
if _, err = rpc.TellWaiting(0, 1); err != nil {
t.Error(err)
}
if _, err = rpc.TellStopped(0, 1); err != nil {
t.Error(err)
}
if _, err = rpc.GetOption(g); err != nil {
t.Error(err)
}
if _, err = rpc.GetGlobalOption(); err != nil {
t.Error(err)
}
if _, err = rpc.GetGlobalStat(); err != nil {
t.Error(err)
}
if _, err = rpc.GetSessionInfo(); err != nil {
t.Error(err)
}
if _, err = rpc.Remove(g); err != nil {
t.Error(err)
}
if _, err = rpc.TellActive(); err != nil {
t.Error(err)
}
}