1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: Apache-2.0
16 type UploadTestSuite struct{}
18 // Gocheck boilerplate
19 var _ = Suite(&UploadTestSuite{})
21 type KeepTestClient struct {
24 func (k KeepTestClient) PutHB(hash string, buf []byte) (string, int, error) {
25 return fmt.Sprintf("%x+%v", md5.Sum(buf), len(buf)), len(buf), nil
28 func (s *TestSuite) TestSimpleUpload(c *C) {
29 tmpdir, _ := ioutil.TempDir("", "")
34 ioutil.WriteFile(tmpdir+"/"+"file1.txt", []byte("foo"), 0600)
36 str, err := WriteTree(KeepTestClient{}, tmpdir)
38 c.Check(str, Equals, ". acbd18db4cc2f85cedef654fccc4a4d8+3 0:3:file1.txt\n")
41 func (s *TestSuite) TestSimpleUploadTwofiles(c *C) {
42 tmpdir, _ := ioutil.TempDir("", "")
47 ioutil.WriteFile(tmpdir+"/"+"file1.txt", []byte("foo"), 0600)
48 ioutil.WriteFile(tmpdir+"/"+"file2.txt", []byte("bar"), 0600)
50 str, err := WriteTree(KeepTestClient{}, tmpdir)
52 c.Check(str, Equals, ". 3858f62230ac3c915f300c664312c63f+6 0:3:file1.txt 3:3:file2.txt\n")
55 func (s *TestSuite) TestSimpleUploadSubdir(c *C) {
56 tmpdir, _ := ioutil.TempDir("", "")
61 os.Mkdir(tmpdir+"/subdir", 0700)
63 ioutil.WriteFile(tmpdir+"/"+"file1.txt", []byte("foo"), 0600)
64 ioutil.WriteFile(tmpdir+"/subdir/file2.txt", []byte("bar"), 0600)
66 str, err := WriteTree(KeepTestClient{}, tmpdir)
68 c.Check(str, Equals, `. acbd18db4cc2f85cedef654fccc4a4d8+3 0:3:file1.txt
69 ./subdir 37b51d194a7513e45b56f6524f2d51f2+3 0:3:file2.txt
73 func (s *TestSuite) TestSimpleUploadLarge(c *C) {
74 tmpdir, _ := ioutil.TempDir("", "")
79 file, _ := os.Create(tmpdir + "/" + "file1.txt")
80 data := make([]byte, 1024*1024-1)
82 data[i] = byte(i % 10)
84 for i := 0; i < 65; i++ {
89 ioutil.WriteFile(tmpdir+"/"+"file2.txt", []byte("bar"), 0600)
91 str, err := WriteTree(KeepTestClient{}, tmpdir)
93 c.Check(str, Equals, ". 00ecf01e0d93385115c9f8bed757425d+67108864 485cd630387b6b1846fe429f261ea05f+1048514 0:68157375:file1.txt 68157375:3:file2.txt\n")
96 func (s *TestSuite) TestUploadEmptySubdir(c *C) {
97 tmpdir, _ := ioutil.TempDir("", "")
102 os.Mkdir(tmpdir+"/subdir", 0700)
104 ioutil.WriteFile(tmpdir+"/"+"file1.txt", []byte("foo"), 0600)
106 str, err := WriteTree(KeepTestClient{}, tmpdir)
108 c.Check(str, Equals, `. acbd18db4cc2f85cedef654fccc4a4d8+3 0:3:file1.txt
112 func (s *TestSuite) TestUploadEmptyFile(c *C) {
113 tmpdir, _ := ioutil.TempDir("", "")
118 ioutil.WriteFile(tmpdir+"/"+"file1.txt", []byte(""), 0600)
120 str, err := WriteTree(KeepTestClient{}, tmpdir)
122 c.Check(str, Equals, `. d41d8cd98f00b204e9800998ecf8427e+0 0:0:file1.txt
126 type KeepErrorTestClient struct {
129 func (k KeepErrorTestClient) PutHB(hash string, buf []byte) (string, int, error) {
130 return "", 0, errors.New("Failed!")
133 func (s *TestSuite) TestUploadError(c *C) {
134 tmpdir, _ := ioutil.TempDir("", "")
139 ioutil.WriteFile(tmpdir+"/"+"file1.txt", []byte("foo"), 0600)
141 str, err := WriteTree(KeepErrorTestClient{}, tmpdir)
143 c.Check(str, Equals, "")