Merge pull request #503 from truecoach/bugfix/handle-504-error

jonnyom · web-flow · commit 7c2ca38674e9 · 2019-11-12T10:11:00.000Z
[feature] handle 504 response
diff --git a/lib/intercom/errors.rb b/lib/intercom/errors.rb
@@ -42,6 +42,9 @@ class ServerError < IntercomError; end
   # Raised when we have bad gateway errors.
   class BadGatewayError < IntercomError; end
 
+  # Raised when we have gateway timeout errors.
+  class GatewayTimeoutError < IntercomError; end
+
   # Raised when we experience a socket read timeout
   class ServiceUnavailableError < IntercomError; end
 
diff --git a/lib/intercom/request.rb b/lib/intercom/request.rb
@@ -168,6 +168,8 @@ def execute(target_base_url=nil, username:, secret: nil, read_timeout: 90, open_
         raise Intercom::BadGatewayError.new('Bad Gateway Error')
       elsif code == 503
         raise Intercom::ServiceUnavailableError.new('Service Unavailable')
+      elsif code == 504
+        raise Intercom::GatewayTimeoutError.new('Gateway Timeout')
       end
     end
 
diff --git a/spec/unit/intercom/request_spec.rb b/spec/unit/intercom/request_spec.rb
@@ -109,6 +109,15 @@ def execute!
 
       expect { execute! }.must_raise(Intercom::RateLimitExceeded)
     end
+
+    it 'raises a GatewayTimeoutError error when the response code is 504' do
+      stub_request(:any, uri).to_return(
+        status: 504,
+        body: '<html> <head><title>504 Gateway Time-out</title></head> <body bgcolor="white"> <center><h1>504 Gateway Time-out</h1></center> </body> </html>'
+      )
+
+      expect { execute! }.must_raise(Intercom::GatewayTimeoutError)
+    end
   end
 
   describe "application error handling" do
