45 lines
888 B
Go
45 lines
888 B
Go
|
package testutil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
metainfo "github.com/nsf/libtorgo/torrent"
|
||
|
|
|
||
|
|
"bytes"
|
||
|
|
)
|
||
|
|
|
||
|
|
const GreetingFileContents = "hello, world\n"
|
||
|
|
|
||
|
|
func CreateDummyTorrentData(dirName string) string {
|
||
|
|
f, _ := os.Create(filepath.Join(dirName, "greeting"))
|
||
|
|
f.WriteString("hello, world\n")
|
||
|
|
return f.Name()
|
||
|
|
}
|
||
|
|
|
||
|
|
func CreateMetaInfo(name string, w io.Writer) {
|
||
|
|
builder := metainfo.Builder{}
|
||
|
|
builder.AddFile(name)
|
||
|
|
builder.AddAnnounceGroup([]string{"lol://cheezburger"})
|
||
|
|
batch, err := builder.Submit()
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
errs, _ := batch.Start(w, 1)
|
||
|
|
<-errs
|
||
|
|
}
|
||
|
|
|
||
|
|
func GreetingTestTorrent() (tempDir string, metaInfo *metainfo.MetaInfo) {
|
||
|
|
tempDir, err := ioutil.TempDir(os.TempDir(), "")
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
name := CreateDummyTorrentData(tempDir)
|
||
|
|
w := &bytes.Buffer{}
|
||
|
|
CreateMetaInfo(name, w)
|
||
|
|
metaInfo, _ = metainfo.Load(w)
|
||
|
|
return
|
||
|
|
}
|