ipvs_litmus 0.0.1

data/spec/ipvs_litmus/app_spec.rb

@@ -0,0 +1,215 @@
+require 'spec_helper'
+
+describe IPVSLitmus::App do
+  def app
+    IPVSLitmus::App
+  end
+
+  describe "POST /force/*" do
+    it "creates a global upfile" do
+      test_service = IPVSLitmus::Service.new('test', [NeverAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      post "/force/up", :reason => "up for testing"
+      last_response.status.should == 201
+
+      get "/test/status"
+      last_response.status.should == 200
+      last_response.body.should match(/up for testing/)
+    end
+
+    it "creates a global downfile" do
+      test_service = IPVSLitmus::Service.new('test', [AlwaysAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      post "/force/down", :reason => "down for testing"
+      last_response.status.should == 201
+
+      get "/test/status"
+      last_response.status.should == 503
+      last_response.body.should match(/down for testing/)
+    end
+
+    it "creates a service specific upfile" do
+      test_service = IPVSLitmus::Service.new('test', [NeverAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      post "/force/up/test", :reason => "up for testing"
+      last_response.status.should == 201
+
+      get "/test/status"
+      last_response.status.should == 200
+      last_response.body.should match(/up for testing/)
+    end
+  end
+
+  describe "DELETE /force/*" do
+    it "removes the global upfile" do
+      test_service = IPVSLitmus::Service.new('test', [NeverAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      post "/force/up", :reason => "up for testing"
+      last_response.status.should == 201
+
+      get "/test/status"
+      last_response.status.should == 200
+
+      delete "/force/up"
+      last_response.status.should == 200
+
+      get "/test/status"
+      last_response.status.should == 503
+      last_response.body.should_not match(/up for testing/)
+    end
+
+    it "removes the global downfile" do
+      test_service = IPVSLitmus::Service.new('test', [AlwaysAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      post "/force/down", :reason => "down for testing"
+      last_response.status.should == 201
+
+      get "/test/status"
+      last_response.status.should == 503
+
+      delete "/force/down"
+      last_response.status.should == 200
+
+      get "/test/status"
+      last_response.should be_ok
+      last_response.body.should_not match(/down for testing/)
+    end
+
+    it "removes a service specific upfile" do
+      test_service = IPVSLitmus::Service.new('test', [NeverAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      post "/force/up/test", :reason => "up for testing"
+      last_response.status.should == 201
+
+      get "/test/status"
+      last_response.status.should == 200
+      last_response.body.should match(/up for testing/)
+
+      delete "/force/up/test"
+      last_response.status.should == 200
+
+      get "/test/status"
+      last_response.status.should == 503
+    end
+
+    it "404s if there is no upfile" do
+      test_service = IPVSLitmus::Service.new('test', [NeverAvailableDependency.new], [ConstantMetric.new(100)])
+
+      delete "/up"
+
+      last_response.status.should == 404
+    end
+  end
+
+  describe "GET /:service/status" do
+    it "is successful when the service is passing" do
+      test_service = IPVSLitmus::Service.new('test', [AlwaysAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      get "/test/status"
+
+      last_response.should be_ok
+      last_response.header["Content-Type"].should == "text/plain"
+      last_response.body.should match(/Health: 100/)
+      last_response.body.should match(/AlwaysAvailableDependency: OK/)
+      last_response.body.should match(/ConstantMetric: 100/)
+    end
+
+    it "is 'service unavailable' when the check fails" do
+      test_service = IPVSLitmus::Service.new('test', [NeverAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      get "/test/status"
+
+      last_response.status.should == 503
+      last_response.header["Content-Type"].should == "text/plain"
+    end
+
+    it "is 'not found' when the service is unknown" do
+      get "/unknown/status"
+
+      last_response.status.should == 404
+      last_response.header["Content-Type"].should == "text/plain"
+    end
+
+    it "is 'service unavailable' when an up file and down file exists" do
+      test_service = IPVSLitmus::Service.new('test', [AlwaysAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      IPVSLitmus::StatusFile.new("up", "test").create("Up for testing")
+      IPVSLitmus::StatusFile.new("down", "test").create("Down for testing")
+
+      get "/test/status"
+
+      last_response.status.should == 503
+      last_response.body.should match(/Down for testing/)
+    end
+
+    it "is 'service available' when an up file exists" do
+      test_service = IPVSLitmus::Service.new('test', [NeverAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      IPVSLitmus::StatusFile.new("up", "test").create("Up for testing")
+
+      get "/test/status"
+
+      last_response.status.should == 200
+      last_response.body.should match(/Up for testing/)
+    end
+
+    it "is 'service available' when an up file exists" do
+      test_service = IPVSLitmus::Service.new('test', [NeverAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      IPVSLitmus::StatusFile.new("up", "test").create("Up for testing")
+
+      get "/test/status"
+
+      last_response.status.should == 200
+      last_response.body.should match(/Up for testing/)
+    end
+
+    it "is 'service unavailable' when a global down file and up file exists" do
+      test_service = IPVSLitmus::Service.new('test', [AlwaysAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      IPVSLitmus::StatusFile.new("global_down").create("Down for testing")
+      IPVSLitmus::StatusFile.new("global_up").create("Up for testing")
+
+      get "/test/status"
+
+      last_response.status.should == 503
+      last_response.body.should match(/Down for testing/)
+    end
+
+    it "is 'service unavailable' when a global down file exists" do
+      test_service = IPVSLitmus::Service.new('test', [AlwaysAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      IPVSLitmus::StatusFile.new("global_down").create("Down for testing")
+
+      get "/test/status"
+
+      last_response.status.should == 503
+      last_response.body.should match(/Down for testing/)
+    end
+
+    it "is successful when a global up file exists" do
+      test_service = IPVSLitmus::Service.new('test', [NeverAvailableDependency.new], [ConstantMetric.new(100)])
+      IPVSLitmus.services['test'] = test_service
+
+      IPVSLitmus::StatusFile.new("global_up").create("Up for testing")
+
+      get "/test/status"
+
+      last_response.status.should == 200
+      last_response.body.should match(/Up for testing/)
+    end
+  end
+end

data/spec/ipvs_litmus/cli/admin_spec.rb

@@ -0,0 +1,58 @@
+require 'spec_helper'
+
+
+describe 'litmusctl' do
+  def _litmusctl(args)
+    `bundle exec ruby -I lib bin/litmusctl #{args} -p 9293`
+  end
+
+  before(:all) do
+    system "bundle exec ruby -I lib bin/litmus -p 9293 -d -D /tmp/ipvs -c #{TEST_CONFIG} -P /tmp/ipvs.pid"
+  end
+
+  after(:all) do
+    system "kill -9 `cat /tmp/ipvs.pid`"
+  end
+
+  describe 'status' do
+    it 'returns the status of a service' do
+      _litmusctl('status test').should match("Health: 0")
+      _litmusctl('status passing_test').should match(/Health: \d\d/)
+    end
+
+    it "returns 'NOT FOUND' for a service that doesn't exist" do
+      _litmusctl('status unknown').should match('NOT FOUND')
+    end
+  end
+
+  describe "force" do
+    it "can create a global downfile" do
+      _litmusctl('force down -r "for testing"').should match("File created")
+
+      status = _litmusctl('status test')
+      status.should match(/Health: 0/)
+      status.should match(/for testing/)
+    end
+
+    it "creates a downfile" do
+      _litmusctl('force down test -r "for testing"').should match("File created")
+
+      status = _litmusctl('status test')
+      status.should match(/Health: 0/)
+      status.should match(/for testing/)
+    end
+
+    it 'removes an upfile for the service' do
+      _litmusctl('force up test -r "for testing"').should match("File created")
+      _litmusctl('force up test -d').should match("File deleted")
+
+      status = _litmusctl('status passing_test')
+      status.should match(/Health: \d\d/)
+      status.should_not match(/for testing/)
+    end
+
+    it "returns not found if downfile doesn't exist" do
+      _litmusctl('force down test -d').should match("NOT FOUND")
+    end
+  end
+end

data/spec/ipvs_litmus/cli/server_spec.rb

@@ -0,0 +1,16 @@
+require 'spec_helper'
+require 'ipvs_litmus/cli/server'
+
+describe IPVSLitmus::CLI::Server do
+  describe 'parse!' do
+    it 'parses litmus config file options' do
+      options = IPVSLitmus::CLI::Server::Options.new.parse!(['-c', 'foo.conf'])
+      options[:litmus_config].should == 'foo.conf'
+    end
+
+    it 'parses the config dir options' do
+      options = IPVSLitmus::CLI::Server::Options.new.parse!(['-D', '/tmp/foo'])
+      options[:config_dir].should == '/tmp/foo'
+    end
+  end
+end

data/spec/ipvs_litmus/configuration_spec.rb

@@ -0,0 +1,11 @@
+require 'spec_helper'
+
+describe IPVSLitmus::Configuration do
+  describe "evaluate" do
+    it "configures a service" do
+      config = IPVSLitmus::Configuration.new(TEST_CONFIG)
+      services = config.evaluate
+      services.has_key?('test').should == true
+    end
+  end
+end

data/spec/ipvs_litmus/dependency/http_spec.rb

@@ -0,0 +1,42 @@
+require 'spec_helper'
+
+describe IPVSLitmus::Dependency::HTTP do
+  describe "#available?" do
+    it "is true when response is 200" do
+      check = IPVSLitmus::Dependency::HTTP.new('http://httpstat.us/200')
+      check.should be_available
+    end
+
+    it "is true when response is 200 and expected content matches" do
+      check = IPVSLitmus::Dependency::HTTP.new('http://httpstat.us/200', :content => "200 OK")
+      check.should be_available
+    end
+
+    it "is false when response is 200, but does not match content" do
+      check = IPVSLitmus::Dependency::HTTP.new('http://httpstat.us/200', :content => "BAD STUFF")
+      check.should be_available
+    end
+
+    it "is true when response is any 200 level response" do
+      check = IPVSLitmus::Dependency::HTTP.new('http://httpstat.us/201')
+      check.should be_available
+      check = IPVSLitmus::Dependency::HTTP.new('http://httpstat.us/202')
+      check.should be_available
+    end
+
+    it "is false when response is 500 " do
+      check = IPVSLitmus::Dependency::HTTP.new('http://httpstat.us/500')
+      check.should_not be_available
+    end
+
+    it "is false when the response is 404" do
+      check = IPVSLitmus::Dependency::HTTP.new('http://httpstat.us/404')
+      check.should_not be_available
+    end
+
+    it "is false when the dependency is not available" do
+      check = IPVSLitmus::Dependency::HTTP.new('http://127.0.0.1:7777')
+      check.should_not be_available
+    end
+  end
+end

data/spec/ipvs_litmus/health_spec.rb

@@ -0,0 +1,71 @@
+require 'spec_helper'
+
+describe IPVSLitmus::Health do
+  describe "ok?" do
+    it "is true when health is greater than 0" do
+      health = IPVSLitmus::Health.new
+      health.perform(ConstantMetric.new(50))
+      health.should be_ok
+    end
+
+    it "is false when health is 0" do
+      health = IPVSLitmus::Health.new
+      health.perform(ConstantMetric.new(0))
+      health.should_not be_ok
+    end
+  end
+
+  describe "perform" do
+    it "executes the check and adds its value to its health" do
+      health = IPVSLitmus::Health.new
+      health.perform(ConstantMetric.new(50))
+      health.perform(ConstantMetric.new(25))
+      health.value.should == 75
+    end
+  end
+
+  describe "ensure" do
+    it "checks the dependency and modifies the value accordingly" do
+      health = IPVSLitmus::Health.new
+      health.ensure(NeverAvailableDependency.new)
+      health.perform(ConstantMetric.new(50))
+      health.value.should == 0
+    end
+  end
+
+  describe "summary" do
+    it "includes the availablilty of dependencies" do
+      health = IPVSLitmus::Health.new
+      health.ensure(NeverAvailableDependency.new)
+      health.ensure(AlwaysAvailableDependency.new)
+
+      health.summary.should match(/NeverAvailableDependency: FAIL/)
+      health.summary.should match(/AlwaysAvailableDependency: OK/)
+    end
+
+    it "includes the health of individual metrics" do
+      health = IPVSLitmus::Health.new
+      health.perform(ConstantMetric.new(12))
+      health.perform(ConstantMetric.new(34))
+
+      health.summary.should match(/ConstantMetric: 12/)
+      health.summary.should match(/ConstantMetric: 34/)
+    end
+
+    it "only runs each metric once" do
+      health = IPVSLitmus::Health.new
+      metric = ConstantMetric.new(12)
+      metric.should_receive(:current_health).once.and_return(12)
+
+      health.perform(metric)
+    end
+
+    it "only runs each dependency once" do
+      health = IPVSLitmus::Health.new
+      dependency = AlwaysAvailableDependency.new
+      dependency.should_receive(:available?).once.and_return(true)
+
+      health.ensure(dependency)
+    end
+  end
+end

data/spec/ipvs_litmus/metric/available_memory_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+
+describe IPVSLitmus::Metric::AvailableMemory do
+  describe "#current_health" do
+    it "multiplies weight by memory available" do
+      facter = StubFacter.new({"memorytotal" => "10 GB", "memoryfree" => "5 GB"})
+      memory = IPVSLitmus::Metric::AvailableMemory.new(50, facter)
+      memory.current_health.should == 25
+    end
+
+    describe "#memory_total" do
+      it "is a positive integer" do
+        metric = IPVSLitmus::Metric::AvailableMemory.new(50)
+        metric.memory_total.should > 1_000
+      end
+
+      it "is cached" do
+        Facter.should_receive(:value).once.and_return("10 MB")
+        metric = IPVSLitmus::Metric::AvailableMemory.new(50)
+        metric.memory_total
+        metric.memory_total
+        metric.memory_total
+      end
+    end
+
+    describe "#memory_free" do
+      it "is a positive integer" do
+        metric = IPVSLitmus::Metric::AvailableMemory.new(50)
+        metric.memory_free.should > 100
+      end
+    end
+  end
+end

data/spec/ipvs_litmus/metric/cpu_load_spec.rb

@@ -0,0 +1,39 @@
+require 'spec_helper'
+
+describe IPVSLitmus::Metric::CPULoad do
+  describe "#current_health" do
+    it "is the percent of available cpu capacity" do
+      facter = StubFacter.new({"processorcount" => "4", "loadaverage" => "1.00 0.40 0.10"})
+      cpu_load = IPVSLitmus::Metric::CPULoad.new(40, facter)
+      cpu_load.current_health.should == 30
+    end
+
+    it "is zero when the load is above one per core" do
+      facter = StubFacter.new({"processorcount" => "4", "loadaverage" => "20.00 11.40 10.00"})
+      cpu_load = IPVSLitmus::Metric::CPULoad.new(50, facter)
+      cpu_load.current_health.should == 0
+    end
+  end
+
+  describe "#processor_count" do
+    it "is a positive integer" do
+      cpu_load = IPVSLitmus::Metric::CPULoad.new(50)
+      cpu_load.processor_count.should > 0
+    end
+
+    it "is cached" do
+      Facter.should_receive(:value).once.and_return("10")
+      cpu_load = IPVSLitmus::Metric::CPULoad.new(50)
+      cpu_load.processor_count
+      cpu_load.processor_count
+      cpu_load.processor_count
+    end
+  end
+
+  describe "#load_average" do
+    it "is a floating point" do
+      cpu_load = IPVSLitmus::Metric::CPULoad.new(50)
+      cpu_load.load_average.should > 0.0
+    end
+  end
+end

data/spec/ipvs_litmus/service_spec.rb

@@ -0,0 +1,65 @@
+require 'spec_helper'
+
+describe IPVSLitmus::Service do
+  describe "health" do
+    it "is the sum of all the metrics' weight" do
+      service = IPVSLitmus::Service.new('test')
+      service.measure_health ConstantMetric, :weight => 50
+      service.measure_health ConstantMetric, :weight => 25
+
+      service.current_health.value.should == 75
+    end
+
+    it "is 0 when a dependency fails" do
+      service = IPVSLitmus::Service.new('test')
+      service.depends NeverAvailableDependency
+      service.measure_health ConstantMetric, :weight => 50
+
+      service.current_health.value.should == 0
+    end
+
+    it "is 0 when a down file exists" do
+      service = IPVSLitmus::Service.new('test')
+      service.depends AlwaysAvailableDependency
+      service.measure_health ConstantMetric, :weight => 50
+
+      IPVSLitmus::StatusFile.new("down", "test").create("Down for testing")
+
+      service.current_health.value.should == 0
+      service.current_health.summary.should == "Down for testing"
+    end
+
+    it "is 0 when a global down file exists" do
+      service = IPVSLitmus::Service.new('test')
+      service.depends AlwaysAvailableDependency
+      service.measure_health ConstantMetric, :weight => 50
+
+      IPVSLitmus::StatusFile.new("down", "test").create("Down for testing")
+
+      service.current_health.value.should == 0
+      service.current_health.summary.should == "Down for testing"
+    end
+
+    it "is 100 when an up file exists" do
+      service = IPVSLitmus::Service.new('test')
+      service.depends NeverAvailableDependency
+      service.measure_health ConstantMetric, :weight => 50
+
+      IPVSLitmus::StatusFile.new("up", "test").create("Up for testing")
+
+      service.current_health.value.should == 100
+      service.current_health.summary.should == "Up for testing"
+    end
+
+    it "is 100 when a global up file exists" do
+      service = IPVSLitmus::Service.new('test')
+      service.depends NeverAvailableDependency
+      service.measure_health ConstantMetric, :weight => 50
+
+      IPVSLitmus::StatusFile.new("global_up").create("Up for testing")
+
+      service.current_health.value.should == 100
+      service.current_health.summary.should == "Up for testing"
+    end
+  end
+end

data/spec/ipvs_litmus/status_file_spec.rb

@@ -0,0 +1,39 @@
+require 'spec_helper'
+
+describe IPVSLitmus::StatusFile do
+  describe "create" do
+    it "creates a nested file" do
+      status_file = IPVSLitmus::StatusFile.new("foo", "bar")
+      status_file.create("for testing")
+
+      status_file.exists?.should == true
+    end
+
+    it "creates a file" do
+      status_file = IPVSLitmus::StatusFile.new("foo")
+      status_file.create("for testing")
+
+      status_file.exists?.should == true
+    end
+
+    it "writes the content" do
+      status_file = IPVSLitmus::StatusFile.new("foo")
+      status_file.create("for testing")
+
+      status_file.content.should == "for testing"
+    end
+  end
+
+  describe "delete" do
+    it "removes the file" do
+      status_file = IPVSLitmus::StatusFile.new("foo")
+      status_file.create("for testing")
+
+      status_file.exists?.should be_true
+
+      status_file.delete
+
+      status_file.exists?.should be_false
+    end
+  end
+end

data/spec/ipvs_litmus_spec.rb

@@ -0,0 +1,22 @@
+require 'spec_helper'
+
+describe IPVSLitmus do
+  describe 'configure' do
+    it 'populates services from the config file' do
+      IPVSLitmus.configure(TEST_CONFIG)
+      IPVSLitmus.services.has_key?('test').should == true
+    end
+  end
+
+  describe "reload" do
+    it "will reconfigure the services" do
+      IPVSLitmus.configure(TEST_CONFIG)
+      IPVSLitmus.services["bar"] = :service
+
+      Process.kill("HUP", Process.pid)
+
+      IPVSLitmus.services.has_key?('bar').should == false
+      IPVSLitmus.services.has_key?('test').should == true
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,21 @@
+ENV['RACK_ENV'] = 'test'
+
+require 'rspec'
+require 'rack/test'
+require 'ipvs_litmus'
+
+Dir.glob("#{File.expand_path('support', File.dirname(__FILE__))}/**/*.rb").each { |f| require f }
+
+RSpec.configure do |config|
+  config.expect_with :rspec
+  config.include Rack::Test::Methods
+
+  config.before :each do
+    FileUtils.rm_rf(IPVSLitmus.config_dir)
+    IPVSLitmus.reset
+  end
+end
+
+IPVSLitmus.config_dir = "/tmp/ipvs"
+
+TEST_CONFIG = File.expand_path('support/test.config', File.dirname(__FILE__))

data/spec/support/always_available_dependency.rb

@@ -0,0 +1,5 @@
+class AlwaysAvailableDependency
+  def available?
+    true
+  end
+end

data/spec/support/constant_metric.rb

@@ -0,0 +1,9 @@
+class ConstantMetric
+  def initialize(constant)
+    @constant = constant
+  end
+
+  def current_health
+    @constant
+  end
+end

data/spec/support/never_available_dependency.rb

@@ -0,0 +1,5 @@
+class NeverAvailableDependency
+  def available?
+    false
+  end
+end

data/spec/support/stub_facter.rb

@@ -0,0 +1,9 @@
+class StubFacter
+  def initialize(values)
+    @values = values
+  end
+
+  def value(key)
+    @values[key]
+  end
+end