mirror of
https://github.com/fog/fog-aws.git
synced 2022-11-09 13:50:52 -05:00
87 lines
3 KiB
Ruby
87 lines
3 KiB
Ruby
module Fog
|
|
module Storage
|
|
class AWS
|
|
module SharedMockMethods
|
|
def define_mock_acl(bucket_name, object_name, options)
|
|
acl = options['x-amz-acl'] || 'private'
|
|
if !['private', 'public-read', 'public-read-write', 'authenticated-read'].include?(acl)
|
|
raise Excon::Errors::BadRequest.new('invalid x-amz-acl')
|
|
else
|
|
self.data[:acls][:object][bucket_name] ||= {}
|
|
self.data[:acls][:object][bucket_name][object_name] = self.class.acls(acl)
|
|
end
|
|
end
|
|
|
|
def parse_mock_data(data)
|
|
data = Fog::Storage.parse_data(data)
|
|
unless data[:body].is_a?(String)
|
|
data[:body] = data[:body].read
|
|
end
|
|
data
|
|
end
|
|
|
|
def verify_mock_bucket_exists(bucket_name)
|
|
if (bucket = self.data[:buckets][bucket_name])
|
|
return bucket
|
|
end
|
|
|
|
response = Excon::Response.new
|
|
response.status = 404
|
|
raise(Excon::Errors.status_error({:expects => 200}, response))
|
|
end
|
|
|
|
def get_upload_info(bucket_name, upload_id, delete = false)
|
|
if delete
|
|
upload_info = self.data[:multipart_uploads][bucket_name].delete(upload_id)
|
|
else
|
|
upload_info = self.data[:multipart_uploads][bucket_name][upload_id]
|
|
end
|
|
|
|
if !upload_info
|
|
response = Excon::Response.new
|
|
response.status = 404
|
|
raise(Excon::Errors.status_error({:expects => 200}, response))
|
|
end
|
|
|
|
upload_info
|
|
end
|
|
|
|
def store_mock_object(bucket, object_name, body, options)
|
|
object = {
|
|
:body => body,
|
|
'Content-Type' => options['Content-Type'],
|
|
'ETag' => Digest::MD5.hexdigest(body),
|
|
'Key' => object_name,
|
|
'Last-Modified' => Fog::Time.now.to_date_header,
|
|
'Content-Length' => options['Content-Length'],
|
|
'StorageClass' => options['x-amz-storage-class'] || 'STANDARD',
|
|
'VersionId' => bucket[:versioning] == 'Enabled' ? Fog::Mock.random_base64(32) : 'null'
|
|
}
|
|
|
|
for key, value in options
|
|
case key
|
|
when 'Cache-Control', 'Content-Disposition', 'Content-Encoding', 'Content-MD5', 'Expires', /^x-amz-meta-/
|
|
object[key] = value
|
|
end
|
|
end
|
|
|
|
if bucket[:versioning]
|
|
bucket[:objects][object_name] ||= []
|
|
|
|
# When versioning is suspended, putting an object will create a new 'null' version if the latest version
|
|
# is a value other than 'null', otherwise it will replace the latest version.
|
|
if bucket[:versioning] == 'Suspended' && bucket[:objects][object_name].first['VersionId'] == 'null'
|
|
bucket[:objects][object_name].shift
|
|
end
|
|
|
|
bucket[:objects][object_name].unshift(object)
|
|
else
|
|
bucket[:objects][object_name] = [object]
|
|
end
|
|
|
|
object
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|