kotovalexarian-likes-github/fog--fog
1
0
Fork 0
mirror of https://github.com/fog/fog.git synced 2022-11-09 13:51:43 -05:00
Code Releases Activity

[aws/sqs] Adding SQS mocking support.

Browse source
This commit is contained in:
Istvan Hoka 2011-09-18 23:00:58 +03:00
parent f812565897
commit a5759ceb7b
13 changed files with 301 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lib/fog/aws.rb
View file

@ -177,6 +177,8 @@ module Fog
class << self
alias :reserved_instances_id :request_id
alias :reserved_instances_offering_id :request_id
alias :sqs_message_id :request_id
alias :sqs_sender_id :request_id
end
def self.reservation_id

30
lib/fog/aws/requests/sqs/change_message_visibility.rb
View file

@ -28,6 +28,36 @@ module Fog
end
class Mock
def change_message_visibility(queue_url, receipt_handle, visibility_timeout)
Excon::Response.new.tap do |response|
if (queue = data[:queues][queue_url])
message_id, _ = queue[:receipt_handles].find { |message_id, receipts|
receipts.keys.include?(receipt_handle)
}
if message_id
queue[:messages][message_id]['Attributes']['VisibilityTimeout'] = visibility_timeout
response.body = {
'ResponseMetadata' => {
'RequestId' => Fog::AWS::Mock.request_id
}
}
response.status = 200
else
response.status = 404
raise(Excon::Errors.status_error({:expects => 200}, response))
end
else
response.status = 404
raise(Excon::Errors.status_error({:expects => 200}, response))
end
end
end
end
end
end
end

34
lib/fog/aws/requests/sqs/create_queue.rb
View file

@ -26,6 +26,40 @@ module Fog
end
class Mock
def create_queue(name, options = {})
Excon::Response.new.tap do |response|
response.status = 200
now = Time.now
queue_url = "https://queue.amazonaws.com/#{data[:owner_id]}/#{name}"
queue = {
'QueueName' => name,
'Attributes' => {
'VisibilityTimeout' => 30,
'ApproximateNumberOfMessages' => 0,
'ApproximateNumberOfMessagesNotVisible' => 0,
'CreatedTimestamp' => now,
'LastModifiedTimestamp' => now,
'QueueArn' => Fog::AWS::Mock.arn('sqs', 'us-east-1', data[:owner_id], name),
'MaximumMessageSize' => 8192,
'MessageRetentionPeriod' => 345600
},
:messages => {},
:receipt_handles => {}
}
data[:queues][queue_url] = queue unless data[:queues][queue_url]
response.body = {
'ResponseMetadata' => {
'RequestId' => Fog::AWS::Mock.request_id
},
'QueueUrl' => queue_url
}
end
end
end
end
end
end

29
lib/fog/aws/requests/sqs/delete_message.rb
View file

@ -26,6 +26,35 @@ module Fog
end
class Mock
def delete_message(queue_url, receipt_handle)
Excon::Response.new.tap do |response|
if (queue = data[:queues][queue_url])
message_id, _ = queue[:receipt_handles].find { |msg_id, receipts|
receipts.keys.include?(receipt_handle)
}
if message_id
queue[:receipt_handles].delete(message_id)
queue[:messages].delete(message_id)
queue['Attributes']['LastModifiedTimestamp'] = Time.now
end
response.body = {
'ResponseMetadata' => {
'RequestId' => Fog::AWS::Mock.request_id
}
}
response.status = 200
else
response.status = 404
raise(Excon::Errors.status_error({:expects => 200}, response))
end
end
end
end
end
end
end

23
lib/fog/aws/requests/sqs/delete_queue.rb
View file

@ -24,6 +24,29 @@ module Fog
end
class Mock
def delete_queue(queue_url)
Excon::Response.new.tap do |response|
if (queue = data[:queues][queue_url])
response.status = 200
data[:queues].delete(queue_url)
response.body = {
'ResponseMetadata' => {
'RequestId' => Fog::AWS::Mock.request_id
}
}
else
response.status = 404
raise(Excon::Errors.status_error({:expects => 200}, response))
end
end
end
end
end
end
end

21
lib/fog/aws/requests/sqs/get_queue_attributes.rb
View file

@ -26,6 +26,27 @@ module Fog
end
class Mock
def get_queue_attributes(queue_url, attribute_name)
Excon::Response.new.tap do |response|
if (queue = data[:queues][queue_url])
response.status = 200
response.body = {
'ResponseMetadata' => {
'RequestId' => Fog::AWS::Mock.request_id
},
'Attributes' => queue['Attributes']
}
else
response.status = 404
raise(Excon::Errors.status_error({:expects => 200}, response))
end
end
end
end
end
end
end

14
lib/fog/aws/requests/sqs/list_queues.rb
View file

@ -23,6 +23,20 @@ module Fog
end
end
class Mock
def list_queues(options = {})
Excon::Response.new.tap do |response|
response.status = 200
response.body = {
'ResponseMetadata' => {
'RequestId' => Fog::AWS::Mock.request_id
},
'QueueUrls' => data[:queues].keys
}
end
end
end
end
end
end

54
lib/fog/aws/requests/sqs/receive_message.rb
View file

@ -29,6 +29,60 @@ module Fog
end
class Mock
def receive_message(queue_url, options = {})
Excon::Response.new.tap do |response|
if (queue = data[:queues][queue_url])
max_number_of_messages = options['MaxNumberOfMessages'] || 1
now = Time.now
keys = queue[:messages].keys[0, max_number_of_messages]
messages = queue[:messages].values_at(*keys).map do |m|
message_id = m['MessageId']
invisible = if (received_handles = queue[:receipt_handles][message_id])
visibility_timeout = m['Attributes']['VisibilityTimeout'] || queue['Attributes']['VisibilityTimeout']
received_handles.any? { |handle, time| now < time + visibility_timeout }
else
false
end
if invisible
nil
else
receipt_handle = Fog::Mock.random_base64(300)
queue[:receipt_handles][message_id] ||= {}
queue[:receipt_handles][message_id][receipt_handle] = now
m['Attributes'].tap do |attrs|
attrs['ApproximateFirstReceiveTimestamp'] ||= now
attrs['ApproximateReceiveCount'] = (attrs['ApproximateReceiveCount'] || 0) + 1
end
m.merge({
'ReceiptHandle' => receipt_handle
})
end
end.compact
response.body = {
'ResponseMetadata' => {
'RequestId' => Fog::AWS::Mock.request_id
},
'Message' => messages
}
response.status = 200
else
response.status = 404
raise(Excon::Errors.status_error({:expects => 200}, response))
end
end
end
end
end
end
end

38
lib/fog/aws/requests/sqs/send_message.rb
View file

@ -26,6 +26,44 @@ module Fog
end
class Mock
def send_message(queue_url, message)
Excon::Response.new.tap do |response|
if (queue = data[:queues][queue_url])
response.status = 200
now = Time.now
message_id = Fog::AWS::Mock.sqs_message_id
md5 = Digest::MD5.hexdigest(message)
queue[:messages][message_id] = {
'MessageId' => message_id,
'Body' => message,
'MD5OfBody' => md5,
'Attributes' => {
'SenderId' => Fog::AWS::Mock.sqs_message_id,
'SentTimestamp' => now
}
}
queue['Attributes']['LastModifiedTimestamp'] = now
response.body = {
'ResponseMetadata' => {
'RequestId' => Fog::AWS::Mock.request_id
},
'MessageId' => message_id,
'MD5OfMessageBody' => md5
}
else
response.status = 404
raise(Excon::Errors.status_error({:expects => 200}, response))
end
end
end
end
end
end
end

18
lib/fog/aws/requests/sqs/set_queue_attributes.rb
View file

@ -28,6 +28,24 @@ module Fog
end
class Mock
def set_queue_attributes(queue_url, attribute_name, attribute_value)
Excon::Response.new.tap do |response|
if (queue = data[:queues][queue_url])
response.status = 200
queue['Attributes'][attribute_name] = attribute_value
response.body = {
'ResponseMetadata' => {
'RequestId' => Fog::AWS::Mock.request_id
}
}
else
response.status = 404
raise(Excon::Errors.status_error({:expects => 200}, response))
end
end
end
end
end
end
end

30
lib/fog/aws/sqs.rb
View file

@ -19,8 +19,38 @@ module Fog
request :set_queue_attributes
class Mock
def self.data
@data ||= Hash.new do |hash, region|
owner_id = Fog::AWS::Mock.owner_id
hash[region] = Hash.new do |region_hash, key|
region_hash[key] = {
:owner_id => Fog::AWS::Mock.owner_id,
:queues => {}
}
end
end
end
def self.reset
@data = nil
end
def initialize(options={})
@aws_access_key_id = options[:aws_access_key_id]
@region = options[:region] || 'us-east-1'
unless ['ap-northeast-1', 'ap-southeast-1', 'eu-west-1', 'us-east-1', 'us-west-1'].include?(@region)
raise ArgumentError, "Unknown region: #{@region.inspect}"
end
end
def data
self.class.data[@region][@aws_access_key_id]
end
def reset_data
self.class.data[@region].delete(@aws_access_key_id)
end
end

8
tests/aws/requests/sqs/message_tests.rb
View file

@ -2,9 +2,7 @@ Shindo.tests('AWS::SQS | message requests', ['aws']) do
tests('success') do
unless Fog.mocking?
@queue_url = Fog::AWS[:sqs].create_queue('fog_message_tests').body['QueueUrl']
end
@queue_url = Fog::AWS[:sqs].create_queue('fog_message_tests').body['QueueUrl']
send_message_format = AWS::SQS::Formats::BASIC.merge({
'MessageId' => String,
@ -12,7 +10,6 @@ Shindo.tests('AWS::SQS | message requests', ['aws']) do
})
tests("#send_message('#{@queue_url}', 'message')").formats(send_message_format) do
pending if Fog.mocking?
Fog::AWS[:sqs].send_message(@queue_url, 'message').body
end
@ -32,19 +29,16 @@ Shindo.tests('AWS::SQS | message requests', ['aws']) do
})
tests("#receive_message").formats(receive_message_format) do
pending if Fog.mocking?
data = Fog::AWS[:sqs].receive_message(@queue_url).body
@receipt_handle = data['Message'].first['ReceiptHandle']
data
end
tests("#change_message_visibility('#{@queue_url}, '#{@receipt_handle}', 60)").formats(AWS::SQS::Formats::BASIC) do
pending if Fog.mocking?
Fog::AWS[:sqs].change_message_visibility(@queue_url, @receipt_handle, 60).body
end
tests("#delete_message('#{@queue_url}', '#{@receipt_handle}')").formats(AWS::SQS::Formats::BASIC) do
pending if Fog.mocking?
Fog::AWS[:sqs].delete_message(@queue_url, @receipt_handle).body
end

5
tests/aws/requests/sqs/queue_tests.rb
View file

@ -7,7 +7,6 @@ Shindo.tests('AWS::SQS | queue requests', ['aws']) do
})
tests("#create_queue('fog_queue_tests')").formats(create_queue_format) do
pending if Fog.mocking?
data = Fog::AWS[:sqs].create_queue('fog_queue_tests').body
@queue_url = data['QueueUrl']
data
@ -18,12 +17,10 @@ Shindo.tests('AWS::SQS | queue requests', ['aws']) do
})
tests("#list_queues").formats(list_queues_format) do
pending if Fog.mocking?
Fog::AWS[:sqs].list_queues.body
end
tests("#set_queue_attributes('#{@queue_url}', 'VisibilityTimeout', 60)").formats(AWS::SQS::Formats::BASIC) do
pending if Fog.mocking?
Fog::AWS[:sqs].set_queue_attributes(@queue_url, 'VisibilityTimeout', 60).body
end
@ -41,12 +38,10 @@ Shindo.tests('AWS::SQS | queue requests', ['aws']) do
})
tests("#get_queue_attributes('#{@queue_url}', 'All')").formats(get_queue_attributes_format) do
pending if Fog.mocking?
Fog::AWS[:sqs].get_queue_attributes(@queue_url, 'All').body
end
tests("#delete_queue('#{@queue_url}')").formats(AWS::SQS::Formats::BASIC) do
pending if Fog.mocking?
Fog::AWS[:sqs].delete_queue(@queue_url).body
end