[rackspace|lb] added error page requests

fog · Dec 15, 2011 · 87778f5 · 87778f5
1 parent 9c3bf8e
commit 87778f5
Show file tree

Hide file tree

Showing 6 changed files with 91 additions and 0 deletions.
diff --git a/lib/fog/rackspace/load_balancers.rb b/lib/fog/rackspace/load_balancers.rb
@@ -62,6 +62,9 @@ class BadRequest < Fog::Rackspace::Errors::BadRequest; end
       request :remove_monitor
       request :get_usage
       request :get_load_balancer_usage
+      request :get_error_page
+      request :set_error_page
+      request :remove_error_page
 
       module Shared
 

diff --git a/lib/fog/rackspace/requests/load_balancers/get_error_page.rb b/lib/fog/rackspace/requests/load_balancers/get_error_page.rb
@@ -0,0 +1,15 @@
+module Fog
+  module Rackspace
+    class LoadBalancers
+      class Real
+        def get_error_page(load_balancer_id)
+          request(
+            :expects => 200,
+            :path => "loadbalancers/#{load_balancer_id}/errorpage",
+            :method => 'GET'
+          )
+         end
+      end
+    end
+  end
+end
diff --git a/lib/fog/rackspace/requests/load_balancers/remove_error_page.rb b/lib/fog/rackspace/requests/load_balancers/remove_error_page.rb
@@ -0,0 +1,15 @@
+module Fog
+  module Rackspace
+    class LoadBalancers
+      class Real
+        def remove_error_page(load_balancer_id)
+          request(
+            :expects => [200, 202],
+            :path => "loadbalancers/#{load_balancer_id}/errorpage",
+            :method => 'DELETE'
+          )
+         end
+      end
+    end
+  end
+end
diff --git a/lib/fog/rackspace/requests/load_balancers/set_error_page.rb b/lib/fog/rackspace/requests/load_balancers/set_error_page.rb
@@ -0,0 +1,21 @@
+module Fog
+  module Rackspace
+    class LoadBalancers
+      class Real
+        def set_error_page(load_balancer_id, content)
+          data = {
+            'errorpage' => {
+              'content' => content
+            }
+          }
+          request(
+            :body     => MultiJson.encode(data),
+            :expects  => [200, 202],
+            :path     => "loadbalancers/#{load_balancer_id}/errorpage",
+            :method   => 'PUT'
+          )
+         end
+      end
+    end
+  end
+end
diff --git a/tests/rackspace/requests/load_balancers/error_page_tests.rb b/tests/rackspace/requests/load_balancers/error_page_tests.rb
@@ -0,0 +1,31 @@
+Shindo.tests('Fog::Rackspace::LoadBalancers | error_page', ['rackspace', 'loadbalancers']) do
+
+  pending if Fog.mocking?
+
+  given_a_load_balancer_service do
+    given_a_load_balancer do
+      tests('success') do
+
+        @lb.wait_for { ready? }
+        tests("#get_error_page(#{@lb.id})").formats(ERROR_PAGE_FORMAT) do
+          @service.get_error_page(@lb.id).body
+        end
+
+        @lb.wait_for { ready? }
+        tests("#set_error_page(#{@lb.id}, '<html><body>hi!</body></html>')").succeeds do
+          @service.set_error_page(@lb.id, '<html><body>hi!</body></html>')
+        end
+
+        @lb.wait_for { ready? }
+        tests("#get_error_page(#{@lb.id})").formats(ERROR_PAGE_FORMAT) do
+          @service.get_error_page(@lb.id).body
+        end
+
+        @lb.wait_for { ready? }
+        tests("#remove_error_page()").succeeds do
+          @service.remove_error_page(@lb.id)
+        end
+      end
+    end
+  end
+end
diff --git a/tests/rackspace/requests/load_balancers/helper.rb b/tests/rackspace/requests/load_balancers/helper.rb
@@ -131,3 +131,9 @@
     'updated' => { 'time' => String },
   }.merge(CONNECTION_LOGGING_FORMAT)
 }
+
+ERROR_PAGE_FORMAT = {
+  'errorpage' => {
+    'content' => String
+  }
+}