Merge branch 'public-body-stats-fixes' into rails-3-develop

1 files changed, 15 insertions, 6 deletions

diff --git a/app/models/info_request.rb b/app/models/info_request.rb
index eba620f53..9463a236e 100644
--- a/app/models/info_request.rb
+++ b/app/models/info_request.rb
@@ -1182,14 +1182,23 @@ public
 
     after_save :update_counter_cache
     after_destroy :update_counter_cache
+    # This method updates the count columns of the PublicBody that
+    # store the number of "not held", "to some extent successful" and
+    # "both visible and classified" requests when saving or destroying
+    # an InfoRequest associated with the body:
     def update_counter_cache
         PublicBody.skip_callback(:save, :after, :purge_in_cache)
-        self.public_body.info_requests_not_held_count = InfoRequest.where(
-            :public_body_id => self.public_body.id,
-            :described_state => 'not_held').count
-        self.public_body.info_requests_successful_count = InfoRequest.where(
-            :public_body_id => self.public_body.id,
-            :described_state => ['successful', 'partially_successful']).count
+        basic_params = {
+            :public_body_id => self.public_body_id,
+            :awaiting_description => false,
+            :prominence => 'normal'
+        }
+        [['info_requests_not_held_count', {:described_state => 'not_held'}],
+         ['info_requests_successful_count', {:described_state => ['successful', 'partially_successful']}],
+         ['info_requests_visible_classified_count', {}]].each do |column, extra_params|
+            params = basic_params.clone.update extra_params
+            self.public_body.send "#{column}=", InfoRequest.where(params).count
+        end
         self.public_body.without_revision do
             public_body.no_xapian_reindex = true
             public_body.save