peak_detector_agent_spec.rb 3.3KB

    require 'spec_helper' describe Agents::PeakDetectorAgent do it_behaves_like LiquidInterpolatable before do @valid_params = { 'name' => "my peak detector agent", 'options' => { 'expected_receive_period_in_days' => "2", 'group_by_path' => "filter", 'value_path' => "count", 'message' => "A peak was found" } } @agent = Agents::PeakDetectorAgent.new(@valid_params) @agent.user = users(:bob) @agent.save! end describe "#receive" do it "tracks and groups by the group_by_path" do events = build_events(:keys => ['count', 'filter'], :values => [[1, "something"], [2, "something"], [3, "else"]]) @agent.receive events @agent.memory['data']['something'].map(&:first).should == [1, 2] @agent.memory['data']['something'].last.last.should be_within(10).of((100 - 1).hours.ago.to_i) @agent.memory['data']['else'].first.first.should == 3 @agent.memory['data']['else'].first.last.should be_within(10).of((100 - 2).hours.ago.to_i) end it "works without a group_by_path as well" do @agent.options['group_by_path'] = "" events = build_events(:keys => ['count'], :values => [[1], [2]]) @agent.receive events @agent.memory['data']['no_group'].map(&:first).should == [1, 2] end it "keeps a rolling window of data" do @agent.options['window_duration_in_days'] = 5/24.0 @agent.receive build_events(:keys => ['count'], :values => [1, 2, 3, 4, 5, 6, 7, 8].map {|i| [i]}, :pattern => { 'filter' => "something" }) @agent.memory['data']['something'].map(&:first).should == [4, 5, 6, 7, 8] end it "finds peaks" do build_events(:keys => ['count'], :values => [5, 6, 4, 5, 4, 5, 15, 11, # peak 8, 50, # ignored because it's too close to the first peak 4, 5].map {|i| [i]}, :pattern => { 'filter' => "something" }).each.with_index do |event, index| lambda { @agent.receive([event]) }.should change { @agent.events.count }.by( index == 6 ? 1 : 0 ) end @agent.events.last.payload['peak'].should == 15.0 @agent.memory['peaks']['something'].length.should == 1 end it "keeps a rolling window of peaks" do @agent.options['min_peak_spacing_in_days'] = 1/24.0 @agent.receive build_events(:keys => ['count'], :values => [1, 1, 1, 1, 1, 1, 10, 1, 1, 1, 1, 1, 1, 1, 10, 1].map {|i| [i]}, :pattern => { 'filter' => "something" }) @agent.memory['peaks']['something'].length.should == 2 end end describe "validation" do before do @agent.should be_valid end it "should validate presence of message" do @agent.options['message'] = nil @agent.should_not be_valid end it "should validate presence of expected_receive_period_in_days" do @agent.options['expected_receive_period_in_days'] = "" @agent.should_not be_valid end it "should validate presence of value_path" do @agent.options['value_path'] = "" @agent.should_not be_valid end end end