1 files changed, 87 insertions, 0 deletions
diff --git a/spec/controllers/public_body_controller_spec.rb b/spec/controllers/public_body_controller_spec.rb
index 34caa40ea..6800765f2 100644
--- a/spec/controllers/public_body_controller_spec.rb
+++ b/spec/controllers/public_body_controller_spec.rb
@@ -308,6 +308,93 @@ describe PublicBodyController, "when showing public body statistics" do
 
 end
 
+describe PublicBodyController, "when converting data for graphing" do
+
+    before(:each) do
+        @raw_count_data = PublicBody.get_request_totals(n=3,
+                                                        highest=true,
+                                                        minimum_requests=1)
+        @percentages_data = PublicBody.get_request_percentages(
+            column='info_requests_successful_count',
+            n=3,
+            highest=false,
+            minimum_requests=1)
+    end
+
+    it "should not include the real public body model instance" do
+        to_draw = controller.simplify_stats_for_graphs(@raw_count_data,
+                                                       column='blah_blah',
+                                                       percentages=false,
+                                                       {} )
+        to_draw['public_bodies'][0].class.should == Hash
+        to_draw['public_bodies'][0].has_key?('request_email').should be_false
+    end
+
+    it "should generate the expected id" do
+        to_draw = controller.simplify_stats_for_graphs(@raw_count_data,
+                                                       column='blah_blah',
+                                                       percentages=false,
+                                                       {:highest => true} )
+        to_draw['id'].should == "blah_blah-highest"
+        to_draw = controller.simplify_stats_for_graphs(@raw_count_data,
+                                                       column='blah_blah',
+                                                       percentages=false,
+                                                       {:highest => false} )
+        to_draw['id'].should == "blah_blah-lowest"
+    end
+
+    it "should have exactly the expected keys" do
+        to_draw = controller.simplify_stats_for_graphs(@raw_count_data,
+                                                       column='blah_blah',
+                                                       percentages=false,
+                                                       {} )
+        to_draw.keys.sort.should == ["errorbars", "id", "public_bodies",
+                                     "title", "tooltips", "totals",
+                                     "x_axis", "x_ticks", "x_values",
+                                     "y_axis", "y_max", "y_values"]
+
+        to_draw = controller.simplify_stats_for_graphs(@percentages_data,
+                                                       column='whatever',
+                                                       percentages=true,
+                                                       {})
+        to_draw.keys.sort.should == ["cis_above", "cis_below",
+                                     "errorbars", "id", "public_bodies",
+                                     "title", "tooltips", "totals",
+                                     "x_axis", "x_ticks", "x_values",
+                                     "y_axis", "y_max", "y_values"]
+    end
+
+    it "should have values of the expected class and length" do
+        [controller.simplify_stats_for_graphs(@raw_count_data,
+                                              column='blah_blah',
+                                              percentages=false,
+                                              {}),
+         controller.simplify_stats_for_graphs(@percentages_data,
+                                              column='whatever',
+                                              percentages=true,
+                                              {})].each do |to_draw|
+            per_pb_keys = ["cis_above", "cis_below", "public_bodies",
+                           "tooltips", "totals", "x_ticks", "x_values",
+                           "y_values"]
+            # These should be all be arrays with one element per public body:
+            per_pb_keys.each do |key|
+                if to_draw.has_key? key
+                    to_draw[key].class.should == Array
+                    to_draw[key].length.should eq(3), "for key #{key}"
+                end
+            end
+            # Just check that the rest aren't of class Array:
+            to_draw.keys.each do |key|
+                unless per_pb_keys.include? key
+                    to_draw[key].class.should_not eq(Array), "for key #{key}"
+                end
+            end
+        end
+    end
+
+end
+
+
 describe PublicBodyController, "when doing type ahead searches" do
 
     render_views