big_brother 0.1.0

data/spec/spec_helper.rb

@@ -0,0 +1,57 @@
+ENV['RACK_ENV'] = 'test'
+
+require 'rspec'
+require 'rack/test'
+require 'big_brother'
+require "socket"
+
+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.around(:each) do |spec|
+    ipvs = BigBrother.ipvs
+    @recording_executor = RecordingExecutor.new
+    BigBrother.ipvs = BigBrother::IPVS.new(@recording_executor)
+    spec.run
+    BigBrother.ipvs = ipvs
+  end
+
+  config.before(:each) do
+    BigBrother.clusters.replace({})
+    FileUtils.rm_rf(BigBrother.config_dir)
+    BigBrother.logger = NullLogger.new
+  end
+end
+
+def run_in_reactor
+  around(:each) do |spec|
+    EM.synchrony do
+      spec.run
+      EM.stop
+    end
+  end
+end
+
+def with_litmus_server(ip, port, health)
+  around(:each) do |spec|
+    server = StubServer.new(<<-HTTP, 0.25, port, ip)
+HTTP/1.0 200 OK
+Connection: close
+
+Health: #{health}
+HTTP
+    spec.run
+    server.stop
+  end
+end
+
+def public_ip_address
+  local_ip = UDPSocket.open {|s| s.connect("64.233.187.99", 1); s.addr.last}
+end
+
+BigBrother.config_dir = "/tmp/big_brother"
+
+TEST_CONFIG = File.expand_path('support/example_config.yml', File.dirname(__FILE__))

data/spec/support/example_config.yml

@@ -0,0 +1,34 @@
+---
+test1:
+  checkInterval: 1
+  scheduler: wrr
+  fwmark: 1
+  nodes:
+  - address: 127.0.0.1
+    port: 9001
+    path: /test/valid
+  - address: 127.0.0.1
+    port: 9002
+    path: /test/valid
+test2:
+  checkInterval: 1
+  scheduler: wrr
+  fwmark: 2
+  nodes:
+  - address: 127.0.0.1
+    port: 9001
+    path: /test/invalid
+  - address: 127.0.0.1
+    port: 9002
+    path: /test/invalid
+test3:
+  checkInterval: 1
+  scheduler: wrr
+  fwmark: 3
+  nodes:
+  - address: 127.0.0.1
+    port: 9001
+    path: /test/valid
+  - address: 127.0.0.1
+    port: 9002
+    path: /test/invalid

data/spec/support/factories/cluster_factory.rb

@@ -0,0 +1,13 @@
+class Factory
+  def self.cluster(overrides = {})
+    BigBrother::Cluster.new(
+      overrides.fetch(:name, 'test'),
+      {
+        'fwmark' => overrides.fetch(:fwmark, 100),
+        'scheduler' => overrides.fetch(:scheduler, 'wrr'),
+        'check_interval' => overrides.fetch(:check_interval, 1),
+        'nodes' => overrides.fetch(:nodes, [])
+      }
+    )
+  end
+end

data/spec/support/factories/node_factory.rb

@@ -0,0 +1,9 @@
+class Factory
+  def self.node(overrides = {})
+    BigBrother::Node.new(
+      overrides.fetch(:address, 'localhost'),
+      overrides.fetch(:port, 8081),
+      overrides.fetch(:path, '/test/status')
+    )
+  end
+end

data/spec/support/ipvsadm

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+echo $@ >> /tmp/ipvsadm.log

data/spec/support/mock_session.rb

@@ -0,0 +1,14 @@
+# inlined from sinatra-synchrony
+require 'rack/test'
+
+module Rack
+  class MockSession
+    alias_method :request_original, :request
+    def request(uri, env)
+      EM.synchrony do
+        request_original uri, env
+        EM.stop
+      end
+    end
+  end
+end

data/spec/support/null_logger.rb

@@ -0,0 +1,7 @@
+class NullLogger
+  def write(msg)
+  end
+
+  def info(msg)
+  end
+end

data/spec/support/playback_executor.rb

@@ -0,0 +1,13 @@
+class PlaybackExecutor
+  def initialize
+    @responses = []
+  end
+
+  def invoke(command)
+    @responses.pop
+  end
+
+  def add_response(output, status)
+    @responses.push [output, status]
+  end
+end

data/spec/support/recording_executor.rb

@@ -0,0 +1,11 @@
+class RecordingExecutor
+  attr_reader :commands
+
+  def initialize
+    @commands = []
+  end
+
+  def invoke(command)
+    @commands << command
+  end
+end

data/spec/support/stub_server.rb

@@ -0,0 +1,22 @@
+class StubServer
+  module Server
+    attr_accessor :response, :delay
+    def receive_data(data)
+      EM.add_timer(@delay) {
+        send_data @response
+        close_connection_after_writing
+      }
+    end
+  end
+
+  def initialize(response, delay = 0, port = 8081, host = "127.0.0.1")
+    @sig = EventMachine::start_server(host, port, Server) { |s|
+      s.response = response
+      s.delay = delay
+    }
+  end
+
+  def stop
+    EventMachine.stop_server @sig
+  end
+end

metadata

@@ -0,0 +1,271 @@
+--- !ruby/object:Gem::Specification
+name: big_brother
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Braintree
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-05-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thin
+  requirement: &70321466649500 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.3.1
+  type: :runtime
+  prerelease: false
+  version_requirements: *70321466649500
+- !ruby/object:Gem::Dependency
+  name: async-rack
+  requirement: &70321466648640 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.5.1
+  type: :runtime
+  prerelease: false
+  version_requirements: *70321466648640
+- !ruby/object:Gem::Dependency
+  name: sinatra
+  requirement: &70321466648140 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70321466648140
+- !ruby/object:Gem::Dependency
+  name: rack-fiber_pool
+  requirement: &70321466647520 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70321466647520
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: &70321466646780 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>'
+      - !ruby/object:Gem::Version
+        version: 1.0.0.beta.1
+    - - <
+      - !ruby/object:Gem::Version
+        version: 1.0.0.beta.100
+  type: :runtime
+  prerelease: false
+  version_requirements: *70321466646780
+- !ruby/object:Gem::Dependency
+  name: em-http-request
+  requirement: &70321466645620 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70321466645620
+- !ruby/object:Gem::Dependency
+  name: em-synchrony
+  requirement: &70321466644700 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70321466644700
+- !ruby/object:Gem::Dependency
+  name: em-resolv-replace
+  requirement: &70321466643840 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70321466643840
+- !ruby/object:Gem::Dependency
+  name: em-syslog
+  requirement: &70321466642720 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.0.2
+  type: :runtime
+  prerelease: false
+  version_requirements: *70321466642720
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &70321466641880 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.9.0
+  type: :development
+  prerelease: false
+  version_requirements: *70321466641880
+- !ruby/object:Gem::Dependency
+  name: rack-test
+  requirement: &70321466641380 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.6.1
+  type: :development
+  prerelease: false
+  version_requirements: *70321466641380
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: &70321466640920 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70321466640920
+- !ruby/object:Gem::Dependency
+  name: rake_commit
+  requirement: &70321466640060 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.13'
+  type: :development
+  prerelease: false
+  version_requirements: *70321466640060
+- !ruby/object:Gem::Dependency
+  name: vagrant
+  requirement: &70321466639400 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70321466639400
+description: IPVS backend supervisor
+email:
+- code@getbraintree.com
+executables:
+- bigbro
+- ocf_big_brother
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rake_commit
+- .rvmrc
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- big_brother.gemspec
+- bin/bigbro
+- bin/ocf_big_brother
+- config.ru
+- lib/big_brother.rb
+- lib/big_brother/app.rb
+- lib/big_brother/cli.rb
+- lib/big_brother/cluster.rb
+- lib/big_brother/configuration.rb
+- lib/big_brother/ipvs.rb
+- lib/big_brother/logger.rb
+- lib/big_brother/node.rb
+- lib/big_brother/shell_executor.rb
+- lib/big_brother/status_file.rb
+- lib/big_brother/ticker.rb
+- lib/big_brother/version.rb
+- lib/sinatra/synchrony.rb
+- lib/thin/backends/tcp_server_with_callbacks.rb
+- lib/thin/callback_rack_handler.rb
+- lib/thin/callbacks.rb
+- spec/big_brother/app_spec.rb
+- spec/big_brother/cluster_spec.rb
+- spec/big_brother/configuration_spec.rb
+- spec/big_brother/ipvs_spec.rb
+- spec/big_brother/node_spec.rb
+- spec/big_brother/shell_executor_spec.rb
+- spec/big_brother/status_file_spec.rb
+- spec/big_brother/ticker_spec.rb
+- spec/big_brother/version_spec.rb
+- spec/big_brother_spec.rb
+- spec/spec_helper.rb
+- spec/support/example_config.yml
+- spec/support/factories/cluster_factory.rb
+- spec/support/factories/node_factory.rb
+- spec/support/ipvsadm
+- spec/support/mock_session.rb
+- spec/support/null_logger.rb
+- spec/support/playback_executor.rb
+- spec/support/recording_executor.rb
+- spec/support/stub_server.rb
+homepage: https://github.com/braintree/big_brother
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: Process to monitor and update weights for servers in an IPVS pool
+test_files:
+- spec/big_brother/app_spec.rb
+- spec/big_brother/cluster_spec.rb
+- spec/big_brother/configuration_spec.rb
+- spec/big_brother/ipvs_spec.rb
+- spec/big_brother/node_spec.rb
+- spec/big_brother/shell_executor_spec.rb
+- spec/big_brother/status_file_spec.rb
+- spec/big_brother/ticker_spec.rb
+- spec/big_brother/version_spec.rb
+- spec/big_brother_spec.rb
+- spec/spec_helper.rb
+- spec/support/example_config.yml
+- spec/support/factories/cluster_factory.rb
+- spec/support/factories/node_factory.rb
+- spec/support/ipvsadm
+- spec/support/mock_session.rb
+- spec/support/null_logger.rb
+- spec/support/playback_executor.rb
+- spec/support/recording_executor.rb
+- spec/support/stub_server.rb