X-Git-Url: https://review.fuel-infra.org/gitweb?a=blobdiff_plain;f=spec%2Funit%2Fplugins%2Fmcollective%2Faggregate%2Fsummary_spec.rb;fp=spec%2Funit%2Fplugins%2Fmcollective%2Faggregate%2Fsummary_spec.rb;h=1f6f71cfde3aeab1bb6a30e2a2d580429ff0333e;hb=b87d2f4e68281062df1913440ca5753ae63314a9;hp=0000000000000000000000000000000000000000;hpb=ab0ea530b8ac956091f17b104ab2311336cfc250;p=packages%2Fprecise%2Fmcollective.git

diff --git a/spec/unit/plugins/mcollective/aggregate/summary_spec.rb b/spec/unit/plugins/mcollective/aggregate/summary_spec.rb
new file mode 100644
index 0000000..1f6f71c
--- /dev/null
+++ b/spec/unit/plugins/mcollective/aggregate/summary_spec.rb
@@ -0,0 +1,55 @@
+#!/usr/bin/env rspec
+
+require 'spec_helper'
+require File.dirname(__FILE__) + "/../../../../../plugins/mcollective/aggregate/summary.rb"
+
+module MCollective
+  class Aggregate
+    describe Summary do
+      describe "#startup_hook" do
+        it "should set the correct result hash" do
+          result = Summary.new(:test, [], "%d", :test_action)
+          result.result.should == {:value => {}, :type => :collection, :output => :test}
+          result.aggregate_format.should == "%d"
+        end
+
+        it "should set a defauly aggregate_format if one isn't defined" do
+          result = Summary.new(:test, [], nil, :test_action)
+          result.aggregate_format.should == :calculate
+        end
+      end
+
+      describe "#process_result" do
+        it "should add the value to the result hash" do
+          sum = Summary.new(:test, [], "%d", :test_action)
+          sum.process_result(:foo, {:test => :foo})
+          sum.result[:value].should == {:foo => 1}
+        end
+
+        it "should add the reply values to the result hash if value is an array" do
+          sum = Summary.new(:test, [], "%d", :test_action)
+          sum.process_result([:foo, :foo, :bar], {:test => [:foo, :foo, :bar]})
+          sum.result[:value].should == {:foo => 2, :bar => 1}
+        end
+      end
+
+      describe "#summarize" do
+        it "should calculate an attractive format" do
+          sum = Summary.new(:test, [], nil, :test_action)
+          sum.result[:value] = {"shrt" => 1, "long key" => 1}
+          sum.summarize.aggregate_format.should == "%8s = %s"
+        end
+
+        it "should calculate an attractive format when result type is not a string" do
+          sum1 = Summary.new(:test, [], nil, :test_action)
+          sum1.result[:value] = {true => 4, false => 5}
+          sum1.summarize.aggregate_format.should == "%5s = %s"
+
+          sum2 = Summary.new(:test, [], nil, :test_action)
+          sum2.result[:value] = {1 => 1, 10 => 2}
+          sum2.summarize.aggregate_format.should == "%2s = %s"
+        end
+      end
+    end
+  end
+end