Как я могу издеваться над AWS SDK (v2) с помощью rspec?
У меня есть класс, который читает / обрабатывает сообщения из очереди SQS, используя гем aws-sdk-rails (который является оболочкой для aws-sdk-ruby v2). Как мне высмеивать вызовы AWS, чтобы я мог проверить свой код, не обращаясь к внешним службам?
communator.rb:
class Communicator
def consume_messages
sqs_client = Aws::SQS::Client.new
# consume messages until the queue is empty
loop do
r = sqs_client.receive_message({
queue_url: "https://sqs.region.amazonaws.com/xxxxxxxxxxxx/foo",
visibility_timeout: 1,
max_number_of_messages: 1
})
break if (response.message.length == 0)
# process r.messages.first.body
r = sqs_client.delete_message({
queue_url: "https://sqs.region.amazonaws.com/xxxxxxxxxxxx/foo",
receipt_handle: r.messages.first.receipt_handle
})
end
end
end
2 ответа
Решение
Мне было трудно найти примеры насмешек над ресурсами AWS. Я потратил несколько дней, чтобы понять это, и хотел поделиться своими результатами по переполнению стека для потомков. Я использовал rspec-mocks ( удваивает и проверяет удвоения). Вот пример с communicator.rb
пример в вопросе.
communator_spec.rb:
RSpec.describe Communicator do
describe "#consume_messages" do
it "can use rspec doubles & verifying doubles to mock AWS SDK calls" do
sqs_client = instance_double(Aws::SQS::Client)
allow(Aws::SQS::Client).to receive(:new).and_return(sqs_client)
SQSResponse = Struct.new(:messages)
SQSMessage = Struct.new(:body, :receipt_handle)
response = SQSResponse.new([SQSMessage.new(File.read('data/expected_body.json'), "receipt_handle")])
empty_response = SQSResponse.new([])
allow(sqs_client).to receive(:receive_message).
and_return(response, empty_response)
allow(sqs_client).to receive(:delete_message).and_return(nil)
Communicator.new.consume_messages
end
end
end
AWS SDK уже обеспечивает заглушку. qv http://docs.aws.amazon.com/sdkforruby/api/Aws/ClientStubs.html для получения дополнительной информации (ссылка на официальную документацию.)