flapjack 0.8.10 → 0.8.11

data/spec/lib/flapjack/gateways/jsonapi/medium_methods_spec.rb

@@ -0,0 +1,232 @@
+require 'spec_helper'
+require 'flapjack/gateways/jsonapi'
+
+describe 'Flapjack::Gateways::JSONAPI::MediumMethods', :sinatra => true, :logger => true do
+
+  include_context "jsonapi"
+
+  let(:medium_data) {
+    {:type => 'email',
+     :address => 'abc@example.com',
+     :interval => 120,
+     :rollup_threshold => 3
+    }
+  }
+
+  let(:contact) { double(Flapjack::Data::Contact, :id => '21') }
+
+  let(:semaphore) {
+    double(Flapjack::Data::Semaphore, :resource => 'folly',
+           :key => 'semaphores:folly', :expiry => 30, :token => 'spatulas-R-us')
+  }
+
+  it "returns a single medium" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :redis => redis, :logger => @logger).and_return(contact)
+
+    expect(contact).to receive(:media_list).and_return([medium_data[:type]])
+    expect(contact).to receive(:media).and_return(medium_data[:type] => medium_data[:address])
+    expect(contact).to receive(:media_intervals).and_return(medium_data[:type] => medium_data[:interval])
+    expect(contact).to receive(:media_rollup_thresholds).and_return(medium_data[:type] => medium_data[:rollup_threshold])
+
+    aget "/media/#{contact.id}_#{medium_data[:type]}"
+    expect(last_response).to be_ok
+    expect(last_response.body).to eq({:media => [{:id => "#{contact.id}_#{medium_data[:type]}"}.merge(medium_data).
+        merge(:links => {:contacts => [contact.id]})]}.to_json)
+  end
+
+  it "returns multiple media" do
+    contact_2 = double(Flapjack::Data::Contact, :id => '53')
+
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :redis => redis, :logger => @logger).and_return(contact)
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact_2.id, :redis => redis, :logger => @logger).and_return(contact_2)
+
+    expect(contact).to receive(:media_list).and_return([medium_data[:type]])
+    expect(contact).to receive(:media).and_return(medium_data[:type] => medium_data[:address])
+    expect(contact).to receive(:media_intervals).and_return(medium_data[:type] => medium_data[:interval])
+    expect(contact).to receive(:media_rollup_thresholds).and_return(medium_data[:type] => medium_data[:rollup_threshold])
+
+    medium_2_data = {:type => 'sms', :address => '0123456789',
+     :interval => 120, :rollup_threshold => 3}
+
+    expect(contact_2).to receive(:media_list).and_return([medium_2_data[:type]])
+    expect(contact_2).to receive(:media).and_return(medium_2_data[:type] => medium_2_data[:address])
+    expect(contact_2).to receive(:media_intervals).and_return(medium_2_data[:type] => medium_2_data[:interval])
+    expect(contact_2).to receive(:media_rollup_thresholds).and_return(medium_2_data[:type] => medium_2_data[:rollup_threshold])
+
+    aget "/media/#{contact.id}_#{medium_data[:type]},#{contact_2.id}_#{medium_2_data[:type]}"
+    expect(last_response).to be_ok
+    expect(last_response.body).to eq({:media => [{:id => "#{contact.id}_#{medium_data[:type]}"}.merge(medium_data).
+        merge(:links => {:contacts => [contact.id]}),
+        {:id => "#{contact_2.id}_#{medium_2_data[:type]}"}.merge(medium_2_data).
+        merge(:links => {:contacts => [contact_2.id]})]}.to_json)
+  end
+
+  it "returns all media" do
+    expect(Flapjack::Data::Contact).to receive(:all).
+      with(:redis => redis).and_return([contact])
+
+    expect(contact).to receive(:media_list).and_return([medium_data[:type]])
+    expect(contact).to receive(:media).and_return(medium_data[:type] => medium_data[:address])
+    expect(contact).to receive(:media_intervals).and_return(medium_data[:type] => medium_data[:interval])
+    expect(contact).to receive(:media_rollup_thresholds).and_return(medium_data[:type] => medium_data[:rollup_threshold])
+
+    aget '/media'
+    expect(last_response).to be_ok
+    expect(last_response.body).to eq({:media => [{:id => "#{contact.id}_#{medium_data[:type]}"}.merge(medium_data).
+        merge(:links => {:contacts => [contact.id]})]}.to_json)
+  end
+
+  it "does not return a medium if the contact is not present" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :redis => redis, :logger => @logger).and_return(nil)
+
+    aget "/media/#{contact.id}_email"
+    expect(last_response.status).to eq(404)
+  end
+
+  it "does not return a medium if the medium is not present" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :redis => redis, :logger => @logger).and_return(contact)
+
+    expect(contact).to receive(:media_list).and_return(['email'])
+
+    aget "/media/#{contact.id}_sms"
+    expect(last_response).to be_ok
+    expect(last_response.body).to eq({:media => []}.to_json)
+  end
+
+  it "creates a medium for a contact" do
+    expect(Flapjack::Data::Semaphore).to receive(:new).
+      with("contact_mass_update", :redis => redis, :expiry => 30).and_return(semaphore)
+
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :redis => redis).and_return(contact)
+
+    expect(contact).to receive(:set_address_for_media).
+      with(medium_data[:type], medium_data[:address])
+    expect(contact).to receive(:set_interval_for_media).
+      with(medium_data[:type], medium_data[:interval])
+    expect(contact).to receive(:set_rollup_threshold_for_media).
+      with(medium_data[:type], medium_data[:rollup_threshold])
+
+    expect(semaphore).to receive(:release).and_return(true)
+
+    apost "/contacts/#{contact.id}/media", {:media => [medium_data]}.to_json, jsonapi_post_env
+    expect(last_response.status).to eq(201)
+    expect(last_response.body).to eq('{"media":[' +
+      medium_data.merge(:id => "#{contact.id}_email",
+                        :links => {:contacts => [contact.id]}).to_json + ']}')
+  end
+
+  it "does not create a medium for a contact that's not present" do
+    expect(Flapjack::Data::Semaphore).to receive(:new).
+      with("contact_mass_update", :redis => redis, :expiry => 30).and_return(semaphore)
+
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :redis => redis).and_return(nil)
+
+    medium_data = {:type => 'email',
+      :address => 'abc@example.com', :interval => 120, :rollup_threshold => 3}
+
+    expect(semaphore).to receive(:release).and_return(true)
+
+    apost "/contacts/#{contact.id}/media", {:media => [medium_data]}.to_json, jsonapi_post_env
+    expect(last_response.status).to eq(422)
+  end
+
+  it "updates a medium" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :logger => @logger, :redis => redis).and_return(contact)
+
+    expect(contact).to receive(:media_list).and_return(['email'])
+    expect(contact).to receive(:set_address_for_media).with('email', 'xyz@example.com')
+
+    apatch "/media/#{contact.id}_email",
+      [{:op => 'replace', :path => '/media/0/address', :value => 'xyz@example.com'}].to_json,
+      jsonapi_patch_env
+    expect(last_response.status).to eq(204)
+  end
+
+  it "updates multiple media" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :logger => @logger, :redis => redis).and_return(contact)
+
+    expect(contact).to receive(:media_list).and_return(['email', 'sms'])
+    expect(contact).to receive(:set_interval_for_media).with('email', 80)
+    expect(contact).to receive(:set_interval_for_media).with('sms', 80)
+
+    apatch "/media/#{contact.id}_email,#{contact.id}_sms",
+      [{:op => 'replace', :path => '/media/0/interval', :value => 80}].to_json,
+      jsonapi_patch_env
+    expect(last_response.status).to eq(204)
+  end
+
+  it "does not update a medium for a contact that's not present" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :logger => @logger, :redis => redis).and_return(nil)
+
+    apatch "/media/#{contact.id}_email",
+      [{:op => 'replace', :path => '/media/0/address', :value => 'xyz@example.com'}].to_json,
+      jsonapi_patch_env
+    expect(last_response.status).to eq(404)
+  end
+
+  it "does not update a medium that's not present" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :logger => @logger, :redis => redis).and_return(contact)
+
+    expect(contact).to receive(:media_list).and_return(['sms'])
+    expect(contact).not_to receive(:set_address_for_media).with('email', 'xyz@example.com')
+
+    apatch "/media/#{contact.id}_email",
+      [{:op => 'replace', :path => '/media/0/address', :value => 'xyz@example.com'}].to_json,
+      jsonapi_patch_env
+    expect(last_response.status).to eq(204)
+  end
+
+  it "deletes a medium" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :logger => @logger, :redis => redis).and_return(contact)
+
+    expect(contact).to receive(:media_list).and_return(['sms', 'email'])
+    expect(contact).to receive(:remove_media).with('email')
+
+    adelete "/media/#{contact.id}_email"
+    expect(last_response.status).to eq(204)
+  end
+
+  it "deletes multiple media" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :logger => @logger, :redis => redis).and_return(contact)
+
+    expect(contact).to receive(:media_list).and_return(['sms', 'email'])
+    expect(contact).to receive(:remove_media).with('email')
+    expect(contact).to receive(:remove_media).with('sms')
+
+    adelete "/media/#{contact.id}_email,#{contact.id}_sms"
+    expect(last_response.status).to eq(204)
+  end
+
+  it "does not delete a medium that's not found" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :logger => @logger, :redis => redis).and_return(contact)
+
+    expect(contact).to receive(:media_list).and_return(['sms'])
+    expect(contact).not_to receive(:remove_media).with('email')
+
+    adelete "/media/#{contact.id}_email"
+    expect(last_response.status).to eq(204)
+  end
+
+  it "does not delete a medium for a contact that's not found" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with('23', :logger => @logger, :redis => redis).and_return(nil)
+
+    adelete '/media/23_email'
+    expect(last_response.status).to eq(404)
+  end
+
+end

data/spec/lib/flapjack/gateways/jsonapi/notification_rule_methods_spec.rb

@@ -0,0 +1,131 @@
+require 'spec_helper'
+require 'flapjack/gateways/jsonapi'
+
+describe 'Flapjack::Gateways::JSONAPI::NotificationRuleMethods', :sinatra => true, :logger => true do
+
+  include_context "jsonapi"
+
+  let(:contact)      { double(Flapjack::Data::Contact, :id => '21') }
+
+  let(:notification_rule) {
+    double(Flapjack::Data::NotificationRule, :id => '1', :contact_id => '21')
+  }
+
+  let(:notification_rule_data) {
+    {"tags"               => ["database","physical"],
+     "regex_tags"         => ["^data.*$","^(physical|bare_metal)$"],
+     "regex_entities"     => ["^foo-\S{3}-\d{2}.example.com$"],
+     "time_restrictions"  => nil,
+     "unknown_media"      => ["jabber"],
+     "warning_media"      => ["email"],
+     "critical_media"     => ["sms", "email"],
+     "unknown_blackhole"  => false,
+     "warning_blackhole"  => false,
+     "critical_blackhole" => false
+    }
+  }
+
+  it "returns a specified notification rule" do
+    expect(notification_rule).to receive(:to_jsonapi).and_return('"rule_1"')
+    expect(Flapjack::Data::NotificationRule).to receive(:find_by_id).
+      with(notification_rule.id, {:redis => redis, :logger => @logger}).and_return(notification_rule)
+
+    aget "/notification_rules/#{notification_rule.id}"
+    expect(last_response).to be_ok
+    expect(last_response.body).to eq('{"notification_rules":["rule_1"]}')
+  end
+
+  it "does not return a notification rule that does not exist" do
+    expect(Flapjack::Data::NotificationRule).to receive(:find_by_id).
+      with(notification_rule.id, {:redis => redis, :logger => @logger}).and_return(nil)
+
+    aget "/notification_rules/#{notification_rule.id}"
+    expect(last_response.status).to eq(404)
+  end
+
+  # POST /notification_rules
+  it "creates a new notification rule" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, {:redis => redis, :logger => @logger}).and_return(contact)
+    expect(notification_rule).to receive(:respond_to?).with(:critical_media).and_return(true)
+    expect(notification_rule).to receive(:to_json).and_return('"rule_1"')
+
+    # symbolize the keys
+    notification_rule_data_sym = notification_rule_data.inject({}){|memo,(k,v)|
+      memo[k.to_sym] = v; memo
+    }
+
+    expect(contact).to receive(:add_notification_rule).
+      with(notification_rule_data_sym, :logger => @logger).and_return(notification_rule)
+
+    apost "/contacts/#{contact.id}/notification_rules",
+      {"notification_rules" => [notification_rule_data]}.to_json, jsonapi_post_env
+    expect(last_response.status).to eq(201)
+    expect(last_response.body).to eq('{"notification_rules":["rule_1"]}')
+  end
+
+  it "does not create a notification_rule for a contact that's not present" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, {:redis => redis, :logger => @logger}).and_return(nil)
+
+    apost "/contacts/#{contact.id}/notification_rules",
+      {"notification_rules" => [notification_rule_data]}.to_json, jsonapi_post_env
+    expect(last_response.status).to eq(404)
+  end
+
+  # PATCH /notification_rules/RULE_ID
+  it "updates a notification rule" do
+    expect(Flapjack::Data::NotificationRule).to receive(:find_by_id).
+      with(notification_rule.id, {:redis => redis, :logger => @logger}).and_return(notification_rule)
+
+    expect(notification_rule).to receive(:update).with({:warning_blackhole => true}, :logger => @logger).and_return(nil)
+
+    apatch "/notification_rules/#{notification_rule.id}",
+      [{:op => 'replace', :path => '/notification_rules/0/warning_blackhole', :value => true}].to_json,
+      jsonapi_patch_env
+    expect(last_response.status).to eq(204)
+  end
+
+  it "does not update a notification rule that's not present" do
+    expect(Flapjack::Data::NotificationRule).to receive(:find_by_id).
+      with(notification_rule.id, {:redis => redis, :logger => @logger}).and_return(nil)
+
+    apatch "/notification_rules/#{notification_rule.id}",
+      [{:op => 'replace', :path => '/notification_rules/0/warning_blackhole', :value => true}].to_json,
+      jsonapi_patch_env
+    expect(last_response.status).to eq(404)
+  end
+
+  # DELETE /notification_rules/RULE_ID
+  it "deletes a notification rule" do
+    expect(notification_rule).to receive(:contact_id).and_return(contact.id)
+    expect(Flapjack::Data::NotificationRule).to receive(:find_by_id).
+      with(notification_rule.id, {:redis => redis, :logger => @logger}).and_return(notification_rule)
+    expect(contact).to receive(:delete_notification_rule).with(notification_rule)
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, {:redis => redis, :logger => @logger}).and_return(contact)
+
+    adelete "/notification_rules/#{notification_rule.id}"
+    expect(last_response.status).to eq(204)
+  end
+
+  it "does not delete a notification rule that's not present" do
+    expect(Flapjack::Data::NotificationRule).to receive(:find_by_id).
+      with(notification_rule.id, {:redis => redis, :logger => @logger}).and_return(nil)
+
+    adelete "/notification_rules/#{notification_rule.id}"
+    expect(last_response.status).to eq(404)
+  end
+
+  it "does not delete a notification rule if the contact is not present" do
+    expect(notification_rule).to receive(:contact_id).and_return(contact.id)
+    expect(Flapjack::Data::NotificationRule).to receive(:find_by_id).
+      with(notification_rule.id, {:redis => redis, :logger => @logger}).and_return(notification_rule)
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, {:redis => redis, :logger => @logger}).and_return(nil)
+
+    adelete "/notification_rules/#{notification_rule.id}"
+    expect(last_response.status).to eq(404)
+  end
+
+end

data/spec/lib/flapjack/gateways/jsonapi/pagerduty_credential_methods_spec.rb

@@ -0,0 +1,113 @@
+require 'spec_helper'
+require 'flapjack/gateways/jsonapi'
+
+describe 'Flapjack::Gateways::JSONAPI::PagerdutyCredentialMethods', :sinatra => true, :logger => true do
+
+  include_context "jsonapi"
+
+  let(:contact) { double(Flapjack::Data::Contact, :id => '21') }
+
+  let(:pagerduty_credentials) {
+    {'service_key' => 'abc',
+     'subdomain'   => 'def',
+     'username'    => 'ghi',
+     'password'    => 'jkl',
+    }
+  }
+
+  let(:semaphore) {
+    double(Flapjack::Data::Semaphore, :resource => 'folly',
+           :key => 'semaphores:folly', :expiry => 30, :token => 'spatulas-R-us')
+  }
+
+  it "returns pagerduty credentials" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :redis => redis, :logger => @logger).and_return(contact)
+
+    expect(contact).to receive(:pagerduty_credentials).and_return(pagerduty_credentials)
+
+    aget "/pagerduty_credentials/#{contact.id}"
+    expect(last_response).to be_ok
+    expect(last_response.body).to eq({:pagerduty_credentials => [pagerduty_credentials.
+        merge(:links => {:contacts => [contact.id]})]}.to_json)
+  end
+
+  it "does not return pagerduty credentials if the contact is not present" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :redis => redis, :logger => @logger).and_return(nil)
+
+    aget "/pagerduty_credentials/#{contact.id}"
+    expect(last_response.status).to eq(404)
+  end
+
+  it "creates pagerduty credentials for a contact" do
+    expect(Flapjack::Data::Semaphore).to receive(:new).
+      with("contact_mass_update", :redis => redis, :expiry => 30).and_return(semaphore)
+
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :redis => redis).and_return(contact)
+
+    expect(contact).to receive(:set_pagerduty_credentials).with(pagerduty_credentials)
+    expect(semaphore).to receive(:release).and_return(true)
+
+    apost "/contacts/#{contact.id}/pagerduty_credentials",
+      {:pagerduty_credentials => [pagerduty_credentials]}.to_json, jsonapi_post_env
+    expect(last_response.status).to eq(201)
+    expect(last_response.body).to eq('{"pagerduty_credentials":[' +
+      pagerduty_credentials.merge(:links => {:contacts => [contact.id]}).to_json + ']}')
+  end
+
+  it "does not create pagerduty credentials for a contact that's not present" do
+    expect(Flapjack::Data::Semaphore).to receive(:new).
+      with("contact_mass_update", :redis => redis, :expiry => 30).and_return(semaphore)
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :redis => redis).and_return(nil)
+    expect(semaphore).to receive(:release).and_return(true)
+
+    apost "/contacts/#{contact.id}/pagerduty_credentials",
+      {:pagerduty_credentials => [pagerduty_credentials]}.to_json, jsonapi_post_env
+    expect(last_response.status).to eq(422)
+  end
+
+  it "updates pagerduty credentials for a contact" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :logger => @logger, :redis => redis).and_return(contact)
+
+    expect(contact).to receive(:pagerduty_credentials).and_return(pagerduty_credentials)
+    expect(contact).to receive(:set_pagerduty_credentials).with(pagerduty_credentials.merge('service_key' => 'xyz'))
+
+    apatch "/pagerduty_credentials/#{contact.id}",
+      [{:op => 'replace', :path => '/pagerduty_credentials/0/service_key', :value => 'xyz'}].to_json,
+      jsonapi_patch_env
+    expect(last_response.status).to eq(204)
+  end
+
+  it "does not update pagerduty credentials for a contact that's not present" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :logger => @logger, :redis => redis).and_return(nil)
+
+    apatch "/pagerduty_credentials/#{contact.id}",
+      [{:op => 'replace', :path => '/pagerduty_credentials/0/service_key', :value => 'xyz'}].to_json,
+      jsonapi_patch_env
+    expect(last_response.status).to eq(404)
+  end
+
+  it "deletes the pagerduty credentials for a contact" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :redis => redis, :logger => @logger).and_return(contact)
+
+    expect(contact).to receive(:delete_pagerduty_credentials)
+
+    adelete "/pagerduty_credentials/#{contact.id}"
+    expect(last_response.status).to eq(204)
+  end
+
+  it "does not delete the pagerduty credentials of a contact that's not present" do
+    expect(Flapjack::Data::Contact).to receive(:find_by_id).
+      with(contact.id, :redis => redis, :logger => @logger).and_return(nil)
+
+    adelete "/pagerduty_credentials/#{contact.id}"
+    expect(last_response.status).to eq(404)
+  end
+
+end