count_hosts.rb in brightbox-cli-0.15.0

- old
+ new

@@ -11,18 +11,37 @@
         # ==== Returns
         # * response<~Excon::Response>: 
         #   * body<~Hash>
         #     * 'count'<~Integer> 
         #   * 'status'<~Integer> - 200 indicates success
-        def count_hosts( zone_id)
+        def count_hosts(zone_id)
           request(
             :expects  => 200,
             :method   => 'GET',
             :parser   => Fog::Parsers::DNS::Zerigo::CountHosts.new,
             :path     => "/api/1.1/zones/#{zone_id}/hosts/count.xml"
           )
         end
 
+      end
+
+      class Mock # :nodoc:all
+        def count_hosts(zone_id)
+          zone = find_by_zone_id(zone_id)
+
+          response = Excon::Response.new
+
+          if zone
+            response.status = 200
+            response.body = {
+              'count' => zone['hosts'].size
+            }
+          else
+            response.status = 404
+          end
+
+          response
+        end
       end
     end
   end
 end