[aws] mock errors on too many addresses

lib/fog/aws/ec2.rb

@@ -77,6 +77,7 @@ module Fog
               :addresses => {},
               :instances => {},
               :key_pairs => {},
+              :limits => { :addresses => 5 },
               :security_groups => {
                 'default' => {
                   'groupDescription'  => 'default group',

lib/fog/aws/requests/ec2/allocate_address.rb

@@ -25,18 +25,24 @@ module Fog
 
         def allocate_address
           response = Excon::Response.new
-          response.status = 200
+          if describe_addresses.body['addressesSet'].size < @data[:limits][:addresses]
-          public_ip = Fog::AWS::Mock.ip_address
+            response.status = 200
-          data ={
+            public_ip = Fog::AWS::Mock.ip_address
-            'instanceId' => '',
+            data ={
-            'publicIp'   => public_ip
+              'instanceId' => '',
-          }
+              'publicIp'   => public_ip
-          @data[:addresses][public_ip] = data
+            }
-          response.body = {
+            @data[:addresses][public_ip] = data
-            'publicIp'  => public_ip,
+            response.body = {
-            'requestId' => Fog::AWS::Mock.request_id
+              'publicIp'  => public_ip,
-          }
+              'requestId' => Fog::AWS::Mock.request_id
-          response
+            }
+            response
+          else
+            response.status = 400
+            response.body = "<?xml version=\"1.0\"?><Response><Errors><Error><Code>AddressLimitExceeded</Code><Message>Too many addresses allocated</Message></Error></Errors><RequestID>#{Fog::AWS::Mock.request_id}</RequestID></Response>"
+            raise(Excon::Errors.status_error({:expects => 200}, response))
+          end
         end
 
       end