contact_spec.rb in flapjack-0.6.40

- old
+ new
@@ -2,44 +2,52 @@
 
 require 'flapjack/data/contact'
 
 describe Flapjack::Data::Contact, :redis => true do
 
-  it "returns a list of all contacts" do
+  def add_contacts
     Flapjack::Data::Contact.add({'id'         => '362',
                                  'first_name' => 'John',
                                  'last_name'  => 'Johnson',
                                  'email'      => 'johnj@example.com' },
                                  :redis       => @redis)
     Flapjack::Data::Contact.add({'id'         => '363',
                                  'first_name' => 'Jane',
                                  'last_name'  => 'Janeley',
                                  'email'      => 'janej@example.com'},
                                  :redis       => @redis)
+  end
 
+  it "returns a list of all contacts" do
+    add_contacts
+
     contacts = Flapjack::Data::Contact.all(:redis => @redis)
     contacts.should_not be_nil
     contacts.should be_an(Array)
     contacts.should have(2).contacts
-    contacts[0].id.should == '362'
-    contacts[1].id.should == '363'
+    contacts[0].name.should == 'Jane Janeley'
+    contacts[1].name.should == 'John Johnson'
   end
 
   it "finds a contact by id" do
-    Flapjack::Data::Contact.add({'id'         => '362',
-                                 'first_name' => 'John',
-                                 'last_name'  => 'Johnson',
-                                 'email'      => 'johnj@example.com' },
-                                 :redis       => @redis)
+    add_contacts
 
     contact = Flapjack::Data::Contact.find_by_id('362', :redis => @redis)
     contact.should_not be_nil
     contact.name.should == "John Johnson"
   end
 
   it "finds all contacts for a check on an entity"
 
-  it "deletes all contacts"
+  it "deletes all contacts" do
+    add_contacts
+
+    Flapjack::Data::Contact.delete_all(:redis => @redis)
+    contact = Flapjack::Data::Contact.find_by_id('362', :redis => @redis)
+    contact.should be_nil
+    contact = Flapjack::Data::Contact.find_by_id('363', :redis => @redis)
+    contact.should be_nil
+  end
 
   it "returns pagerduty credentials for a contact"
 
   it "returns a list of entities for a contact"
 