Merge branch '1748-workbench-view-files'
[arvados.git] / sdk / cli / test / test_arv-put.rb
1 require 'minitest/autorun'
2 require 'digest/md5'
3
4 class TestArvPut < Minitest::Test
5   def setup
6     begin Dir.mkdir './tmp' rescue Errno::EEXIST end
7     begin Dir.mkdir './tmp/empty_dir' rescue Errno::EEXIST end
8     File.open './tmp/empty_file', 'wb' do
9     end
10     File.open './tmp/foo', 'wb' do |f|
11       f.write 'foo'
12     end
13   end
14
15   def test_help
16     out, err = capture_subprocess_io do
17       assert_equal(true, arv_put('-h'),
18                    'arv-put -h exits zero')
19     end
20     $stderr.write err
21     assert_equal '', err
22     assert_match /^usage:/, out
23   end
24
25   def test_raw_stdin
26     out, err = capture_subprocess_io do
27       r,w = IO.pipe
28       wpid = fork do
29         r.close
30         w << 'foo'
31       end
32       w.close
33       assert_equal true, arv_put('--raw', {in: r})
34       r.close
35       Process.waitpid wpid
36     end
37     $stderr.write err
38     assert_match '', err
39     assert_equal "acbd18db4cc2f85cedef654fccc4a4d8+3\n", out
40   end
41
42   def test_raw_file
43     out, err = capture_subprocess_io do
44       assert_equal true, arv_put('--raw', './tmp/foo')
45     end
46     $stderr.write err
47     assert_match '', err
48     assert_equal "acbd18db4cc2f85cedef654fccc4a4d8+3\n", out
49   end
50
51   def test_raw_empty_file
52     out, err = capture_subprocess_io do
53       assert_equal true, arv_put('--raw', './tmp/empty_file')
54     end
55     $stderr.write err
56     assert_match '', err
57     assert_equal "d41d8cd98f00b204e9800998ecf8427e+0\n", out
58   end
59
60   def test_filename_arg_with_directory
61     out, err = capture_subprocess_io do
62       assert_equal(false, arv_put('--filename', 'foo', './tmp/empty_dir/.'),
63                    'arv-put --filename refuses directory')
64     end
65     assert_match /^usage:.*error:/m, err
66     assert_equal '', out
67   end
68
69   def test_filename_arg_with_multiple_files
70     out, err = capture_subprocess_io do
71       assert_equal(false, arv_put('--filename', 'foo',
72                                   './tmp/empty_file',
73                                   './tmp/empty_file'),
74                    'arv-put --filename refuses directory')
75     end
76     assert_match /^usage:.*error:/m, err
77     assert_equal '', out
78   end
79
80   def test_filename_arg_with_empty_file
81     out, err = capture_subprocess_io do
82       assert_equal true, arv_put('--filename', 'foo', './tmp/empty_file')
83     end
84     $stderr.write err
85     assert_match '', err
86     assert_equal "aa4f15cbf013142a7d98b1e273f9c661+45\n", out
87   end
88
89   def test_as_stream
90     out, err = capture_subprocess_io do
91       assert_equal true, arv_put('--as-stream', './tmp/foo')
92     end
93     $stderr.write err
94     assert_match '', err
95     assert_equal foo_manifest, out
96   end
97
98   def test_progress
99     out, err = capture_subprocess_io do
100       assert_equal true, arv_put('--manifest', '--progress', './tmp/foo')
101     end
102     assert_match /%/, err
103     assert_equal foo_manifest_locator+"\n", out
104   end
105
106   def test_batch_progress
107     out, err = capture_subprocess_io do
108       assert_equal true, arv_put('--manifest', '--batch-progress', './tmp/foo')
109     end
110     assert_match /: 0 written 3 total/, err
111     assert_match /: 3 written 3 total/, err
112     assert_equal foo_manifest_locator+"\n", out
113   end
114
115   def test_progress_and_batch_progress
116     out, err = capture_subprocess_io do
117       assert_equal(false,
118                    arv_put('--progress', '--batch-progress', './tmp/foo'),
119                    'arv-put --progress --batch-progress is contradictory')
120     end
121     assert_match /^usage:.*error:/m, err
122     assert_equal '', out
123   end
124
125   def test_read_from_implicit_stdin
126     test_read_from_stdin(specify_stdin_as='--manifest')
127   end
128
129   def test_read_from_dev_stdin
130     test_read_from_stdin(specify_stdin_as='/dev/stdin')
131   end
132
133   def test_read_from_stdin(specify_stdin_as='-')
134     out, err = capture_subprocess_io do
135       r,w = IO.pipe
136       wpid = fork do
137         r.close
138         w << 'foo'
139       end
140       w.close
141       assert_equal true, arv_put('--filename', 'foo', specify_stdin_as,
142                                  { in: r })
143       r.close
144       Process.waitpid wpid
145     end
146     $stderr.write err
147     assert_match '', err
148     assert_equal foo_manifest_locator+"\n", out
149   end
150
151   def test_read_from_implicit_stdin_implicit_manifest
152     test_read_from_stdin_implicit_manifest(specify_stdin_as=nil,
153                                            expect_filename='stdin')
154   end
155
156   def test_read_from_dev_stdin_implicit_manifest
157     test_read_from_stdin_implicit_manifest(specify_stdin_as='/dev/stdin')
158   end
159
160   def test_read_from_stdin_implicit_manifest(specify_stdin_as='-',
161                                              expect_filename=nil)
162     expect_filename = expect_filename || specify_stdin_as.split('/').last
163     out, err = capture_subprocess_io do
164       r,w = IO.pipe
165       wpid = fork do
166         r.close
167         w << 'foo'
168       end
169       w.close
170       args = []
171       args.push specify_stdin_as if specify_stdin_as
172       assert_equal true, arv_put(*args, { in: r })
173       r.close
174       Process.waitpid wpid
175     end
176     $stderr.write err
177     assert_match '', err
178     assert_equal(foo_manifest_locator(expect_filename)+"\n",
179                  out)
180   end
181
182   protected
183   def arv_put(*args)
184     system ['./bin/arv-put', 'arv-put'], *args
185   end
186
187   def foo_manifest(filename='foo')
188     ". #{Digest::MD5.hexdigest('foo')}+3 0:3:#{filename}\n"
189   end
190
191   def foo_manifest_locator(filename='foo')
192     Digest::MD5.hexdigest(foo_manifest(filename)) +
193       "+#{foo_manifest(filename).length}"
194   end
195 end