[aws|storage] Add mock for Fog::Storage::AWS#put_bucket_website

5apps · Jan 8, 2012 · d91d4e1 · d91d4e1
1 parent f07e594
commit d91d4e1
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/lib/fog/aws/requests/storage/put_bucket_website.rb b/lib/fog/aws/requests/storage/put_bucket_website.rb
@@ -44,6 +44,23 @@ def put_bucket_website(bucket_name, suffix, options = {})
         end
 
       end
+
+      class Mock # :nodoc:all
+
+        def put_bucket_website(bucket_name, suffix, options = {})
+          response = Excon::Response.new
+          if self.data[:buckets][bucket_name]
+            response.status = 200
+          else
+            response.status = 403
+            raise(Excon::Errors.status_error({:expects => 200}, response))
+          end
+
+          response
+        end
+
+      end
+
     end
   end
 end
diff --git a/tests/aws/requests/storage/bucket_tests.rb b/tests/aws/requests/storage/bucket_tests.rb
@@ -117,7 +117,6 @@
     end
 
     tests("#put_bucket_website('#{@aws_bucket_name}', 'index.html')").succeeds do
-      pending if Fog.mocking?
       Fog::Storage[:aws].put_bucket_website(@aws_bucket_name, 'index.html')
     end
 
@@ -222,6 +221,10 @@
       Fog::Storage[:aws].put_bucket_acl('fognonbucket', 'invalid')
     end
 
+    tests("#put_bucket_website('fognonbucket', 'index.html')").raises(Excon::Errors::Forbidden) do
+      Fog::Storage[:aws].put_bucket_website('fognonbucket', 'index.html')
+    end
+
   end
 
   # don't keep the bucket around