bosh_aws_registry 0.2.1

data/README

@@ -0,0 +1,3 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+BOSH AWS Cloud Provider Interface

data/Rakefile

@@ -0,0 +1,50 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+$:.unshift(File.expand_path("../../rake", __FILE__))
+
+ENV["BUNDLE_GEMFILE"] ||= File.expand_path("../Gemfile", __FILE__)
+
+require "rubygems"
+require "bundler"
+Bundler.setup(:default, :test)
+
+require "rake"
+begin
+  require "rspec/core/rake_task"
+rescue LoadError
+end
+
+require "bundler_task"
+require "ci_task"
+
+gem_helper = Bundler::GemHelper.new(Dir.pwd)
+
+desc "Build AWS Registry gem into the pkg directory"
+task "build" do
+  gem_helper.build_gem
+end
+
+desc "Build and install AWS Registry into system gems"
+task "install" do
+  Rake::Task["bundler:install"].invoke
+  gem_helper.install_gem
+end
+
+BundlerTask.new
+
+if defined?(RSpec)
+  namespace :spec do
+    desc "Run Unit Tests"
+    rspec_task = RSpec::Core::RakeTask.new(:unit) do |t|
+      t.pattern = "spec/unit/**/*_spec.rb"
+      t.rspec_opts = %w(--format progress --colour)
+    end
+
+    CiTask.new do |task|
+      task.rspec_task = rspec_task
+    end
+  end
+
+  desc "Run tests"
+  task :spec => %w(spec:unit)
+end

data/bin/aws_registry

@@ -0,0 +1,39 @@
+#!/usr/bin/env ruby
+
+gemfile = File.expand_path("../../Gemfile", __FILE__)
+
+if File.exists?(gemfile)
+  ENV["BUNDLE_GEMFILE"] = gemfile
+  require "rubygems"
+  require "bundler/setup"
+end
+
+$:.unshift(File.expand_path("../../lib", __FILE__))
+
+require "aws_registry"
+require "optparse"
+
+config_file = nil
+
+opts = OptionParser.new do |opts|
+  opts.on("-c", "--config FILE", "configuration file") do |opt|
+    config_file = opt
+  end
+end
+
+opts.parse!(ARGV.dup)
+
+if config_file.nil?
+  puts opts
+  exit 1
+end
+
+runner = Bosh::AwsRegistry::Runner.new(config_file)
+
+Signal.trap("INT") do
+  runner.stop
+  exit(1)
+end
+
+runner.run
+

data/bin/migrate

@@ -0,0 +1,42 @@
+#!/usr/bin/env ruby
+
+ENV["BUNDLE_GEMFILE"] = File.expand_path("../../Gemfile", __FILE__)
+
+require "rubygems"
+require "bundler/setup"
+require "logger"
+require "sequel"
+
+$:.unshift(File.dirname(__FILE__) + "/../lib")
+
+require "aws_registry"
+
+config_file = nil
+
+opts = OptionParser.new do |opts|
+  opts.on("-c", "--config FILE", "configuration file") do |opt|
+    config_file = opt
+  end
+end
+
+opts.parse!(ARGV.dup)
+
+if config_file.nil?
+  puts opts
+  exit 1
+end
+
+include Bosh::AwsRegistry::YamlHelper
+
+config = load_yaml_file(config_file)
+
+db = Bosh::AwsRegistry.connect_db(config["db"])
+migrations_dir = File.expand_path("../../db/migrations", __FILE__)
+
+options = {
+  :table => "aws_registry_schema"
+}
+
+Sequel.extension :migration
+Sequel::TimestampMigrator.run(db, migrations_dir, options)
+

data/lib/aws_registry/api_controller.rb

@@ -0,0 +1,66 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::AwsRegistry
+
+  class ApiController < Sinatra::Base
+
+    not_found do
+      json(:status => "not_found")
+    end
+
+    error do
+      exception = request.env["sinatra.error"]
+      @logger.error(exception)
+      status(500)
+      json(:status => "error")
+    end
+
+    get "/instances/:instance_id/settings" do
+      ip_check = authorized? ? nil : request.ip
+      settings = @instance_manager.read_settings(params[:instance_id], ip_check)
+      json(:status => "ok", :settings => settings)
+    end
+
+    put "/instances/:instance_id/settings" do
+      protected!
+      @instance_manager.update_settings(params[:instance_id], request.body.read)
+      json(:status => "ok")
+    end
+
+    delete "/instances/:instance_id/settings" do
+      protected!
+      @instance_manager.delete_settings(params[:instance_id])
+      json(:status => "ok")
+    end
+
+    def initialize
+      super
+      @logger = Bosh::AwsRegistry.logger
+
+      @users = Set.new
+      @users << [Bosh::AwsRegistry.http_user, Bosh::AwsRegistry.http_password]
+      @instance_manager = InstanceManager.new
+    end
+
+    def protected!
+      unless authorized?
+        headers("WWW-Authenticate" => 'Basic realm="EC2 Registry"')
+        halt(401, json("status" => "access_denied"))
+      end
+    end
+
+    def authorized?
+      @auth ||= Rack::Auth::Basic::Request.new(request.env)
+      @auth.provided? &&
+        @auth.basic? &&
+        @auth.credentials &&
+        @users.include?(@auth.credentials)
+    end
+
+    def json(payload)
+      Yajl::Encoder.encode(payload)
+    end
+
+  end
+
+end

data/lib/aws_registry/config.rb

@@ -0,0 +1,84 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::AwsRegistry
+
+  class << self
+
+    AWS_MAX_RETRIES = 2
+    AWS_EC2_ENDPOINT = "ec2.amazonaws.com"
+
+    attr_accessor :logger
+    attr_accessor :http_port
+    attr_accessor :http_user
+    attr_accessor :http_password
+    attr_accessor :db
+
+    attr_writer :ec2
+
+    def configure(config)
+      validate_config(config)
+
+      @logger ||= Logger.new(config["logfile"] || STDOUT)
+
+      if config["loglevel"].kind_of?(String)
+        @logger.level = Logger.const_get(config["loglevel"].upcase)
+      end
+
+      @http_port = config["http"]["port"]
+      @http_user = config["http"]["user"]
+      @http_password = config["http"]["password"]
+
+      @aws = config["aws"]
+
+      @aws_options = {
+        :access_key_id => @aws["access_key_id"],
+        :secret_access_key => @aws["secret_access_key"],
+        :max_retries => @aws["max_retries"] || AWS_MAX_RETRIES,
+        :ec2_endpoint => @aws["ec2_endpoint"] || AWS_EC2_ENDPOINT,
+        :logger => @logger
+      }
+
+      @db = connect_db(config["db"])
+    end
+
+    def ec2
+      @ec2 ||= AWS::EC2.new(@aws_options)
+    end
+
+    def connect_db(db_config)
+      connection_options = {
+        :max_connections => db_config["max_connections"],
+        :pool_timeout => db_config["pool_timeout"]
+      }
+
+      db = Sequel.connect(db_config["database"], connection_options)
+      db.logger = @logger
+      db.sql_log_level = :debug
+      db
+    end
+
+    def validate_config(config)
+      unless config.is_a?(Hash)
+        raise ConfigError, "Invalid config format, Hash expected, " \
+                           "#{config.class} given"
+      end
+
+      unless config.has_key?("http") && config["http"].is_a?(Hash)
+        raise ConfigError, "HTTP configuration is missing from " \
+                           "config file"
+      end
+
+      unless config.has_key?("db") && config["db"].is_a?(Hash)
+        raise ConfigError, "Database configuration is missing from " \
+                           "config file"
+      end
+
+      unless config.has_key?("aws") && config["aws"].is_a?(Hash)
+        raise ConfigError, "AWS configuration is missing from " \
+                           "config file"
+      end
+    end
+
+  end
+
+end

data/lib/aws_registry/errors.rb

@@ -0,0 +1,20 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::AwsRegistry
+
+  class Error < StandardError
+    def self.code(code = 500)
+      define_method(:code) { code }
+    end
+  end
+
+  class FatalError < Error; end
+
+  class ConfigError < Error; end
+  class ConnectionError < Error; end
+
+  class AwsError < Error; end
+
+  class InstanceError < Error; end
+  class InstanceNotFound < Error; code(404); end
+end

data/lib/aws_registry/instance_manager.rb

@@ -0,0 +1,73 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::AwsRegistry
+
+  class InstanceManager
+
+    def initialize
+      @logger = Bosh::AwsRegistry.logger
+      @ec2 = Bosh::AwsRegistry.ec2
+    end
+
+    ##
+    # Updates instance settings
+    # @param [String] instance_id EC2 instance id (instance record
+    #        will be created in DB if it doesn't already exist)
+    # @param [String] settings New settings for the instance
+    def update_settings(instance_id, settings)
+      params = {
+        :instance_id => instance_id
+      }
+
+      instance = Models::AwsInstance[params] || Models::AwsInstance.new(params)
+      instance.settings = settings
+      instance.save
+    end
+
+    ##
+    # Reads instance settings
+    # @param [String] instance_id EC2 instance id
+    # @param [optional, String] remote_ip If this IP is provided,
+    #        check will be performed to see if it instance id
+    #        actually has this IP address according to EC2.
+    def read_settings(instance_id, remote_ip = nil)
+      check_instance_ip(remote_ip, instance_id) if remote_ip
+
+      get_instance(instance_id).settings
+    end
+
+    def delete_settings(instance_id)
+      get_instance(instance_id).destroy
+    end
+
+    private
+
+    def check_instance_ip(ip, instance_id)
+      return if ip == "127.0.0.1"
+      actual_ip = instance_private_ip(instance_id)
+      unless ip == actual_ip
+        raise InstanceError, "Instance IP mismatch, expected IP is " \
+                             "`%s', actual IP is `%s'" % [ ip, actual_ip ]
+      end
+    end
+
+    def get_instance(instance_id)
+      instance = Models::AwsInstance[:instance_id => instance_id]
+
+      if instance.nil?
+        raise InstanceNotFound, "Can't find instance `#{instance_id}'"
+      end
+
+      instance
+    end
+
+    def instance_private_ip(instance_id)
+      @ec2.instances[instance_id].private_ip_address
+    rescue AWS::Errors::Base => e
+      raise Bosh::AwsRegistry::AwsError, "AWS error: #{e}"
+    end
+
+  end
+
+end
+

data/lib/aws_registry/models/aws_instance.rb

@@ -0,0 +1,13 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::AwsRegistry::Models
+  class AwsInstance < Sequel::Model
+
+    def validate
+      validates_presence [:instance_id, :settings]
+      validates_unique :instance_id
+    end
+
+  end
+end
+

data/lib/aws_registry/models.rb

@@ -0,0 +1,11 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::AwsRegistry
+  module Models
+  end
+end
+
+require "aws_registry/models/aws_instance"
+
+
+

data/lib/aws_registry/runner.rb

@@ -0,0 +1,56 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::AwsRegistry
+  class Runner
+    include YamlHelper
+
+    def initialize(config_file)
+      Bosh::AwsRegistry.configure(load_yaml_file(config_file))
+
+      @logger = Bosh::AwsRegistry.logger
+      @http_port = Bosh::AwsRegistry.http_port
+      @http_user = Bosh::AwsRegistry.http_user
+      @http_password = Bosh::AwsRegistry.http_password
+    end
+
+    def run
+      @logger.info("BOSH AWS Registry starting...")
+      EM.kqueue if EM.kqueue?
+      EM.epoll if EM.epoll?
+
+      EM.error_handler { |e| handle_em_error(e) }
+
+      EM.run do
+        start_http_server
+      end
+    end
+
+    def stop
+      @logger.info("BOSH AWS Registry shutting down...")
+      @http_server.stop! if @http_server
+      EM.stop
+    end
+
+    def start_http_server
+      @logger.info "HTTP server is starting on port #{@http_port}..."
+      @http_server = Thin::Server.new("0.0.0.0", @http_port, :signals => false) do
+        Thin::Logging.silent = true
+        map "/" do
+          run Bosh::AwsRegistry::ApiController.new
+        end
+      end
+      @http_server.start!
+    end
+
+    private
+
+    def handle_em_error(e)
+      @logger.send(level, e.to_s)
+      if e.respond_to?(:backtrace) && e.backtrace.respond_to?(:join)
+        @logger.send(level, e.backtrace.join("\n"))
+      end
+      stop
+    end
+
+  end
+end

data/lib/aws_registry/version.rb

@@ -0,0 +1,7 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh
+  module AwsRegistry
+    VERSION = "0.2.1"
+  end
+end

data/lib/aws_registry/yaml_helper.rb

@@ -0,0 +1,24 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::AwsRegistry
+  module YamlHelper
+
+    def load_yaml_file(path, expected_type = Hash)
+      unless File.exists?(path)
+        raise(ConfigError, "Cannot find file `#{path}'")
+      end
+
+      yaml = YAML.load_file(path)
+
+      if expected_type && !yaml.is_a?(expected_type)
+        raise ConfigError, "Incorrect file format in `#{path}', " \
+                           "#{expected_type} expected"
+      end
+
+      yaml
+    rescue SystemCallError => e
+      raise ConfigError, "Cannot load YAML file at `#{path}': #{e}"
+    end
+
+  end
+end

data/lib/aws_registry.rb

@@ -0,0 +1,26 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh
+  module AwsRegistry
+    autoload :Models, "aws_registry/models"
+  end
+end
+
+require "aws-sdk"
+require "logger"
+require "sequel"
+require "sinatra/base"
+require "thin"
+require "yajl"
+
+require "aws_registry/yaml_helper"
+
+require "aws_registry/api_controller"
+require "aws_registry/config"
+require "aws_registry/errors"
+require "aws_registry/instance_manager"
+require "aws_registry/runner"
+require "aws_registry/version"
+
+Sequel::Model.plugin :validation_helpers
+

data/spec/assets/sample_config.yml

@@ -0,0 +1,17 @@
+---
+loglevel: debug
+
+http:
+  port: 25695
+  user: admin
+  password: admin
+
+db:
+  database: "sqlite:///:memory:"
+  max_connections: 32
+  pool_timeout: 10
+
+aws:
+  access_key_id: foo
+  secret_access_key: bar
+  max_retries: 2

data/spec/spec_helper.rb

@@ -0,0 +1,113 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+$:.unshift(File.expand_path("../../lib", __FILE__))
+
+ENV["BUNDLE_GEMFILE"] ||= File.expand_path("../../Gemfile", __FILE__)
+
+require "rubygems"
+require "bundler"
+Bundler.setup(:default, :test)
+
+require "fileutils"
+require "logger"
+require "tmpdir"
+
+require "rspec"
+require "rack/test"
+
+module SpecHelper
+  class << self
+    attr_accessor :logger
+    attr_accessor :temp_dir
+
+    def init
+      ENV["RACK_ENV"] = "test"
+      configure_logging
+      configure_temp_dir
+
+      require "aws_registry"
+      init_database
+    end
+
+    def configure_logging
+      if ENV["DEBUG"]
+        @logger = Logger.new(STDOUT)
+      else
+        path = File.expand_path("../spec.log", __FILE__)
+        log_file = File.open(path, "w")
+        log_file.sync = true
+        @logger = Logger.new(log_file)
+      end
+    end
+
+    def configure_temp_dir
+      @temp_dir = Dir.mktmpdir
+      ENV["TMPDIR"] = @temp_dir
+      FileUtils.mkdir_p(@temp_dir)
+      at_exit { FileUtils.rm_rf(@temp_dir) }
+    end
+
+    def init_database
+      @migrations_dir = File.expand_path("../../db/migrations", __FILE__)
+
+      Sequel.extension :migration
+
+      @db = Sequel.sqlite(:database => nil, :max_connections => 32, :pool_timeout => 10)
+      @db.loggers << @logger
+      Bosh::AwsRegistry.db = @db
+
+      run_migrations
+    end
+
+    def run_migrations
+      Sequel::Migrator.apply(@db, @migrations_dir, nil)
+    end
+
+    def reset_database
+      @db.execute("PRAGMA foreign_keys = OFF")
+      @db.tables.each do |table|
+        @db.drop_table(table)
+      end
+      @db.execute("PRAGMA foreign_keys = ON")
+    end
+
+    def reset
+      reset_database
+      run_migrations
+
+      Bosh::AwsRegistry.db = @db
+      Bosh::AwsRegistry.logger = @logger
+    end
+  end
+end
+
+SpecHelper.init
+
+def valid_config
+  {
+    "logfile" => nil,
+    "loglevel" => "debug",
+    "http" => {
+      "user" => "admin",
+      "password" => "admin",
+      "port" => 25777
+    },
+    "db" => {
+      "max_connections" => 433,
+      "pool_timeout" => 227,
+      "database" => "sqlite:///:memory:"
+    },
+    "aws" => {
+      "access_key_id" => "foo",
+      "secret_access_key" => "bar",
+      "max_retries" => 5
+    }
+  }
+end
+
+RSpec.configure do |rspec|
+  rspec.before(:each) do
+    SpecHelper.reset
+    Bosh::AwsRegistry.logger = Logger.new(StringIO.new)
+  end
+end

data/spec/unit/api_controller_spec.rb

@@ -0,0 +1,74 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+require File.expand_path("../../spec_helper", __FILE__)
+
+describe Bosh::AwsRegistry::ApiController do
+
+  before(:each) do
+    Bosh::AwsRegistry.http_user = "admin"
+    Bosh::AwsRegistry.http_password = "admin"
+
+    @instance_manager = mock("instance manager")
+    Bosh::AwsRegistry::InstanceManager.stub!(:new).and_return(@instance_manager)
+
+    rack_mock = Rack::MockSession.new(Bosh::AwsRegistry::ApiController.new)
+    @session = Rack::Test::Session.new(rack_mock)
+  end
+
+  def expect_json_response(response, status, body)
+    response.status.should == status
+    Yajl::Parser.parse(response.body).should == body
+  end
+
+  it "returns settings for given EC2 instance (IP check)" do
+    @instance_manager.should_receive(:read_settings).
+      with("foo", "127.0.0.1").and_return("bar")
+
+    @session.get("/instances/foo/settings")
+
+    expect_json_response(@session.last_response, 200,
+                         { "status" => "ok", "settings" => "bar" })
+  end
+
+  it "returns settings (authorized user, no IP check)" do
+    @instance_manager.should_receive(:read_settings).
+      with("foo", nil).and_return("bar")
+
+    @session.basic_authorize("admin", "admin")
+    @session.get("/instances/foo/settings")
+
+    expect_json_response(@session.last_response, 200,
+                         { "status" => "ok", "settings" => "bar" })
+  end
+
+  it "updates settings" do
+    @session.put("/instances/foo/settings", {}, { :input => "bar" })
+    expect_json_response(@session.last_response, 401,
+                         { "status" => "access_denied" })
+
+    @instance_manager.should_receive(:update_settings).
+      with("foo", "bar").and_return(true)
+
+    @session.basic_authorize("admin", "admin")
+    @session.put("/instances/foo/settings", {}, { :input => "bar" })
+
+    expect_json_response(@session.last_response, 200,
+                         { "status" => "ok" })
+  end
+
+  it "deletes settings" do
+    @session.delete("/instances/foo/settings")
+    expect_json_response(@session.last_response, 401,
+                         { "status" => "access_denied" })
+
+    @instance_manager.should_receive(:delete_settings).
+      with("foo").and_return(true)
+
+    @session.basic_authorize("admin", "admin")
+    @session.delete("/instances/foo/settings")
+
+    expect_json_response(@session.last_response, 200,
+                         { "status" => "ok" })
+  end
+
+end

data/spec/unit/config_spec.rb

@@ -0,0 +1,56 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+require File.expand_path("../../spec_helper", __FILE__)
+
+describe Bosh::AwsRegistry do
+
+  describe "configuring AWS registry" do
+    it "reads provided configuration file and sets singletons" do
+      Bosh::AwsRegistry.configure(valid_config)
+
+      logger = Bosh::AwsRegistry.logger
+
+      logger.should be_kind_of(Logger)
+      logger.level.should == Logger::DEBUG
+
+      Bosh::AwsRegistry.http_port.should == 25777
+      Bosh::AwsRegistry.http_user.should == "admin"
+      Bosh::AwsRegistry.http_password.should == "admin"
+
+      db = Bosh::AwsRegistry.db
+      db.should be_kind_of(Sequel::SQLite::Database)
+      db.opts[:database].should == "/:memory:"
+      db.opts[:max_connections].should == 433
+      db.opts[:pool_timeout].should == 227
+    end
+
+    it "validates configuration file" do
+      expect {
+        Bosh::AwsRegistry.configure("foobar")
+      }.to raise_error(Bosh::AwsRegistry::ConfigError,
+                       /Invalid config format/)
+
+      config = valid_config.merge("http" => nil)
+
+      expect {
+        Bosh::AwsRegistry.configure(config)
+      }.to raise_error(Bosh::AwsRegistry::ConfigError,
+                       /HTTP configuration is missing/)
+
+      config = valid_config.merge("db" => nil)
+
+      expect {
+        Bosh::AwsRegistry.configure(config)
+      }.to raise_error(Bosh::AwsRegistry::ConfigError,
+                       /Database configuration is missing/)
+
+      config = valid_config.merge("aws" => nil)
+
+      expect {
+        Bosh::AwsRegistry.configure(config)
+      }.to raise_error(Bosh::AwsRegistry::ConfigError,
+                       /AWS configuration is missing/)
+    end
+
+  end
+end

data/spec/unit/instance_manager_spec.rb

@@ -0,0 +1,92 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+require File.expand_path("../../spec_helper", __FILE__)
+
+describe Bosh::AwsRegistry::InstanceManager do
+
+  before(:each) do
+    @ec2 = mock("ec2")
+    Bosh::AwsRegistry.ec2 = @ec2
+  end
+
+  let(:manager) do
+    Bosh::AwsRegistry::InstanceManager.new
+  end
+
+  def create_instance(params)
+    Bosh::AwsRegistry::Models::AwsInstance.create(params)
+  end
+
+  def actual_ip_is(ip)
+    instances = mock("instances")
+    instance = mock("instance")
+    @ec2.should_receive(:instances).and_return(instances)
+    instances.should_receive(:[]).with("foo").and_return(instance)
+    instance.should_receive(:private_ip_address).and_return(ip)
+  end
+
+  describe "reading settings" do
+    it "returns settings after verifying IP address" do
+      create_instance(:instance_id => "foo", :settings => "bar")
+      actual_ip_is("10.0.0.1")
+      manager.read_settings("foo", "10.0.0.1").should == "bar"
+    end
+
+    it "raises an error if IP cannot be verified" do
+      create_instance(:instance_id => "foo", :settings => "bar")
+      actual_ip_is("10.0.0.2")
+
+      expect {
+        manager.read_settings("foo", "10.0.0.1")
+      }.to raise_error(Bosh::AwsRegistry::InstanceError,
+                       "Instance IP mismatch, expected IP is `10.0.0.1', " \
+                       "actual IP is `10.0.0.2'")
+    end
+
+    it "doesn't check remote IP if it's not provided" do
+      create_instance(:instance_id => "foo", :settings => "bar")
+      manager.read_settings("foo").should == "bar"
+    end
+
+    it "raises an error if instance not found" do
+      expect {
+        manager.read_settings("foo")
+      }.to raise_error(Bosh::AwsRegistry::InstanceNotFound,
+                       "Can't find instance `foo'")
+    end
+  end
+
+  describe "updating settings" do
+    it "updates settings (new instance)" do
+      manager.update_settings("foo", "baz")
+      manager.read_settings("foo").should == "baz"
+    end
+
+    it "updates settings (existing instance)" do
+      create_instance(:instance_id => "foo", :settings => "bar")
+      manager.read_settings("foo").should == "bar"
+      manager.update_settings("foo", "baz")
+      manager.read_settings("foo").should == "baz"
+    end
+  end
+
+  describe "deleting settings" do
+    it "deletes settings" do
+      manager.update_settings("foo", "baz")
+      manager.delete_settings("foo")
+
+      expect {
+        manager.read_settings("foo")
+      }.to raise_error(Bosh::AwsRegistry::InstanceNotFound,
+                       "Can't find instance `foo'")
+    end
+
+    it "raises an error if instance not found" do
+      expect {
+        manager.delete_settings("foo")
+      }.to raise_error(Bosh::AwsRegistry::InstanceNotFound,
+                       "Can't find instance `foo'")
+    end
+  end
+
+end

data/spec/unit/runner_spec.rb

@@ -0,0 +1,84 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+require File.expand_path("../../spec_helper", __FILE__)
+
+describe Bosh::AwsRegistry::Runner do
+
+  def make_runner(config_file)
+    Bosh::AwsRegistry::Runner.new(config_file)
+  end
+
+  def write_config(file, config)
+    File.open(file, "w") do |f|
+      YAML.dump(config, f)
+    end
+  end
+
+  describe "initializing" do
+    it "configures AWS registry using provided file" do
+      config_file = Tempfile.new("config")
+      config = { "key" => "value" }
+      write_config(config_file.path, config)
+
+      Bosh::AwsRegistry.should_receive(:configure).with(config)
+      make_runner(config_file.path)
+    end
+
+    it "fails if config file not found" do
+      expect {
+        make_runner("foo")
+      }.to raise_error(Bosh::AwsRegistry::ConfigError, "Cannot find file `foo'")
+    end
+
+    it "fails when config file has incorrect format" do
+      config_file = Tempfile.new("config")
+      config = "foo"
+      write_config(config_file.path, config)
+
+      expect {
+        make_runner(config_file.path)
+      }.to raise_error(Bosh::AwsRegistry::ConfigError, /Incorrect file format/)
+    end
+
+    it "fails when some syscall fails" do
+      config_file = Tempfile.new("config")
+      write_config(config_file.path, { "foo" => "bar" })
+
+      YAML.stub!(:load_file).and_raise(SystemCallError.new("baz"))
+
+      expect {
+        make_runner(config_file.path)
+      }.to raise_error(Bosh::AwsRegistry::ConfigError, /baz/)
+    end
+  end
+
+  describe "running/stopping" do
+    before(:each) do
+      @config_file = Tempfile.new("config")
+      write_config(@config_file.path, valid_config)
+    end
+
+    it "spins up/shuts down reactor and HTTP server" do
+      runner = make_runner(@config_file)
+      mock_thin = mock("thin")
+
+      EM.should_receive(:run).and_yield
+
+      Thin::Server.should_receive(:new).
+        with("0.0.0.0", 25777, :signals => false).
+        and_return(mock_thin)
+
+      mock_thin.should_receive(:start!)
+
+      runner.run
+
+      mock_thin.should_receive(:stop!)
+      EM.should_receive(:stop)
+
+      runner.stop
+    end
+  end
+
+end
+
+

metadata

@@ -0,0 +1,157 @@
+--- !ruby/object:Gem::Specification
+name: bosh_aws_registry
+version: !ruby/object:Gem::Version
+  version: 0.2.1
+  prerelease: 
+platform: ruby
+authors:
+- VMware
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-07-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sequel
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: sinatra
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: thin
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: yajl-ruby
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.3.6
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.3.6
+description: BOSH AWS registry
+email: support@vmware.com
+executables:
+- aws_registry
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/aws_registry
+- bin/migrate
+- lib/aws_registry.rb
+- lib/aws_registry/api_controller.rb
+- lib/aws_registry/config.rb
+- lib/aws_registry/errors.rb
+- lib/aws_registry/instance_manager.rb
+- lib/aws_registry/models.rb
+- lib/aws_registry/models/aws_instance.rb
+- lib/aws_registry/runner.rb
+- lib/aws_registry/version.rb
+- lib/aws_registry/yaml_helper.rb
+- README
+- Rakefile
+- spec/assets/sample_config.yml
+- spec/spec_helper.rb
+- spec/unit/api_controller_spec.rb
+- spec/unit/config_spec.rb
+- spec/unit/instance_manager_spec.rb
+- spec/unit/runner_spec.rb
+homepage: http://www.vmware.com
+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'
+      segments:
+      - 0
+      hash: -3082160961986360018
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -3082160961986360018
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: BOSH AWS registry
+test_files:
+- spec/assets/sample_config.yml
+- spec/spec_helper.rb
+- spec/unit/api_controller_spec.rb
+- spec/unit/config_spec.rb
+- spec/unit/instance_manager_spec.rb
+- spec/unit/runner_spec.rb