flapjack 0.5.5 → 0.6.23

data/spec/spec_helper.rb

@@ -0,0 +1,44 @@
+if ENV['COVERAGE']
+  require 'simplecov'
+
+  SimpleCov.start do
+    add_filter '/spec/'
+  end
+end
+
+FLAPJACK_ENV = ENV["FLAPJACK_ENV"] || 'test'
+require 'bundler'
+Bundler.require(:default, :test)
+
+$:.unshift(File.dirname(__FILE__) + '/../lib')
+
+# Requires supporting files with custom matchers and macros, etc,
+# in ./support/ and its subdirectories.
+Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each {|f| require f}
+
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# Require this file using `require "spec_helper"` to ensure that it is only
+# loaded once.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+RSpec.configure do |config|
+  config.treat_symbols_as_metadata_keys_with_true_values = true
+  config.run_all_when_everything_filtered = true
+  config.filter_run :focus
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = 'random'
+
+  config.around(:each, :redis => true) do |example|
+    @redis = ::Redis.new(:db => 14, :driver => :ruby)
+    @redis.flushdb
+    example.run
+    @redis.quit
+  end
+
+  config.include Rack::Test::Methods, :sinatra => true
+end

data/spec/support/profile_all_formatter.rb

@@ -0,0 +1,44 @@
+require 'rspec/core/formatters/base_formatter'
+
+class ProfileAllFormatter < RSpec::Core::Formatters::BaseFormatter
+
+  def initialize(output)
+    super(output)
+    @example_times = []
+  end
+
+  def start(count)
+    super(count)
+    @output.puts "Profiling enabled."
+  end
+
+  def example_started(example)
+    super(example)
+    @time = ((Time.respond_to?(:zone) && Time.zone) ? Time.zone.now : Time.now)
+  end
+
+  def example_passed(example)
+    @example_times << [
+      example_group.description,
+      example.description,
+      ((Time.respond_to?(:zone) && Time.zone) ? Time.zone.now : Time.now) - @time
+    ]
+    super(example)
+  end
+
+  def start_dump
+    super
+    @output.puts "\n\nExample times:\n"
+
+    @example_times = @example_times.sort_by do |description, example, time|
+      time
+    end.reverse
+
+    @example_times.each do |description, example, time|
+      @output.print sprintf("%.7f", time)
+      @output.puts " #{description} #{example}"
+    end
+    @output.flush
+  end
+
+end

data/spec/support/uncolored_doc_formatter.rb

@@ -0,0 +1,9 @@
+require 'rspec/core/formatters/documentation_formatter'
+
+class UncoloredDocFormatter < RSpec::Core::Formatters::DocumentationFormatter
+
+  def color(text, color_code)
+    text
+  end
+
+end

data/tasks/events.rake

@@ -0,0 +1,85 @@
+
+require 'redis'
+require 'yaml'
+require 'yajl'
+
+namespace :events do
+
+  # FIXME: add arguments, make more flexible
+  desc "send events to trigger some notifications"
+  task :test_notification do
+
+    # config file reading stuff ...
+    # FIXME: move this into somewhere reusable
+    FLAPJACK_ENV = ENV['FLAPJACK_ENV'] || 'development'
+    config_file = File.join('etc', 'flapjack_config.yaml')
+    if File.file?(config_file)
+      config = YAML::load(File.open(config_file))
+    else
+      puts "Could not find 'etc/flapjack_config.yaml'"
+      exit(false)
+    end
+    config_env = config[FLAPJACK_ENV]
+    @redis_host = config_env['redis']['host'] || 'localhost'
+    @redis_port = config_env['redis']['port'] || '6379'
+    @redis_path = config_env['redis']['path'] || nil
+    @redis_db   = config_env['redis']['db']   || 0
+
+    if config_env.nil? || config_env.empty?
+        puts "No config data for environment '#{FLAPJACK_ENV}'"
+          exit(false)
+    end
+
+    # add lib to the default include path
+    unless $:.include?(File.dirname(__FILE__) + '/../lib/')
+      $: << File.dirname(__FILE__) + '/../lib'
+    end
+
+    def get_redis_connection
+      if @redis_path
+        redis = Redis.new(:db => @redis_db, :path => @redis_path)
+      else
+        redis = Redis.new(:db => @redis_db, :host => @redis_host, :port => @redis_port)
+      end
+      redis
+    end
+
+    # creates an event object and adds it to the events list in redis
+    #   'entity'    => entity,
+    #   'check'     => check,
+    #   'type'      => 'service',
+    #   'state'     => state,
+    #   'summary'   => check_output,
+    #   'time'      => timestamp,
+    def create_event(event)
+      redis = get_redis_connection
+      evt = Yajl::Encoder.encode(event)
+      puts "sending #{evt}"
+      redis.rpush('events', evt)
+    end
+
+    create_event( 'entity'  => 'clientx-app-01',
+                  'check'   => 'ping',
+                  'type'    => 'service',
+                  'state'   => 'ok',
+                  'summary' => 'testing' )
+
+    sleep(8)
+
+    create_event( 'entity'  => 'clientx-app-01',
+                  'check'   => 'ping',
+                  'type'    => 'service',
+                  'state'   => 'critical',
+                  'summary' => 'testing' )
+
+    sleep(8)
+
+    create_event( 'entity'  => 'clientx-app-01',
+                  'check'   => 'ping',
+                  'type'    => 'service',
+                  'state'   => 'ok',
+                  'summary' => 'testing' )
+
+  end
+
+end

data/tmp/acknowledge.rb

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require 'redis'
+require 'json'
+
+if not id = ARGV.first then
+  puts "Usage: acknowledge.rb <id>"
+  exit 1
+end
+
+
+redis = Redis.new
+
+redis.hset 'acknowledged', id, 'true'

data/tmp/create_config_yaml.rb

@@ -0,0 +1,16 @@
+#!/usr/bin/env ruby
+
+
+require 'yaml'
+
+config = { 'sausage' => true,
+           'sauce'   => 'tomato sauce',
+           'count'   => 23,
+           'foobar'  => 'jkfldjsk',
+           'zero_s'  => '0',
+           'true_s'  => 'true',
+         }
+
+
+puts config.to_yaml
+

data/tmp/create_events_failure.rb

@@ -0,0 +1,33 @@
+#!/usr/bin/env ruby
+
+require 'redis'
+require 'json'
+
+id = "%.2d" % (1..10).to_a[rand(9)]
+
+events = []
+
+events << {
+  'entity' => "app-#{id}",
+  'check' => 'http',
+  'type' => 'service',
+  'state' => 'critical',
+}.to_json
+
+redis = Redis.new
+
+2000.times do
+  events.each {|event|
+    redis.rpush 'events', event
+  }
+end
+
+puts "#{Time.now} - finished loading up events"
+previous_events_size = redis.llen 'events'
+while previous_events_size > 0
+  sleep 1
+  events_size = redis.llen 'events'
+  throughput = previous_events_size - events_size
+  previous_events_size = events_size
+  puts "#{Time.now} - #{events_size} (#{throughput})"
+end

data/tmp/create_events_ok.rb

@@ -0,0 +1,33 @@
+#!/usr/bin/env ruby
+
+require 'redis'
+require 'json'
+
+id = "%.2d" % (1..10).to_a[rand(9)]
+
+events = []
+
+events << {
+  'entity' => "client1-test-app-#{id}",
+  'check' => 'http',
+  'type' => 'service',
+  'state' => 'ok',
+}.to_json
+
+redis = Redis.new
+
+4000.times do
+  events.each {|event|
+    redis.rpush 'events', event
+  }
+end
+
+puts "#{Time.now} - finished loading up events"
+previous_events_size = redis.llen 'events'
+while previous_events_size > 0
+  sleep 1
+  events_size = redis.llen 'events'
+  throughput = previous_events_size - events_size
+  previous_events_size = events_size
+  puts "#{Time.now} - #{events_size} (#{throughput})"
+end

data/tmp/create_events_ok_fail_ack_ok.rb

@@ -0,0 +1,54 @@
+#!/usr/bin/env ruby
+
+require 'redis'
+require 'json'
+
+id = "%.2d" % (1..10).to_a[rand(9)]
+
+events = []
+
+events << {
+  'entity' => "app-#{id}",
+  'check' => 'http',
+  'type' => 'service',
+  'state' => 'ok',
+}.to_json
+
+events << {
+  'entity' => "app-#{id}",
+  'check' => 'http',
+  'type' => 'service',
+  'state' => 'critical',
+}.to_json
+
+events << {
+  'entity' => "app-#{id}",
+  'check' => 'http',
+  'type' => 'action',
+  'state' => 'acknowledgement',
+}.to_json
+
+events << {
+  'entity' => "app-#{id}",
+  'check' => 'http',
+  'type' => 'service',
+  'state' => 'ok',
+}.to_json
+
+redis = Redis.new
+
+2000.times do
+  events.each {|event|
+    redis.rpush 'events', event
+  }
+end
+
+puts "#{Time.now} - finished loading up events"
+previous_events_size = redis.llen 'events'
+while previous_events_size > 0
+  sleep 1
+  events_size = redis.llen 'events'
+  throughput = previous_events_size - events_size
+  previous_events_size = events_size
+  puts "#{Time.now} - #{events_size} (#{throughput})"
+end

data/tmp/create_events_ok_failure.rb

@@ -0,0 +1,40 @@
+#!/usr/bin/env ruby
+
+require 'redis'
+require 'json'
+
+id = "%.2d" % (1..10).to_a[rand(9)]
+
+events = []
+
+events << {
+  'entity' => "app-#{id}",
+  'check' => 'http',
+  'type' => 'service',
+  'state' => 'ok',
+}.to_json
+
+events << {
+  'entity' => "app-#{id}",
+  'check' => 'http',
+  'type' => 'service',
+  'state' => 'critical',
+}.to_json
+
+redis = Redis.new
+
+2000.times do
+  events.each {|event|
+    redis.rpush 'events', event
+  }
+end
+
+puts "#{Time.now} - finished loading up events"
+previous_events_size = redis.llen 'events'
+while previous_events_size > 0
+  sleep 1
+  events_size = redis.llen 'events'
+  throughput = previous_events_size - events_size
+  previous_events_size = events_size
+  puts "#{Time.now} - #{events_size} (#{throughput})"
+end

data/tmp/create_events_ok_failure_ack.rb

@@ -0,0 +1,54 @@
+#!/usr/bin/env ruby
+
+require 'redis'
+require 'json'
+
+id = "%.2d" % (1..10).to_a[rand(9)]
+
+events = []
+
+events << {
+  'entity' => "app-#{id}",
+  'check' => 'http',
+  'type' => 'service',
+  'state' => 'ok',
+}.to_json
+
+events << {
+  'entity' => "app-#{id}",
+  'check' => 'http',
+  'type' => 'service',
+  'state' => 'critical',
+}.to_json
+
+events << {
+  'entity' => "app-#{id}",
+  'check' => 'http',
+  'type' => 'action',
+  'state' => 'acknowledgement',
+}.to_json
+
+#events << {
+#  'entity' => "app-#{id}",
+#  'check' => 'http',
+#  'type' => 'service',
+#  'state' => 'ok',
+#}.to_json
+
+redis = Redis.new
+
+2000.times do
+  events.each {|event|
+    redis.rpush 'events', event
+  }
+end
+
+puts "#{Time.now} - finished loading up events"
+previous_events_size = redis.llen 'events'
+while previous_events_size > 0
+  sleep 1
+  events_size = redis.llen 'events'
+  throughput = previous_events_size - events_size
+  previous_events_size = events_size
+  puts "#{Time.now} - #{events_size} (#{throughput})"
+end

data/tmp/dummy_entities.json

@@ -0,0 +1 @@
+[{"id":"10001","name":"clientx-app-01","contacts":["0362","0363","0364"]},{"id":"10002","name":"clientx-app-02","contacts":["0362"]},{"id":"10003","name":"clienty-app-01","contacts":["0363","0364"]}]