Adding count_results to Custodian::Article

lib/custodian/article_class.rb

@@ -1,7 +1,12 @@
 module Custodian
   class Article
     require 'hpricot'
-
+
+    def self.count(conditions)
+      conditions = normalise_conditions(conditions)
+      count_results(request('/content/search', conditions))
+    end
+
     def self.find(type, conditions={})
       if type.class == Fixnum
         find_article_by_id(type)
@@ -31,6 +36,10 @@ def self.extract_articles(xml)
     def self.extract_article(xml)
       Article.new(Hpricot(xml).search("/content").first)
     end
+
+    def self.count_results(xml)
+      Hpricot(xml).search("/search").first.attributes["count"].to_i
+    end
 
     private
 

spec/article_spec.rb

@@ -33,6 +33,17 @@
   end
 end
 
+describe "Counting results for a particular query" do
+  before(:each) do
+    Custodian::Article.stub!(:count_tags).and_return(Hpricot(File.open(File.dirname(__FILE__) + '/search.xml')).search("/search").first.attributes["count"].to_i)
+  end
+
+  it "should return the total number of available results for a query" do
+    Custodian::Article.count_tags(:q => "uk").should == 1007029
+  end
+end
+
+
 describe "An article" do
   before(:each) do
     @article = Custodian::Article.new(Hpricot(File.open(File.dirname(__FILE__) + '/article.xml')))