1 # Copyright 2012 Google Inc.
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
17 require 'google/api_client'
18 require 'google/api_client/version'
20 fixtures_path = File.expand_path('../../../fixtures', __FILE__)
22 describe Google::APIClient::UploadIO do
23 it 'should reject invalid file paths' do
25 media = Google::APIClient::UploadIO.new('doesnotexist', 'text/plain')
26 end).should raise_error
29 describe 'with a file' do
31 @file = File.expand_path('files/sample.txt', fixtures_path)
32 @media = Google::APIClient::UploadIO.new(@file, 'text/plain')
35 it 'should report the correct file length' do
36 @media.length.should == File.size(@file)
39 it 'should have a mime type' do
40 @media.content_type.should == 'text/plain'
44 describe 'with StringIO' do
46 @content = "hello world"
47 @media = Google::APIClient::UploadIO.new(StringIO.new(@content), 'text/plain', 'test.txt')
50 it 'should report the correct file length' do
51 @media.length.should == @content.length
54 it 'should have a mime type' do
55 @media.content_type.should == 'text/plain'
60 describe Google::APIClient::ResumableUpload do
62 @client = Google::APIClient.new
63 @drive = @client.discovered_api('drive', 'v1')
64 @file = File.expand_path('files/sample.txt', fixtures_path)
65 @media = Google::APIClient::UploadIO.new(@file, 'text/plain')
66 @uploader = Google::APIClient::ResumableUpload.new(
69 'https://www.googleapis.com/upload/drive/v1/files/12345')
72 it 'should consider 20x status as complete' do
73 api_client = stub('api', :execute => mock_result(200))
74 @uploader.send_chunk(api_client)
75 @uploader.complete?.should == true
78 it 'should consider 30x status as incomplete' do
79 api_client = stub('api', :execute => mock_result(308))
80 @uploader.send_chunk(api_client)
81 @uploader.complete?.should == false
82 @uploader.expired?.should == false
85 it 'should consider 40x status as fatal' do
86 api_client = stub('api', :execute => mock_result(404))
87 @uploader.send_chunk(api_client)
88 @uploader.expired?.should == true
91 it 'should detect changes to location' do
92 api_client = stub('api', :execute => mock_result(308, 'location' => 'https://www.googleapis.com/upload/drive/v1/files/abcdef'))
93 @uploader.send_chunk(api_client)
94 @uploader.location.should == 'https://www.googleapis.com/upload/drive/v1/files/abcdef'
97 it 'should resume from the saved range reported by the server' do
98 api_client = mock('api')
99 api_client.should_receive(:execute).and_return(mock_result(308, 'range' => '0-99'))
100 api_client.should_receive(:execute).with(
101 hash_including(:headers => hash_including(
102 "Content-Range" => "bytes 100-299/#{@media.length}",
103 "Content-Length" => "200"
104 ))).and_return(mock_result(308))
106 @uploader.chunk_size = 200
107 @uploader.send_chunk(api_client) # Send bytes 0-199, only 0-99 saved
108 @uploader.send_chunk(api_client) # Send bytes 100-299
111 it 'should resync the offset after 5xx errors' do
112 api_client = mock('api')
113 api_client.should_receive(:execute).and_return(mock_result(500))
114 api_client.should_receive(:execute).with(
115 hash_including(:headers => hash_including(
116 "Content-Range" => "bytes */#{@media.length}",
117 "Content-Length" => "0"
118 ))).and_return(mock_result(308, 'range' => '0-99'))
119 api_client.should_receive(:execute).with(
120 hash_including(:headers => hash_including(
121 "Content-Range" => "bytes 100-299/#{@media.length}",
122 "Content-Length" => "200"
123 ))).and_return(mock_result(308))
125 @uploader.chunk_size = 200
126 @uploader.send_chunk(api_client) # 500, invalidate
127 @uploader.send_chunk(api_client) # Just resyncs, doesn't actually upload
128 @uploader.send_chunk(api_client) # Send next chunk at correct range
131 def mock_result(status, headers = {})
132 reference = Google::APIClient::Reference.new(:api_method => @drive.files.insert)
133 stub('result', :status => status, :headers => headers, :reference => reference)