flapjack 1.6.0 → 2.0.0b1

data/spec/service_consumers/provider_states_for_flapjack-diner.rb

@@ -1,420 +1,377 @@
 Pact.provider_states_for "flapjack-diner" do
 
-  provider_state "no contact exists" do
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
-    end
+  provider_state "no data exists" do
+    set_up    { default_tear_down }
+    tear_down { default_tear_down }
   end
 
-  provider_state "no entity exists" do
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+  provider_state "a check exists" do
+    set_up do
+      check = Flapjack::Data::Check.new(check_data)
+      check.save!
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "no check exists" do
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+  provider_state "two checks exist" do
+    set_up do
+      check = Flapjack::Data::Check.new(check_data)
+      check.save!
+
+      check_2 = Flapjack::Data::Check.new(check_2_data)
+      check_2.save!
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "no notification rule exists" do
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+  provider_state "a contact exists" do
+    set_up do
+      contact = Flapjack::Data::Contact.new(contact_data)
+      contact.save!
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "an entity 'www.example.com' with id '1234' exists" do
+  provider_state "a contact with one medium exists" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
+      contact = Flapjack::Data::Contact.new(contact_data)
+      contact.save!
 
-      entity_data = {'id' => '1234', 'name' => 'www.example.com'}
-      Flapjack::Data::Entity.add(entity_data, :redis => redis)
-    end
+      email = Flapjack::Data::Medium.new(email_data)
+      email.save!
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+      contact.media << email
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "entities 'www.example.com', id '1234' and 'www2.example.com', id '5678' exist" do
+  provider_state "a contact with one medium and one rule exists" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
+      contact = Flapjack::Data::Contact.new(contact_data)
+      contact.save!
 
-      entity_data = {'id' => '1234', 'name' => 'www.example.com'}
-      Flapjack::Data::Entity.add(entity_data, :redis => redis)
-      entity_data_2 = {'id' => '5678', 'name' => 'www2.example.com'}
-      Flapjack::Data::Entity.add(entity_data_2, :redis => redis)
-    end
+      email = Flapjack::Data::Medium.new(email_data)
+      email.save!
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+      contact.media << email
+
+      rule = Flapjack::Data::Rule.new(rule_data)
+      rule.save!
+
+      contact.rules << rule
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "a check 'www.example.com:SSH' exists" do
+  provider_state "two contacts exist" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
+      contact = Flapjack::Data::Contact.new(contact_data)
+      contact.save!
 
-      entity_data = {'id' => '1234', 'name' => 'www.example.com'}
-      Flapjack::Data::Entity.add(entity_data, :redis => redis)
-      check_data = {'entity_id' => '1234', 'name' => 'SSH'}
-      Flapjack::Data::EntityCheck.add(check_data, :redis => redis)
+      contact_2 = Flapjack::Data::Contact.new(contact_2_data)
+      contact_2.save!
     end
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
-    end
+    tear_down { default_tear_down }
   end
 
-  provider_state "checks 'www.example.com:SSH' and 'www2.example.com:PING' exist" do
+  provider_state "a medium exists" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
+      sms = Flapjack::Data::Medium.new(sms_data)
+      sms.save!
+    end
 
-      entity_data = {'id' => '1234', 'name' => 'www.example.com'}
-      Flapjack::Data::Entity.add(entity_data, :redis => redis)
-      entity_data_2 = {'id' => '5678', 'name' => 'www2.example.com'}
-      Flapjack::Data::Entity.add(entity_data_2, :redis => redis)
+    tear_down { default_tear_down }
+  end
 
-      check_data = {'entity_id' => '1234', 'name' => 'SSH'}
-      Flapjack::Data::EntityCheck.add(check_data, :redis => redis)
-      check_data_2 = {'entity_id' => '5678', 'name' => 'PING'}
-      Flapjack::Data::EntityCheck.add(check_data_2, :redis => redis)
-    end
+  provider_state "two media exist" do
+    set_up do
+      sms = Flapjack::Data::Medium.new(sms_data)
+      sms.save!
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+      email = Flapjack::Data::Medium.new(email_data)
+      email.save!
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "a contact with id 'abc' exists" do
+  provider_state "a rule exists" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      contact_data = {'id'         => 'abc',
-                      'first_name' => 'Jim',
-                      'last_name'  => 'Smith',
-                      'email'      => 'jims@example.com',
-                      'timezone'   => 'UTC',
-                      'tags'       => ['admin', 'night_shift']}
-      Flapjack::Data::Contact.add(contact_data, :redis => redis)
+      rule = Flapjack::Data::Rule.new(rule_data)
+      rule.save!
     end
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+    tear_down { default_tear_down }
+  end
+
+  provider_state "a state exists" do
+    set_up do
+      sd = state_data.dup
+      [:created_at, :updated_at].each {|t| sd[t] = Time.parse(sd[t]) }
+
+      state = Flapjack::Data::State.new(sd)
+      state.save!
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "a contact with id 'abc' has email and sms media" do
+  provider_state "two rules exist" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      contact_data = {'id'         => 'abc',
-                      'first_name' => 'Jim',
-                      'last_name'  => 'Smith',
-                      'email'      => 'jims@example.com',
-                      'timezone'   => 'UTC',
-                      'tags'       => ['admin', 'night_shift']}
-      contact = Flapjack::Data::Contact.add(contact_data, :redis => redis)
-
-      email_data = {
-        'type'             => 'email',
-        'address'          => 'ablated@example.org',
-        'interval'         => 180,
-        'rollup_threshold' => 3
-      }
-
-      sms_data = {
-        'type'             => 'sms',
-        'address'          => '0123456789',
-        'interval'         => 300,
-        'rollup_threshold' => 5
-      }
-
-      [email_data, sms_data].each do |medium_data|
-        type = medium_data['type']
-        contact.set_address_for_media(type, medium_data['address'])
-        contact.set_interval_for_media(type, medium_data['interval'])
-        contact.set_rollup_threshold_for_media(type, medium_data['rollup_threshold'])
-      end
+      rule = Flapjack::Data::Rule.new(rule_data)
+      rule.save!
+
+      rule_2 = Flapjack::Data::Rule.new(rule_2_data)
+      rule_2.save!
     end
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+    tear_down { default_tear_down }
+  end
+
+  provider_state "a tag exists" do
+    set_up do
+      tag = Flapjack::Data::Tag.new(tag_data)
+      tag.save!
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "a contact with id '872' exists" do
+  provider_state "two tags exist" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      contact_data = {'id' => '872', 'first_name' => 'John',
-                      'last_name' => 'Smith', 'email' => 'jsmith@example.com'}
-      Flapjack::Data::Contact.add(contact_data, :redis => redis)
+      tag = Flapjack::Data::Tag.new(tag_data)
+      tag.save!
+
+      tag_2 = Flapjack::Data::Tag.new(tag_2_data)
+      tag_2.save!
     end
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+    tear_down { default_tear_down }
+  end
+
+  provider_state "a scheduled maintenance period exists" do
+    set_up do
+      smd = scheduled_maintenance_data.dup
+      [:start_time, :end_time].each {|t| smd[t] = Time.parse(smd[t]) }
+      scheduled_maintenance = Flapjack::Data::ScheduledMaintenance.new(smd)
+      scheduled_maintenance.save!
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "contacts with ids 'abc' and '872' exist" do
+  provider_state "two scheduled maintenance periods exist" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      contact_data = {'id'         => 'abc',
-                      'first_name' => 'Jim',
-                      'last_name'  => 'Smith',
-                      'email'      => 'jims@example.com',
-                      'timezone'   => 'UTC',
-                      'tags'       => ['admin', 'night_shift']}
-      Flapjack::Data::Contact.add(contact_data, :redis => redis)
-      contact_data_2 = {'id'         => '872',
-                        'first_name' => 'John',
-                        'last_name'  => 'Smith',
-                        'email'      => 'jsmith@example.com'}
-      Flapjack::Data::Contact.add(contact_data_2, :redis => redis)
+      smd = scheduled_maintenance_data.dup
+      [:start_time, :end_time].each {|t| smd[t] = Time.parse(smd[t]) }
+      scheduled_maintenance = Flapjack::Data::ScheduledMaintenance.new(smd)
+      scheduled_maintenance.save!
+
+      smd_2 = scheduled_maintenance_2_data.dup
+      [:start_time, :end_time].each {|t| smd_2[t] = Time.parse(smd_2[t]) }
+      scheduled_maintenance_2 = Flapjack::Data::ScheduledMaintenance.new(smd_2)
+      scheduled_maintenance_2.save!
     end
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+    tear_down { default_tear_down }
+  end
+
+  provider_state "an unscheduled maintenance period exists" do
+    set_up do
+      usmd = unscheduled_maintenance_data.dup
+      [:start_time, :end_time].each {|t| usmd[t] = Time.parse(usmd[t]) }
+      unscheduled_maintenance = Flapjack::Data::UnscheduledMaintenance.new(usmd)
+      unscheduled_maintenance.save!
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "a contact 'abc' with generic notification rule '05983623-fcef-42da-af44-ed6990b500fa' exists" do
+  provider_state "two unscheduled maintenance periods exist" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-
-      contact_data = {'id'         => 'abc',
-                      'first_name' => 'Jim',
-                      'last_name'  => 'Smith',
-                      'email'      => 'jims@example.com',
-                      'timezone'   => 'UTC',
-                      'tags'       => ['admin', 'night_shift']}
-      contact = Flapjack::Data::Contact.add(contact_data, :redis => redis)
-      existing_nr = contact.notification_rules.first
-
-      nr_data = {
-        :id                 => '05983623-fcef-42da-af44-ed6990b500fa',
-        :tags               => [],
-        :regex_tags         => [],
-        :entities           => [],
-        :regex_entities     => [],
-        :time_restrictions  => [],
-        :warning_media      => ["email"],
-        :critical_media     => ["sms", "email"],
-        :warning_blackhole  => false,
-        :critical_blackhole => false
-      }
-      contact.add_notification_rule(nr_data)
-      contact.delete_notification_rule(existing_nr)
+      usmd = unscheduled_maintenance_data.dup
+      [:start_time, :end_time].each {|t| usmd[t] = Time.parse(usmd[t]) }
+      unscheduled_maintenance = Flapjack::Data::UnscheduledMaintenance.new(usmd)
+      unscheduled_maintenance.save!
+
+      usmd_2 = unscheduled_maintenance_2_data.dup
+      [:start_time, :end_time].each {|t| usmd_2[t] = Time.parse(usmd_2[t]) }
+      unscheduled_maintenance_2 = Flapjack::Data::UnscheduledMaintenance.new(usmd_2)
+      unscheduled_maintenance_2.save!
     end
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+    tear_down { default_tear_down }
+  end
+
+  provider_state "a global statistics object exists" do
+    set_up do
+      gsd = global_statistics_data.dup
+      gsd[:created_at] = Time.parse(gsd[:created_at])
+      global_stats = Flapjack::Data::Statistic.new(gsd)
+      global_stats.save!
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "a contact 'abc' with generic notification rule '05983623-fcef-42da-af44-ed6990b500fa' and notification rule '20f182fc-6e32-4794-9007-97366d162c51' exists" do
+  provider_state "a check and a tag exist" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-
-      contact_data = {'id'         => 'abc',
-                      'first_name' => 'Jim',
-                      'last_name'  => 'Smith',
-                      'email'      => 'jims@example.com',
-                      'timezone'   => 'UTC',
-                      'tags'       => ['admin', 'night_shift']}
-      contact = Flapjack::Data::Contact.add(contact_data, :redis => redis)
-      existing_nr = contact.notification_rules.first
-
-      nr_data = {
-        :id                 => '05983623-fcef-42da-af44-ed6990b500fa',
-        :tags               => [],
-        :regex_tags         => [],
-        :entities           => [],
-        :regex_entities     => [],
-        :time_restrictions  => [],
-        :warning_media      => ["email"],
-        :critical_media     => ["sms", "email"],
-        :warning_blackhole  => false,
-        :critical_blackhole => false
-      }
-      contact.add_notification_rule(nr_data)
-
-      nr_data_2 = {
-        :id                 => '20f182fc-6e32-4794-9007-97366d162c51',
-        :tags               => ['physical'],
-        :regex_tags         => [],
-        :entities           => ['example.com'],
-        :regex_entities     => [],
-        :time_restrictions  => [],
-        :warning_media      => ["email"],
-        :critical_media     => ["sms", "email"],
-        :warning_blackhole  => true,
-        :critical_blackhole => true
-      }
-      contact.add_notification_rule(nr_data_2)
-      contact.delete_notification_rule(existing_nr)
+      check = Flapjack::Data::Check.new(check_data)
+      check.save!
+
+      tag = Flapjack::Data::Tag.new(tag_data)
+      tag.save!
     end
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+    tear_down { default_tear_down }
+  end
+
+  provider_state "a contact and a medium exist" do
+    set_up do
+      contact = Flapjack::Data::Contact.new(contact_data)
+      contact.save!
+
+      email = Flapjack::Data::Medium.new(email_data)
+      email.save!
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "a contact with id 'abc' has pagerduty credentials" do
+  provider_state "a contact with a medium exists" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      contact_data = {'id'         => 'abc',
-                      'first_name' => 'Jim',
-                      'last_name'  => 'Smith',
-                      'email'      => 'jims@example.com',
-                      'timezone'   => 'UTC',
-                      'tags'       => ['admin', 'night_shift']}
-      contact = Flapjack::Data::Contact.add(contact_data, :redis => redis)
-
-      pdc_data = {
-        'service_key' => 'abc',
-        'subdomain'   => 'def',
-        'token'       => 'ghi',
-      }
-      contact.set_pagerduty_credentials(pdc_data)
+      contact = Flapjack::Data::Contact.new(contact_data)
+      contact.save!
+
+      email = Flapjack::Data::Medium.new(email_data)
+      email.save!
+
+      contact.media << email
     end
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+    tear_down { default_tear_down }
+  end
+
+  provider_state "a contact with a rule exists" do
+    set_up do
+      contact = Flapjack::Data::Contact.new(contact_data)
+      contact.save!
+
+      rule = Flapjack::Data::Rule.new(rule_data)
+      rule.save!
+
+      contact.rules << rule
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "contacts with ids 'abc' and '872' have pagerduty credentials" do
+  provider_state "a check with a tag exists" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      contact_data = {'id'         => 'abc',
-                      'first_name' => 'Jim',
-                      'last_name'  => 'Smith',
-                      'email'      => 'jims@example.com',
-                      'timezone'   => 'UTC',
-                      'tags'       => ['admin', 'night_shift']}
-      contact = Flapjack::Data::Contact.add(contact_data, :redis => redis)
-      contact_data_2 = {'id'         => '872',
-                        'first_name' => 'John',
-                        'last_name'  => 'Smith',
-                        'email'      => 'jsmith@example.com'}
-      contact_2 = Flapjack::Data::Contact.add(contact_data_2, :redis => redis)
-
-      pdc_data = {
-        'service_key' => 'abc',
-        'subdomain'   => 'def',
-        'token'       => 'ghi',
-      }
-      contact.set_pagerduty_credentials(pdc_data)
-      pdc_data_2 = {
-        'service_key' => 'mno',
-        'subdomain'   => 'pqr',
-        'token'       => 'stu',
-    }
-    contact_2.set_pagerduty_credentials(pdc_data_2)
+      check = Flapjack::Data::Check.new(check_data)
+      check.save!
+
+      tag = Flapjack::Data::Tag.new(tag_data)
+      tag.save!
+
+      check.tags << tag
     end
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+    tear_down { default_tear_down }
+  end
+
+  provider_state "a check and two tags exist" do
+    set_up do
+      check = Flapjack::Data::Check.new(check_data)
+      check.save!
+
+      tag = Flapjack::Data::Tag.new(tag_data)
+      tag.save!
+
+      tag_2 = Flapjack::Data::Tag.new(tag_2_data)
+      tag_2.save!
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "flapjack has a simplistic check freshness distribution" do
+  provider_state "a check with two tags exists" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      check_freshness_data = {
-        :from_0_to_60     => 0,
-        :from_60_to_300   => 1,
-        :from_300_to_900  => 2,
-        :from_900_to_3600 => 3,
-        :from_3600_to_∞   => 4
-      }
-      entity = Flapjack::Data::Entity.find_by_name('sally', :create => true, :redis => redis)
-
-      checks = {
-        "check 1" => 61,
-        "check 2" => 301,
-        "check 3" => 301,
-        "check 4" => 901,
-        "check 5" => 901,
-        "check 6" => 901,
-        "check 7" => 3601,
-        "check 8" => 3601,
-        "check 9" => 3601,
-        "check 10" => 3601,
-      }
-      checks.each_pair {|check_name, age|
-        check = Flapjack::Data::EntityCheck.new(entity, check_name, :redis => redis)
-        check.update_state(Flapjack::Data::EntityCheck::STATE_OK, :timestamp => Time.new.to_i - age)
-      } 
+      check = Flapjack::Data::Check.new(check_data)
+      check.save!
+
+      tag = Flapjack::Data::Tag.new(tag_data)
+      tag.save!
+
+      tag_2 = Flapjack::Data::Tag.new(tag_2_data)
+      tag_2.save!
+
+      check.tags.add(tag, tag_2)
     end
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+    tear_down { default_tear_down }
+  end
+
+  provider_state "a tag with two checks exists" do
+    set_up do
+      tag = Flapjack::Data::Tag.new(tag_data)
+      tag.save!
+
+      check = Flapjack::Data::Check.new(check_data)
+      check.save!
+
+      check_2 = Flapjack::Data::Check.new(check_2_data)
+      check_2.save!
+
+      tag.checks.add(check, check_2)
     end
+
+    tear_down { default_tear_down }
   end
 
-  provider_state "flapjack has processed 0 ok, 1 failure, 2 ack, and 3 invalid events with 4 on the queue" do
+  provider_state "a check with a tag and a rule exists" do
     set_up do
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      failure_event = {
-        'type'    => 'service',
-        'state'   => 'critical',
-        'summary' => '50% packet loss',
-        'entity'  => 'foo-app-01.example',
-        'check'   => 'ping'
-      }
-
-      # fake out the metrics
-      redis.hmset('event_counters',
-        'all'     , 6,
-        'ok'      , 0,
-        'failure' , 1,
-        'action'  , 2,
-        'invalid' , 3
-        )
-
-      4.times{ redis.lpush('events', Flapjack.dump_json(failure_event)) }
+      check = Flapjack::Data::Check.new(check_data)
+      check.save!
+
+      tag = Flapjack::Data::Tag.new(tag_data)
+      tag.save!
+
+      rule = Flapjack::Data::Rule.new(rule_data)
+      rule.save!
+
+      check.tags << tag
+      rule.tags << tag
     end
 
-    tear_down do
-      Flapjack::Gateways::JSONAPI.instance_variable_get('@logger').messages.clear
-      redis = Flapjack::Gateways::JSONAPI.instance_variable_get('@redis')
-      redis.flushdb
+    tear_down { default_tear_down }
+  end
+
+  provider_state "a check with a tag, current state and a latest notification exists" do
+    set_up do
+      check = Flapjack::Data::Check.new(check_data)
+      check.save!
+
+      tag = Flapjack::Data::Tag.new(tag_data)
+      tag.save!
+
+      sd = state_data.dup
+      [:created_at, :updated_at].each {|t| sd[t] = Time.parse(sd[t]) }
+      state = Flapjack::Data::State.new(sd)
+      state.save!
+
+      check.tags << tag
+      check.current_state = state
+      check.latest_notifications << state
     end
+
+    tear_down { default_tear_down }
   end
 
-end
+end