143 lines
4.0 KiB
Ruby
143 lines
4.0 KiB
Ruby
# Copyright 2015 Google Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
require 'spec_helper'
|
|
require 'google/apis/core/batch'
|
|
require 'google/apis/core/json_representation'
|
|
|
|
RSpec.describe Google::Apis::Core::BatchCommand do
|
|
include TestHelpers
|
|
include_context 'HTTP client'
|
|
|
|
let(:command) do
|
|
command = Google::Apis::Core::BatchCommand.new(:post, 'https://www.googleapis.com/batch')
|
|
end
|
|
|
|
let(:get_command) { Google::Apis::Core::HttpCommand.new(:get, 'https://www.googleapis.com/zoo/animals/1') }
|
|
|
|
let(:post_with_string_command) do
|
|
command = Google::Apis::Core::HttpCommand.new(:post, 'https://www.googleapis.com/zoo/animals/2')
|
|
command.body = 'Hello world'
|
|
command.header['Content-Type'] = 'text/plain'
|
|
command
|
|
end
|
|
|
|
let(:post_with_io_command) do
|
|
command = Google::Apis::Core::HttpCommand.new(:post, 'https://www.googleapis.com/zoo/animals/3')
|
|
command.body = StringIO.new('Goodbye!')
|
|
command.header['Content-Type'] = 'text/plain'
|
|
command
|
|
end
|
|
|
|
before(:example) do
|
|
allow(SecureRandom).to receive(:uuid).and_return('ffe23d1b-e8f7-47f5-8c01-2a30cf8ecb8f')
|
|
allow(Digest::SHA1).to receive(:hexdigest).and_return('123abc')
|
|
|
|
response = <<EOF
|
|
--batch123
|
|
Content-Type: application/http
|
|
Content-ID: <response-ffe23d1b-e8f7-47f5-8c01-2a30cf8ecb8f+0>
|
|
|
|
HTTP/1.1 200 OK
|
|
Content-Type: text/plain; charset=UTF-8
|
|
|
|
Hello
|
|
--batch123
|
|
Content-Type: application/http
|
|
Content-ID: <response-ffe23d1b-e8f7-47f5-8c01-2a30cf8ecb8f+2>
|
|
|
|
HTTP/1.1 500 Server Error
|
|
Content-Type: text/plain; charset=UTF-8
|
|
|
|
Error!
|
|
--batch123
|
|
Content-Type: application/http
|
|
Content-ID: <response-ffe23d1b-e8f7-47f5-8c01-2a30cf8ecb8f+1>
|
|
|
|
HTTP/1.1 200 OK
|
|
Content-Type: text/plain; charset=UTF-8
|
|
|
|
world
|
|
--batch123--
|
|
EOF
|
|
stub_request(:post, 'https://www.googleapis.com/batch')
|
|
.to_return(headers: { 'Content-Type' => 'multipart/mixed; boundary=batch123' }, body: response)
|
|
end
|
|
|
|
it 'should send content' do
|
|
b = ->(_res, _err) {}
|
|
command.add(get_command, &b)
|
|
command.add(post_with_string_command, &b)
|
|
command.add(post_with_io_command, &b)
|
|
command.execute(client)
|
|
|
|
expected_body = <<EOF.gsub(/\n/, "\r\n")
|
|
--123abc
|
|
Content-Type: application/http
|
|
Content-Id: <ffe23d1b-e8f7-47f5-8c01-2a30cf8ecb8f+0>
|
|
Content-Length: 58
|
|
Content-Transfer-Encoding: binary
|
|
|
|
GET /zoo/animals/1? HTTP/1.1
|
|
Host: www.googleapis.com
|
|
|
|
|
|
--123abc
|
|
Content-Type: application/http
|
|
Content-Id: <ffe23d1b-e8f7-47f5-8c01-2a30cf8ecb8f+1>
|
|
Content-Length: 96
|
|
Content-Transfer-Encoding: binary
|
|
|
|
POST /zoo/animals/2? HTTP/1.1
|
|
Content-Type: text/plain
|
|
Host: www.googleapis.com
|
|
|
|
Hello world
|
|
--123abc
|
|
Content-Type: application/http
|
|
Content-Id: <ffe23d1b-e8f7-47f5-8c01-2a30cf8ecb8f+2>
|
|
Content-Length: 93
|
|
Content-Transfer-Encoding: binary
|
|
|
|
POST /zoo/animals/3? HTTP/1.1
|
|
Content-Type: text/plain
|
|
Host: www.googleapis.com
|
|
|
|
Goodbye!
|
|
--123abc--
|
|
|
|
EOF
|
|
expect(a_request(:post, 'https://www.googleapis.com/batch').with(body: expected_body)).to have_been_made
|
|
end
|
|
|
|
it 'should send decode responses' do
|
|
expect do |b|
|
|
command.add(get_command) do |res, err|
|
|
b.to_proc.call(1, res, err)
|
|
end
|
|
command.add(post_with_string_command) do |res, err|
|
|
b.to_proc.call(2, res, err)
|
|
end
|
|
command.add(post_with_io_command) do |res, err|
|
|
b.to_proc.call(3, res, err)
|
|
end
|
|
command.execute(client)
|
|
end.to yield_successive_args([1, 'Hello', nil], [3, nil, an_instance_of(Google::Apis::ServerError)], [2, 'world', nil],)
|
|
end
|
|
|
|
it 'should raise error if batch is empty' do
|
|
expect { command.execute(client) }.to raise_error(Google::Apis::BatchError)
|
|
end
|
|
end
|