sdk/go/crunchrunner/upload_test.go

   1 package main
   2
   3 import (
   4         "crypto/md5"
   5         "errors"
   6         "fmt"
   7         . "gopkg.in/check.v1"
   8         "io/ioutil"
   9         "log"
  10         "os"
  11 )
  12
  13 type UploadTestSuite struct{}
  14
  15 // Gocheck boilerplate
  16 var _ = Suite(&UploadTestSuite{})
  17
  18 type KeepTestClient struct {
  19 }
  20
  21 func (k KeepTestClient) PutHB(hash string, buf []byte) (string, int, error) {
  22         return fmt.Sprintf("%x+%v", md5.Sum(buf), len(buf)), len(buf), nil
  23 }
  24
  25 func (s *TestSuite) TestSimpleUpload(c *C) {
  26         log.Print("--TestSimpleUpload--")
  27
  28         tmpdir, _ := ioutil.TempDir("", "")
  29         defer func() {
  30                 os.RemoveAll(tmpdir)
  31         }()
  32
  33         ioutil.WriteFile(tmpdir+"/"+"file1.txt", []byte("foo"), 0600)
  34
  35         str, err := WriteTree(KeepTestClient{}, tmpdir)
  36         c.Check(err, IsNil)
  37         c.Check(str, Equals, ". acbd18db4cc2f85cedef654fccc4a4d8+3 0:3:file1.txt\n")
  38 }
  39
  40 func (s *TestSuite) TestSimpleUploadTwofiles(c *C) {
  41         log.Print("--TestSimpleUploadTwofiles--")
  42
  43         tmpdir, _ := ioutil.TempDir("", "")
  44         defer func() {
  45                 os.RemoveAll(tmpdir)
  46         }()
  47
  48         ioutil.WriteFile(tmpdir+"/"+"file1.txt", []byte("foo"), 0600)
  49         ioutil.WriteFile(tmpdir+"/"+"file2.txt", []byte("bar"), 0600)
  50
  51         str, err := WriteTree(KeepTestClient{}, tmpdir)
  52         c.Check(err, IsNil)
  53         c.Check(str, Equals, ". 3858f62230ac3c915f300c664312c63f+6 0:3:file1.txt 3:3:file2.txt\n")
  54 }
  55
  56 func (s *TestSuite) TestSimpleUploadSubdir(c *C) {
  57         log.Print("--TestSimpleUploadSubdir--")
  58
  59         tmpdir, _ := ioutil.TempDir("", "")
  60         defer func() {
  61                 os.RemoveAll(tmpdir)
  62         }()
  63
  64         os.Mkdir(tmpdir+"/subdir", 0700)
  65
  66         ioutil.WriteFile(tmpdir+"/"+"file1.txt", []byte("foo"), 0600)
  67         ioutil.WriteFile(tmpdir+"/subdir/file2.txt", []byte("bar"), 0600)
  68
  69         str, err := WriteTree(KeepTestClient{}, tmpdir)
  70         c.Check(err, IsNil)
  71         c.Check(str, Equals, `. acbd18db4cc2f85cedef654fccc4a4d8+3 0:3:file1.txt
  72 ./subdir 37b51d194a7513e45b56f6524f2d51f2+3 0:3:file2.txt
  73 `)
  74 }
  75
  76 func (s *TestSuite) TestSimpleUploadLarge(c *C) {
  77         log.Print("--TestSimpleUploadLarge--")
  78
  79         tmpdir, _ := ioutil.TempDir("", "")
  80         defer func() {
  81                 os.RemoveAll(tmpdir)
  82         }()
  83
  84         file, _ := os.Create(tmpdir + "/" + "file1.txt")
  85         data := make([]byte, 1024*1024-1)
  86         for i := 0; i < 1024*1024-1; i++ {
  87                 data[i] = byte(i % 10)
  88         }
  89         for i := 0; i < 65; i++ {
  90                 file.Write(data)
  91         }
  92         file.Close()
  93
  94         ioutil.WriteFile(tmpdir+"/"+"file2.txt", []byte("bar"), 0600)
  95
  96         str, err := WriteTree(KeepTestClient{}, tmpdir)
  97         c.Check(err, IsNil)
  98         c.Check(str, Equals, ". 00ecf01e0d93385115c9f8bed757425d+67108864 485cd630387b6b1846fe429f261ea05f+1048514 0:68157375:file1.txt 68157375:3:file2.txt\n")
  99 }
 100
 101 func (s *TestSuite) TestUploadEmptySubdir(c *C) {
 102         log.Print("--TestUploadEmptySubdir--")
 103
 104         tmpdir, _ := ioutil.TempDir("", "")
 105         defer func() {
 106                 os.RemoveAll(tmpdir)
 107         }()
 108
 109         os.Mkdir(tmpdir+"/subdir", 0700)
 110
 111         ioutil.WriteFile(tmpdir+"/"+"file1.txt", []byte("foo"), 0600)
 112
 113         str, err := WriteTree(KeepTestClient{}, tmpdir)
 114         c.Check(err, IsNil)
 115         c.Check(str, Equals, `. acbd18db4cc2f85cedef654fccc4a4d8+3 0:3:file1.txt
 116 `)
 117 }
 118
 119 func (s *TestSuite) TestUploadEmptyFile(c *C) {
 120         log.Print("--TestUploadEmptyFile--")
 121
 122         tmpdir, _ := ioutil.TempDir("", "")
 123         defer func() {
 124                 os.RemoveAll(tmpdir)
 125         }()
 126
 127         ioutil.WriteFile(tmpdir+"/"+"file1.txt", []byte(""), 0600)
 128
 129         str, err := WriteTree(KeepTestClient{}, tmpdir)
 130         c.Check(err, IsNil)
 131         c.Check(str, Equals, `. d41d8cd98f00b204e9800998ecf8427e+0 0:0:file1.txt
 132 `)
 133 }
 134
 135 type KeepErrorTestClient struct {
 136 }
 137
 138 func (k KeepErrorTestClient) PutHB(hash string, buf []byte) (string, int, error) {
 139         return "", 0, errors.New("Failed!")
 140 }
 141
 142 func (s *TestSuite) TestUploadError(c *C) {
 143         log.Print("--TestSimpleUpload--")
 144
 145         tmpdir, _ := ioutil.TempDir("", "")
 146         defer func() {
 147                 os.RemoveAll(tmpdir)
 148         }()
 149
 150         ioutil.WriteFile(tmpdir+"/"+"file1.txt", []byte("foo"), 0600)
 151
 152         str, err := WriteTree(KeepErrorTestClient{}, tmpdir)
 153         c.Check(err, NotNil)
 154         c.Check(str, Equals, "")
 155 }