internal/testutil/spec.go

   1 package testutil
   2
   3 import (
   4         "io"
   5         "strings"
   6
   7         "github.com/anacrolix/missinggo/expect"
   8
   9         "github.com/anacrolix/torrent/bencode"
  10         "github.com/anacrolix/torrent/metainfo"
  11 )
  12
  13 type File struct {
  14         Name string
  15         Data string
  16 }
  17
  18 type Torrent struct {
  19         Files []File
  20         Name  string
  21 }
  22
  23 func (t *Torrent) IsDir() bool {
  24         return len(t.Files) == 1 && t.Files[0].Name == ""
  25 }
  26
  27 func (t *Torrent) GetFile(name string) *File {
  28         if t.IsDir() && t.Name == name {
  29                 return &t.Files[0]
  30         }
  31         for _, f := range t.Files {
  32                 if f.Name == name {
  33                         return &f
  34                 }
  35         }
  36         return nil
  37 }
  38
  39 func (t *Torrent) Info(pieceLength int64) metainfo.Info {
  40         info := metainfo.Info{
  41                 Name:        t.Name,
  42                 PieceLength: pieceLength,
  43         }
  44         if t.IsDir() {
  45                 info.Length = int64(len(t.Files[0].Data))
  46         } else {
  47                 for _, f := range t.Files {
  48                         info.Files = append(info.Files, metainfo.FileInfo{
  49                                 Path:   []string{f.Name},
  50                                 Length: int64(len(f.Data)),
  51                         })
  52                 }
  53         }
  54         err := info.GeneratePieces(func(fi metainfo.FileInfo) (io.ReadCloser, error) {
  55                 return io.NopCloser(strings.NewReader(t.GetFile(strings.Join(fi.Path, "/")).Data)), nil
  56         })
  57         expect.Nil(err)
  58         return info
  59 }
  60
  61 func (t *Torrent) Metainfo(pieceLength int64) *metainfo.MetaInfo {
  62         mi := metainfo.MetaInfo{}
  63         var err error
  64         mi.InfoBytes, err = bencode.Marshal(t.Info(pieceLength))
  65         expect.Nil(err)
  66         return &mi
  67 }