flapjack-diner 0.12

data/lib/flapjack-diner/argument_validator.rb

@@ -0,0 +1,60 @@
+module Flapjack
+  class ArgumentValidator
+
+    attr_reader :query
+
+    def initialize(query = {})
+      @errors = []
+      @query = query
+    end
+
+    def validate(args)
+      args = args.dup
+      validations = args.delete(:as)
+      validations = [validations] unless validations.is_a?(Array)
+
+      if elements = args[:query]
+        elements = [elements] unless elements.is_a?(Array)
+        validations.each do |validation|
+          __send__(validation.to_s.downcase, *elements)
+        end
+      end
+
+      raise ArgumentError.new(@errors.join('; ')) unless @errors.empty?
+    end
+
+    private
+
+    def time(*elements)
+      elements.each do |element|
+        if target = @query[element]
+          @errors << "'#{target}' should contain some kind of time object which responds to." unless target.respond_to?(:iso8601)
+        end
+      end
+    end
+
+    def required(*elements)
+      elements.each do |element|
+        @errors << "'#{element}' is required." if @query[element].nil?
+      end
+    end
+
+    def respond_to?(name, include_private = false)
+      !classify_name(name).nil? || super
+    end
+
+    def method_missing(name, *args)
+      return super unless klass = classify_name(name)
+      elements = args
+      elements.each do |element|
+        @errors << "'#{element}' is expected to be a #{klass}" unless @query[element].is_a?(klass)
+      end
+    end
+
+    def classify_name(name)
+      class_name = name.to_s.split('_').map(&:capitalize).join
+      Module.const_get(class_name)
+    rescue NameError
+    end
+  end
+end

data/lib/flapjack-diner/version.rb

@@ -0,0 +1,5 @@
+module Flapjack
+  module Diner
+    VERSION = "0.12"
+  end
+end

data/spec/argument_validator_spec.rb

@@ -0,0 +1,101 @@
+require 'spec_helper'
+require "flapjack-diner/argument_validator"
+
+describe Flapjack::ArgumentValidator do
+
+  context 'required' do
+
+    let(:query) do
+      {:entity => 'myservice', :check => 'HOST'}
+    end
+
+    subject { Flapjack::ArgumentValidator.new(query) }
+
+    it 'does not raise an exception when query entity is valid' do
+      lambda { subject.validate(:query => :entity, :as => :required) }.should_not raise_exception(ArgumentError)
+    end
+
+    it 'raises ArgumentError when query entity is invalid' do
+      query[:entity] = nil
+      lambda { subject.validate(:query => :entity, :as => :required) }.should raise_exception(ArgumentError)
+    end
+
+    it 'handles arrays as query values valid' do
+      lambda { subject.validate(:query => [:entity, :check], :as => :required) }.should_not raise_exception(ArgumentError)
+    end
+
+    it 'handles arrays as query values invalid' do
+      query[:check] = nil
+      lambda { subject.validate(:query => [:entity, :check], :as => :required) }.should raise_exception(ArgumentError)
+    end
+  end
+
+  context 'time' do
+
+    let(:query) do
+      {:start_time => Time.now, :duration => 10}
+    end
+
+    subject { Flapjack::ArgumentValidator.new(query) }
+
+    it 'does not raise an exception when query start_time is valid' do
+      lambda { subject.validate(:query => :start_time, :as => :time) }.should_not raise_exception(ArgumentError)
+    end
+
+    it 'raises an exception when query start_time is invalid' do
+      query[:start_time] = 1234
+      lambda { subject.validate(:query => :start_time, :as => :time) }.should raise_exception(ArgumentError)
+    end
+
+    it 'handles arrays as query values valid' do
+      query[:end_time] = Time.now
+      lambda { subject.validate(:query => [:start_time, :end_time], :as => :time) }.should_not raise_exception(ArgumentError)
+    end
+
+    it 'handles arrays as query values invalid' do
+      query[:end_time] = 3904
+      lambda { subject.validate(:query => [:start_time, :end_time], :as => :time) }.should raise_exception(ArgumentError)
+    end
+
+    it 'handles dates as query values' do
+      query[:end_time] = Date.today
+      lambda { subject.validate(:query => :end_time, :as => :time) }.should_not raise_exception(ArgumentError)
+    end
+  end
+
+  context 'integer via method missing' do
+
+    let(:query) do
+      {:start_time => Time.now, :duration => 10}
+    end
+
+    subject { Flapjack::ArgumentValidator.new(query) }
+
+    it 'does not raise an exception when query duration is valid' do
+      lambda { subject.validate(:query => :duration, :as => :integer) }.should_not raise_exception(ArgumentError)
+    end
+
+    it 'raises an exception when query duration is invalid' do
+      query[:duration] = '23'
+      lambda { subject.validate(:query => :duration, :as => :integer) }.should raise_exception(ArgumentError)
+    end
+  end
+
+  context 'multiple validations' do
+
+    let(:query) do
+      {:start_time => Time.now, :duration => 10}
+    end
+
+    subject { Flapjack::ArgumentValidator.new(query) }
+
+    it 'does not raise an exception when query start_time is valid' do
+      lambda { subject.validate(:query => :start_time, :as => [:time, :required]) }.should_not raise_exception(ArgumentError)
+    end
+
+    it 'raises an exception when query start_time is invalid' do
+      query[:start_time] = nil
+      lambda { subject.validate(:query => :start_time, :as => [:time, :required]) }.should raise_exception(ArgumentError)
+    end
+  end
+end

data/spec/flapjack-diner_spec.rb

@@ -0,0 +1,877 @@
+require 'spec_helper'
+require 'flapjack-diner'
+
+describe Flapjack::Diner do
+
+  let(:server) { 'flapjack.com' }
+  let(:entity) { 'ex-abcd-data-17.example.com' }
+  let(:check)  { 'ping'}
+
+  let(:rule_data) {
+    {"contact_id"         => "21",
+     "entity_tags"        => ["database","physical"],
+     "entities"           => ["foo-app-01.example.com"],
+     "time_restrictions"  => nil,
+     "warning_media"      => ["email"],
+     "critical_media"     => ["sms", "email"],
+     "warning_blackhole"  => false,
+     "critical_blackhole" => false
+    }
+  }
+
+  let(:response)      { '{"key":"value"}' }
+  let(:response_body) { {'key' => 'value'} }
+
+  before(:each) do
+    Flapjack::Diner.base_uri(server)
+    Flapjack::Diner.logger = nil
+  end
+
+  after(:each) do
+    WebMock.reset!
+  end
+
+  it "returns a json list of entities" do
+    req = stub_request(:get, "http://#{server}/entities").to_return(
+      :body => response)
+
+    result = Flapjack::Diner.entities
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a json list of entities from a non-standard port" do
+    Flapjack::Diner.base_uri('flapjack.com:54321')
+
+    req = stub_request(:get, "http://#{server}:54321/entities").to_return(
+      :body => response)
+
+    result = Flapjack::Diner.entities
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a json list of checks for an entity" do
+    req = stub_request(:get, "http://#{server}/checks/#{entity}").to_return(
+      :body => response)
+
+    result = Flapjack::Diner.checks(entity)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a json list of check statuses for an entity" do
+    req = stub_request(:get, "http://#{server}/status").
+            with(:query => {:entity => entity}).
+            to_return(:body => response)
+
+    result = Flapjack::Diner.status(entity)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of entity statuses for a bulk query" do
+    req = stub_request(:get, "http://#{server}/status").
+      with(:query => {:entity => ['abc.net', 'def.com'], :check => {'ghi.net' => 'ping', 'jkl.org' => ['ping', 'ssh']}}).
+      to_return(:body => response)
+
+    result = Flapjack::Diner.bulk_status(:entity => ['abc.net', 'def.com'], :check => {'ghi.net' => 'ping', 'jkl.org' => ['ping', 'ssh']})
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a single check status for an entity" do
+    req = stub_request(:get, "http://#{server}/status").
+            with(:query => {:check => {entity => check}}).
+            to_return(:body => response)
+
+    result = Flapjack::Diner.status(entity, :check => check)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of scheduled maintenance periods for all checks on an entity" do
+    req = stub_request(:get, "http://#{server}/scheduled_maintenances").
+            with(:query => {:entity => entity}).
+            to_return(:body => response)
+
+    result = Flapjack::Diner.scheduled_maintenances(entity)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of scheduled maintenance periods for a check on an entity" do
+    req = stub_request(:get, "http://#{server}/scheduled_maintenances").
+            with(:query => {:check => {entity => check}}).
+            to_return(:body => response)
+
+    result = Flapjack::Diner.scheduled_maintenances(entity, :check => check)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of scheduled maintenance periods for a bulk query" do
+    req = stub_request(:get, "http://#{server}/scheduled_maintenances").
+      with(:query => {:entity => 'abc.net', :check => {'def.org' => 'ping'}}).
+      to_return(:body => response)
+
+    result = Flapjack::Diner.bulk_scheduled_maintenances(:entity => 'abc.net', :check => {'def.org' => 'ping'})
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of unscheduled maintenance periods for all checks on an entity" do
+    req = stub_request(:get, "http://#{server}/unscheduled_maintenances").
+            with(:query => {:entity => entity}).
+            to_return(:body => response)
+
+    result = Flapjack::Diner.unscheduled_maintenances(entity)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of unscheduled maintenance periods for a check on an entity" do
+    req = stub_request(:get, "http://#{server}/unscheduled_maintenances").
+            with(:query => {:check => {entity => check}}).
+            to_return(:body => response)
+
+    result = Flapjack::Diner.unscheduled_maintenances(entity, :check => check)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of unscheduled maintenance periods for a bulk query" do
+    req = stub_request(:get, "http://#{server}/unscheduled_maintenances").
+      with(:query => {:entity => 'abc.net', :check => {'def.org' => 'ping'}}).
+      to_return(:body => response)
+
+    result = Flapjack::Diner.bulk_unscheduled_maintenances(:entity => 'abc.net', :check => {'def.org' => 'ping'})
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of outages for all checks on an entity" do
+    req = stub_request(:get, "http://#{server}/outages").
+            with(:query => {:entity => entity}).
+            to_return(:body => response)
+
+    result = Flapjack::Diner.outages(entity)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of outages for a check on an entity" do
+    req = stub_request(:get, "http://#{server}/outages").
+            with(:query => {:check => {entity => check}}).
+            to_return(:body => response)
+
+    result = Flapjack::Diner.outages(entity, :check => check)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of outages for a bulk query" do
+    req = stub_request(:get, "http://#{server}/outages").
+      with(:query => {:entity => 'abc.net', :check => {'def.org' => 'ping'}}).
+      to_return(:body => response)
+
+    result = Flapjack::Diner.bulk_outages(:entity => 'abc.net', :check => {'def.org' => 'ping'})
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of downtimes for all checks on an entity" do
+    req = stub_request(:get, "http://#{server}/downtime").
+            with(:query => {:entity => entity}).
+            to_return(:body => response)
+
+    result = Flapjack::Diner.downtime(entity)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of downtimes for all checks on an entity between two times" do
+    start  = Time.iso8601('2011-08-01T00:00:00+10:00')
+    finish = Time.iso8601('2011-08-31T00:00:00+10:00')
+
+    req = stub_request(:get, "http://#{server}/downtime").
+            with(:query => {:entity => entity, :start_time => start.iso8601, :end_time => finish.iso8601}).
+            to_return(:body => response)
+
+    result = Flapjack::Diner.downtime(entity, :start_time => start, :end_time => finish)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of downtimes for a check on an entity" do
+    req = stub_request(:get, "http://#{server}/downtime").
+            with(:query => {:check => {entity => check}}).
+            to_return(:body => response)
+
+    result = Flapjack::Diner.downtime(entity, :check => check)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a list of downtimes for a bulk query between two times" do
+    start  = Time.iso8601('2011-08-01T00:00:00+10:00')
+    finish = Time.iso8601('2011-08-31T00:00:00+10:00')
+
+    req = stub_request(:get, "http://#{server}/downtime").
+      with(:query => {:entity => 'abc.net', :check => {'def.org' => 'ping'},
+                      :start_time => start.iso8601, :end_time => finish.iso8601}).
+      to_return(:body => response)
+
+    result = Flapjack::Diner.bulk_downtime(:entity => 'abc.net', :check => {'def.org' => 'ping'},
+                                           :start_time => start, :end_time => finish)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "acknowledges a check's state for an entity" do
+    req = stub_request(:post, "http://#{server}/acknowledgements").with(
+      :body => {:check => {entity => check}, :summary => 'dealing with it'}).to_return(
+      :status => 204)
+
+    result = Flapjack::Diner.acknowledge!(entity, check, :summary => 'dealing with it')
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "acknowledges all checks on an entity" do
+    req = stub_request(:post, "http://#{server}/acknowledgements").with(
+      :body => {:entity => entity, :summary => 'dealing with it'},
+                :headers => {'Content-Type' => 'application/json'}).to_return(
+      :status => 204)
+
+    result = Flapjack::Diner.bulk_acknowledge!(:entity => entity, :summary => 'dealing with it')
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "acknowledges checks from multiple entities" do
+    req = stub_request(:post, "http://#{server}/acknowledgements").with(
+      :body => {:entity => [entity, 'lmn.net'], :summary => 'dealing with it'}).to_return(
+      :status => 204)
+
+    result = Flapjack::Diner.bulk_acknowledge!(:entity => [entity, 'lmn.net'], :summary => 'dealing with it')
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "generates test notifications for a check on an entity" do
+    req = stub_request(:post, "http://#{server}/test_notifications").with(
+      :body => {:check => {entity => check}, :summary => 'testing notifications'}).to_return(
+      :status => 204)
+
+    result = Flapjack::Diner.test_notifications!(entity, check, :summary => 'testing notifications')
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "generates test notifications for all checks on an entity" do
+    req = stub_request(:post, "http://#{server}/test_notifications").with(
+      :body => {:entity => entity, :summary => 'testing notifications'}).to_return(
+      :status => 204)
+
+    result = Flapjack::Diner.bulk_test_notifications!(:entity => entity, :summary => 'testing notifications')
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "generates test notifications for checks from multiple entities" do
+    req = stub_request(:post, "http://#{server}/test_notifications").with(
+      :body => {:entity => [entity, 'lmn.net'], :summary => 'testing notifications'}).to_return(
+      :status => 204)
+
+    result = Flapjack::Diner.bulk_test_notifications!(:entity => [entity, 'lmn.net'], :summary => 'testing notifications')
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "creates a scheduled maintenance period for an entity" do
+    start_time = Time.now
+    duration = 60 * 30 # in seconds, so 30 minutes
+    summary = "fixing everything"
+
+    req = stub_request(:post, "http://#{server}/scheduled_maintenances").
+            with(:body => {:check => {entity => check}, :start_time => start_time.iso8601,
+                           :duration => duration, :summary => summary},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:status => 204)
+
+    result = Flapjack::Diner.create_scheduled_maintenance!(entity, check,
+      :start_time => start_time, :duration => duration, :summary => summary)
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "creates scheduled maintenance periods for all checks on an entity" do
+    start_time = Time.now
+    duration = 60 * 30 # in seconds, so 30 minutes
+    summary = "fixing everything"
+
+    req = stub_request(:post, "http://#{server}/scheduled_maintenances").
+            with(:body => {:entity => entity, :start_time => start_time.iso8601,
+                           :duration => duration, :summary => summary},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:status => 204)
+
+    result = Flapjack::Diner.bulk_create_scheduled_maintenance!(:entity => entity,
+      :start_time => start_time, :duration => duration, :summary => summary)
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "creates scheduled maintenance periods for checks from multiple entities" do
+    start_time = Time.now
+    duration = 60 * 30 # in seconds, so 30 minutes
+    summary = "fixing everything"
+
+    req = stub_request(:post, "http://#{server}/scheduled_maintenances").
+            with(:body => {:check => {entity => 'ping', 'pqr.org' => 'ssh'}, :start_time => start_time.iso8601,
+                           :duration => duration, :summary => summary},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:status => 204)
+
+    result = Flapjack::Diner.bulk_create_scheduled_maintenance!(:check => {entity => 'ping', 'pqr.org' => 'ssh'},
+      :start_time => start_time, :duration => duration, :summary => summary)
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "deletes a scheduled maintenance period for a check on an entity" do
+    start_time = Time.now
+
+    req = stub_request(:delete, "http://#{server}/scheduled_maintenances").
+            with(:body => {:check => {entity => check}, :start_time => start_time.iso8601},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:status => 204)
+
+    result = Flapjack::Diner.delete_scheduled_maintenance!(entity, check,
+      :start_time => start_time)
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "deletes scheduled maintenance periods for all checks on an entity" do
+    start_time = Time.now
+
+    req = stub_request(:delete, "http://#{server}/scheduled_maintenances").
+            with(:body => {:entity => entity, :start_time => start_time.iso8601},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:status => 204)
+
+    result = Flapjack::Diner.bulk_delete_scheduled_maintenance!(:entity => entity,
+      :start_time => start_time)
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "deletes scheduled maintenance periods for checks from multiple entities" do
+    start_time = Time.now
+
+    req = stub_request(:delete, "http://#{server}/scheduled_maintenances").
+            with(:body => {:check => {entity => 'ping', 'pqr.org' => 'ssh'},
+                           :start_time => start_time.iso8601},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:status => 204)
+
+    result = Flapjack::Diner.bulk_delete_scheduled_maintenance!(:check => {entity => 'ping', 'pqr.org' => 'ssh'},
+      :start_time => start_time)
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "deletes an unscheduled maintenance period for a check on an entity" do
+    end_time = Time.now
+
+    req = stub_request(:delete, "http://#{server}/unscheduled_maintenances").
+            with(:body => {:check => {entity => check}, :end_time => end_time.iso8601},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:status => 204)
+
+    result = Flapjack::Diner.delete_unscheduled_maintenance!(entity, check,
+      :end_time => end_time)
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "deletes scheduled maintenance periods for all checks on an entity" do
+    end_time = Time.now
+
+    req = stub_request(:delete, "http://#{server}/unscheduled_maintenances").
+            with(:body => {:entity => entity, :end_time => end_time.iso8601},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:status => 204)
+
+    result = Flapjack::Diner.bulk_delete_unscheduled_maintenance!(:entity => entity,
+      :end_time => end_time)
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "deletes scheduled maintenance periods for checks from multiple entities" do
+    end_time = Time.now
+
+    req = stub_request(:delete, "http://#{server}/unscheduled_maintenances").
+            with(:body => {:check => {entity => 'ping', 'pqr.org' => 'ssh'},
+                           :end_time => end_time.iso8601},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:status => 204)
+
+    result = Flapjack::Diner.bulk_delete_unscheduled_maintenance!(:check => {entity => 'ping', 'pqr.org' => 'ssh'},
+      :end_time => end_time)
+    req.should have_been_requested
+    result.should be_true
+  end
+
+
+  it "returns a list of contacts" do
+    req = stub_request(:get, "http://#{server}/contacts").to_return(
+      :body => response)
+
+    result = Flapjack::Diner.contacts
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a single contact" do
+    contact_id = '21'
+    req = stub_request(:get, "http://#{server}/contacts/#{contact_id}").to_return(
+      :body => response)
+
+    result = Flapjack::Diner.contact(contact_id)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns notification rules for a contact" do
+    contact_id = '21'
+    req = stub_request(:get, "http://#{server}/contacts/#{contact_id}/notification_rules").to_return(
+      :body => response)
+
+    result = Flapjack::Diner.notification_rules(contact_id)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns a single notification rule" do
+    rule_id = '00001'
+    req = stub_request(:get, "http://#{server}/notification_rules/#{rule_id}").
+            to_return(:body => response)
+
+    result = Flapjack::Diner.notification_rule(rule_id)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "creates a notification rule" do
+    rule_result = rule_data.merge('id' => '00001')
+
+    req = stub_request(:post, "http://#{server}/notification_rules").
+            with(:body => rule_data,
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:body => rule_result.to_json)
+
+    result = Flapjack::Diner.create_notification_rule!(rule_data)
+    req.should have_been_requested
+    result.should == rule_result
+  end
+
+  it "updates a notification rule" do
+    rule_id = '00001'
+
+    rule_data_with_id = rule_data.merge('id' => rule_id)
+
+    req = stub_request(:put, "http://#{server}/notification_rules/#{rule_id}").
+            with(:body => rule_data_with_id,
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:body => rule_data_with_id.to_json)
+
+    result = Flapjack::Diner.update_notification_rule!(rule_id, rule_data_with_id)
+    req.should have_been_requested
+    result.should == rule_data_with_id
+  end
+
+  it "deletes a notification rule" do
+    rule_id = '00001'
+    req = stub_request(:delete, "http://#{server}/notification_rules/#{rule_id}").to_return(
+      :status => 204)
+
+    result = Flapjack::Diner.delete_notification_rule!(rule_id)
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "gets a list of entity tags" do
+    req = stub_request(:get, "http://#{server}/entities/#{entity}/tags").to_return(
+      :body => ['web', 'app'].to_json)
+
+    result = Flapjack::Diner.entity_tags(entity)
+    req.should have_been_requested
+    result.should == ['web', 'app']
+  end
+
+  it "adds tags to an entity" do
+    req = stub_request(:post, "http://#{server}/entities/#{entity}/tags").
+            with(:body => {:tag => ['web', 'app']},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:body => ['web', 'app'].to_json)
+
+    result = Flapjack::Diner.add_entity_tags!(entity, 'web', 'app')
+    req.should have_been_requested
+    result.should == ['web', 'app']
+  end
+
+  it "deletes tags from an entity" do
+    req = stub_request(:delete, "http://#{server}/entities/#{entity}/tags").
+            with(:body => {:tag => ['web']},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:status => 204)
+
+    result = Flapjack::Diner.delete_entity_tags!(entity, 'web')
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "gets a list of contact tags" do
+    contact_id = 21
+    req = stub_request(:get, "http://#{server}/contacts/#{contact_id}/tags").to_return(
+      :body => ['user', 'admin'].to_json)
+
+    result = Flapjack::Diner.contact_tags(contact_id)
+    req.should have_been_requested
+    result.should == ['user', 'admin']
+  end
+
+  it "gets tags for a contact's linked entities" do
+    contact_id = 21
+    req = stub_request(:get, "http://#{server}/contacts/#{contact_id}/entity_tags").to_return(
+      :body => {'entity_1' => ['web', 'app']}.to_json)
+
+    result = Flapjack::Diner.contact_entitytags(contact_id)
+    req.should have_been_requested
+    result.should == {'entity_1' => ['web', 'app']}
+  end
+
+  it "adds tags to a contact's linked entities" do
+    contact_id = 21
+    req = stub_request(:post, "http://#{server}/contacts/#{contact_id}/entity_tags").
+            with(:body => {:entity => {'entity_1' => ['web', 'app']}},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:body => {'entity_1' => ['web', 'app']}.to_json)
+
+    result = Flapjack::Diner.add_contact_entitytags!(contact_id, {'entity_1' => ['web', 'app']})
+    req.should have_been_requested
+    result.should == {'entity_1' => ['web', 'app']}
+  end
+
+  it "deletes tags from a contact's linked entities" do
+    contact_id = 21
+    req = stub_request(:delete, "http://#{server}/contacts/#{contact_id}/entity_tags").
+            with(:body => {:entity => {'entity_1' => ['web', 'app']}},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:status => 204)
+
+    result = Flapjack::Diner.delete_contact_entitytags!(contact_id, {'entity_1' => ['web', 'app']})
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "adds tags to a contact" do
+    contact_id = 21
+    req = stub_request(:post, "http://#{server}/contacts/#{contact_id}/tags").
+            with(:body => {:tag => ['admin', 'user']},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:body => ['admin', 'user'].to_json)
+
+    result = Flapjack::Diner.add_contact_tags!(contact_id, 'admin', 'user')
+    req.should have_been_requested
+    result.should == ['admin', 'user']
+  end
+
+  it "deletes tags from a contact" do
+    contact_id = 21
+    req = stub_request(:delete, "http://#{server}/contacts/#{contact_id}/tags").
+            with(:body => {:tag => ['admin']},
+                 :headers => {'Content-Type'=>'application/json'}).
+            to_return(:status => 204)
+
+    result = Flapjack::Diner.delete_contact_tags!(contact_id, 'admin')
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "returns a list of a contact's notification media values" do
+    contact_id = '21'
+    req = stub_request(:get, "http://#{server}/contacts/#{contact_id}/media").to_return(
+      :body => response)
+
+    result = Flapjack::Diner.contact_media(contact_id)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "returns the values for a contact's notification medium" do
+    contact_id = '21'
+    media = 'sms'
+    req = stub_request(:get, "http://#{server}/contacts/#{contact_id}/media/#{media}").to_return(
+      :body => response)
+
+    result = Flapjack::Diner.contact_medium(contact_id, media)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "updates a contact's notification medium" do
+    contact_id = '21'
+    media_type = 'sms'
+    media_data = {"address" => "dmitri@example.com",
+                  "interval" => 900}
+
+    req = stub_request(:put, "http://#{server}/contacts/#{contact_id}/media/#{media_type}").with(
+      :body => media_data).to_return(:body => media_data.to_json)
+
+    result = Flapjack::Diner.update_contact_medium!(contact_id, media_type, media_data)
+    req.should have_been_requested
+    result.should == media_data
+  end
+
+  it "deletes a contact's notification medium" do
+    contact_id = '21'
+    media = 'sms'
+    req = stub_request(:delete, "http://#{server}/contacts/#{contact_id}/media/#{media}").to_return(
+      :status => 204)
+
+    result = Flapjack::Diner.delete_contact_medium!(contact_id, media)
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "returns a contact's timezone" do
+    contact_id = '21'
+    req = stub_request(:get, "http://#{server}/contacts/#{contact_id}/timezone").to_return(
+      :body => response)
+
+    result = Flapjack::Diner.contact_timezone(contact_id)
+    req.should have_been_requested
+    result.should_not be_nil
+    result.should == response_body
+  end
+
+  it "updates a contact's timezone" do
+    contact_id = '21'
+    timezone_data = {'timezone' => "Australia/Perth"}
+
+    req = stub_request(:put, "http://#{server}/contacts/#{contact_id}/timezone").with(
+      :body => timezone_data).to_return(:body => timezone_data.to_json)
+
+    result = Flapjack::Diner.update_contact_timezone!(contact_id, timezone_data['timezone'])
+    req.should have_been_requested
+    result.should == timezone_data
+  end
+
+  it "deletes a contact's timezone" do
+    contact_id = '21'
+    req = stub_request(:delete, "http://#{server}/contacts/#{contact_id}/timezone").to_return(
+      :status => 204)
+
+    result = Flapjack::Diner.delete_contact_timezone!(contact_id)
+    req.should have_been_requested
+    result.should be_true
+  end
+
+  it "returns nil with last_error available when requesting the timezone for a non existant contact" do
+    contact_id = 'jkfldsj'
+    req = stub_request(:get, "http://#{server}/contacts/#{contact_id}/timezone").to_return(
+      :body => '{"errors": ["Not found"]}', :status => 404)
+
+    result = Flapjack::Diner.contact_timezone(contact_id)
+    last_error = Flapjack::Diner.last_error
+    req.should have_been_requested
+    result.should be_nil
+    last_error.should_not be_nil
+  end
+
+  context "logging" do
+
+    let(:logger) { mock('logger') }
+
+    before do
+      Flapjack::Diner.logger = logger
+    end
+
+    it "logs a GET request without a path" do
+      req = stub_request(:get, "http://#{server}/entities").to_return(
+        :body => response)
+
+      logger.should_receive(:info).with("GET http://#{server}/entities")
+      logger.should_receive(:info).with("  Response Code: 200")
+      logger.should_receive(:info).with("  Response Body: #{response}")
+
+      result = Flapjack::Diner.entities
+      req.should have_been_requested
+      result.should_not be_nil
+      result.should == response_body
+    end
+
+    it "logs a GET request with a path" do
+      req = stub_request(:get, "http://#{server}/checks/#{entity}").to_return(
+        :body => response)
+
+      logger.should_receive(:info).with("GET http://#{server}/checks/#{entity}")
+      logger.should_receive(:info).with("  Response Code: 200")
+      logger.should_receive(:info).with("  Response Body: #{response}")
+
+      result = Flapjack::Diner.checks(entity)
+      req.should have_been_requested
+      result.should_not be_nil
+      result.should == response_body
+    end
+
+    it "logs a POST request" do
+      req = stub_request(:post, "http://#{server}/acknowledgements").
+              with(:body => {:check => {entity => check}, :summary => 'dealing with it'},
+                   :headers => {'Content-Type'=>'application/json'}).
+              to_return(:status => 204)
+      logger.should_receive(:info).with("POST http://#{server}/acknowledgements\n" +
+        "  Params: {:summary=>\"dealing with it\", :check=>{\"ex-abcd-data-17.example.com\"=>\"ping\"}}")
+      logger.should_receive(:info).with("  Response Code: 204")
+
+      result = Flapjack::Diner.acknowledge!(entity, check, :summary => 'dealing with it')
+      req.should have_been_requested
+      result.should be_true
+    end
+
+    it "logs a JSON put request" do
+      contact_id = '21'
+      timezone_data = {:timezone => "Australia/Perth"}
+
+      req = stub_request(:put, "http://#{server}/contacts/#{contact_id}/timezone").with(
+        :body => timezone_data).to_return(:body => timezone_data.to_json, :status => [200, ' OK'])
+
+      logger.should_receive(:info).
+        with("PUT http://#{server}/contacts/#{contact_id}/timezone\n  Params: #{timezone_data.inspect}")
+      logger.should_receive(:info).with("  Response Code: 200 OK")
+      logger.should_receive(:info).with("  Response Body: #{timezone_data.to_json}")
+
+      result = Flapjack::Diner.update_contact_timezone!(contact_id, timezone_data[:timezone])
+      req.should have_been_requested
+      result.should == {'timezone' => "Australia/Perth"}
+    end
+
+    it "logs a DELETE request" do
+      contact_id = '21'
+      req = stub_request(:delete, "http://#{server}/contacts/#{contact_id}/timezone").to_return(
+        :status => 204)
+
+      logger.should_receive(:info).with("DELETE http://#{server}/contacts/#{contact_id}/timezone")
+      logger.should_receive(:info).with("  Response Code: 204")
+
+      result = Flapjack::Diner.delete_contact_timezone!(contact_id)
+      req.should have_been_requested
+      result.should be_true
+    end
+
+  end
+
+  context "problems" do
+
+    it "raises an exception on network failure" do
+      req = stub_request(:get, "http://#{server}/entities").to_timeout
+
+      expect {
+        Flapjack::Diner.entities
+      }.to raise_error
+      req.should have_been_requested
+    end
+
+    it "raises an exception on invalid JSON data" do
+      req = stub_request(:get, "http://#{server}/entities").to_return(
+        :body => "{")
+
+      expect {
+        Flapjack::Diner.entities
+      }.to raise_error
+      req.should have_been_requested
+    end
+
+    it "raises an exception if a required argument is not provided" do
+      req = stub_request(:get, /http:\/\/#{server}\/*/)
+
+      expect {
+        Flapjack::Diner.check_status(entity, nil)
+      }.to raise_error
+      req.should_not have_been_requested
+    end
+
+    it "raises an exception if bulk queries don't have entity or check arguments" do
+      req = stub_request(:get, /http:\/\/#{server}\/*/)
+
+      expect {
+        Flapjack::Diner.bulk_downtime({})
+      }.to raise_error
+      req.should_not have_been_requested
+    end
+
+    it "raises an exception if bulk queries have invalid entity arguments" do
+      req = stub_request(:get, /http:\/\/#{server}\/*/)
+
+      expect {
+        Flapjack::Diner.bulk_scheduled_maintenances(:entity => 23)
+      }.to raise_error
+      req.should_not have_been_requested
+    end
+
+    it "raises an exception if bulk queries have invalid check arguments" do
+      req = stub_request(:get, /http:\/\/#{server}\/*/)
+
+      expect {
+        Flapjack::Diner.bulk_outages(:check => {'abc.com' => ['ping', 5]})
+      }.to raise_error
+      req.should_not have_been_requested
+    end
+
+    it "raises an exception if a time argument is provided with the wrong data type" do
+      start_str  = '2011-08-01T00:00:00+10:00'
+      finish_str = '2011-08-31T00:00:00+10:00'
+
+      start  = Time.iso8601(start_str)
+
+      req = stub_request(:get, /http:\/\/#{server}\/*/)
+
+      expect {
+        Flapjack::Diner.downtime(entity, :start_time => start, :end_time => finish_str)
+      }.to raise_error
+      req.should_not have_been_requested
+    end
+
+  end
+
+end