Add ability to specify s3 restore object tier in post_object_restore #740

Author: Ankk98

lib/fog/aws/requests/storage/post_object_restore.rb

@@ -7,6 +7,7 @@ class Real
         # @param bucket_name [String] Name of bucket containing object
         # @param object_name [String] Name of object to restore
         # @option days [Integer] Number of days to restore object for. Defaults to 100000 (a very long time)
+        # @option tier [String] Glacier retrieval tier. Can be 'Standard', 'Bulk', or 'Expedited'
         #
         # @return [Excon::Response] response:
         #   * status [Integer] 200 (OK) Object is previously restored
@@ -15,11 +16,16 @@ class Real
         #
         # @see http://docs.amazonwebservices.com/AmazonS3/latest/API/RESTObjectPOSTrestore.html
         #
-        def post_object_restore(bucket_name, object_name, days = 100000)
+        def post_object_restore(bucket_name, object_name, days = 100000, tier = nil)
           raise ArgumentError.new('bucket_name is required') unless bucket_name
           raise ArgumentError.new('object_name is required') unless object_name
 
-          data = '<RestoreRequest xmlns="http://s3.amazonaws.com/doc/2006-3-01"><Days>' + days.to_s + '</Days></RestoreRequest>'
+          data = '<RestoreRequest xmlns="http://s3.amazonaws.com/doc/2006-03-01/">'
+          data += '<Days>' + days.to_s + '</Days>'
+          if tier
+            data += '<GlacierJobParameters><Tier>' + tier + '</Tier></GlacierJobParameters>'
+          end
+          data += '</RestoreRequest>'
 
           headers = {}
           headers['Content-MD5'] = Base64.encode64(OpenSSL::Digest::MD5.digest(data)).strip
@@ -39,7 +45,7 @@ def post_object_restore(bucket_name, object_name, days = 100000)
       end
 
       class Mock # :nodoc:all
-        def post_object_restore(bucket_name, object_name, days = 100000)
+        def post_object_restore(bucket_name, object_name, days = 100000, tier = nil)
           response = get_object(bucket_name, object_name)
           response.body = nil
           response

tests/requests/storage/object_tests.rb

@@ -96,6 +96,11 @@
       Fog::Storage[:aws].post_object_restore(@directory.identity, 'fog_object')
     end
 
+    tests("#post_object_restore('#{@directory.identity}', 'fog_object', 5, 'Bulk')").succeeds do
+      pending unless Fog.mocking?
+      Fog::Storage[:aws].post_object_restore(@directory.identity, 'fog_object', 5, 'Bulk')
+    end
+
     tests("#put_object_acl('#{@directory.identity}', 'fog_object', 'private')").succeeds do
       Fog::Storage[:aws].put_object_acl(@directory.identity, 'fog_object', 'private')
     end