engineyard-cloud-client 0.1.2

data/lib/engineyard-cloud-client/ruby_ext.rb

@@ -0,0 +1,9 @@
+class Object
+  unless respond_to?(:tap)
+    # Ruby 1.9 has it, 1.8 doesn't
+    def tap
+      yield self
+      self
+    end
+  end
+end

data/lib/engineyard-cloud-client/test.rb

@@ -0,0 +1,31 @@
+require 'engineyard-cloud-client'
+
+module EY::CloudClient::Test
+end
+
+begin
+  require 'dm-core'
+  require 'dm-migrations'
+  require 'dm-aggregates'
+  require 'dm-sqlite-adapter'
+  require 'ey_resolver'
+  require 'rabl'
+rescue LoadError
+  raise LoadError, <<-ERROR
+engineyard-cloud-client needs the following gems to run in test mode:
+
+group 'engineyard-cloud-client-test' do
+  gem 'dm-core'
+  gem 'dm-migrations'
+  gem 'dm-aggregates'
+  gem 'dm-sqlite-adapter'
+  gem 'ey_resolver', '~>0.2.1'
+  gem 'rabl'
+end
+
+Please add the above to your Gemfile.
+  ERROR
+end
+
+require 'engineyard-cloud-client/test/ui'
+require 'engineyard-cloud-client/test/scenario'

data/lib/engineyard-cloud-client/test/fake_awsm.rb

@@ -0,0 +1,22 @@
+require 'engineyard-cloud-client/test'
+require 'engineyard-cloud-client/test'
+require 'realweb'
+
+module EY::CloudClient::Test
+  module FakeAwsm
+    def self.uri
+      @uri ||= load_server
+    end
+
+    protected
+
+    def self.load_server
+      config_ru = File.expand_path("fake_awsm/config.ru", File.dirname(__FILE__))
+      unless system("ruby -c '#{config_ru}' > /dev/null")
+        raise SyntaxError, "There is a syntax error in fake_awsm/config.ru! FIX IT!"
+      end
+      @server = RealWeb.start_server_in_fork(config_ru)
+      @server.base_uri.to_s
+    end
+  end
+end

data/lib/engineyard-cloud-client/test/fake_awsm/config.ru

@@ -0,0 +1,207 @@
+require 'rubygems'
+require 'sinatra/base'
+require 'json'
+require 'rabl'
+require 'gitable'
+require 'ey_resolver'
+require File.expand_path('../scenarios', __FILE__)
+require File.expand_path('../models', __FILE__)
+
+Rabl.register!
+
+class FakeAwsm < Sinatra::Base
+  disable :show_exceptions
+  enable :raise_errors
+  set :views, File.expand_path('../views', __FILE__)
+
+  SCENARIOS = [
+    Scenario::Base.new,
+    Scenario::AppWithoutEnv.new,
+    Scenario::UnlinkedApp.new,
+    Scenario::TwoApps.new,
+    Scenario::LinkedApp.new,
+    Scenario::MultipleAmbiguousAccounts.new,
+    Scenario::LinkedAppNotRunning.new,
+    Scenario::LinkedAppRedMaster.new,
+    Scenario::OneAppManyEnvs.new,
+    Scenario::OneAppManySimilarlyNamedEnvs.new,
+    Scenario::TwoAppsSameGitUri.new,
+  ]
+
+  def initialize(*)
+    super
+    @user = Scenario::Base.new.user
+  end
+
+  before do
+    content_type "application/json"
+    token = request.env['HTTP_X_EY_CLOUD_TOKEN']
+    if token
+      @user = User.first(:api_token => token)
+    end
+  end
+
+  get "/" do
+    content_type :html
+    "OMG"
+  end
+
+  get "/scenario" do
+    new_scenario = SCENARIOS.detect { |scen| scen.user.name == params[:scenario] }
+    unless new_scenario
+      status(404)
+      return {"ok" => "false", "message" => "wtf is the #{params[:scenario]} scenario?"}.to_json
+    end
+    user = new_scenario.user
+    {
+      "scenario" => {
+        "email"     => user.email,
+        "password"  => user.password,
+        "api_token" => user.api_token,
+      }
+    }.to_json
+  end
+
+  get "/scenarios" do
+    scenarios = SCENARIOS.map do |scen|
+      user = scen.user
+      {
+        :name      => user.name,
+        :email     => user.email,
+        :password  => user.password,
+        :api_token => user.api_token,
+      }
+    end
+    {'scenarios' => scenarios}.to_json
+  end
+
+  get "/api/v2/current_user" do
+    render :rabl, :user, :format => "json"
+  end
+
+  get "/api/v2/accounts" do
+    @accounts = @user.accounts
+    render :rabl, :accounts, :format => "json"
+  end
+
+  get "/api/v2/apps" do
+    raise('No user agent header') unless env['HTTP_USER_AGENT'] =~ %r#^EngineYardCloudClient/#
+    @apps = @user.accounts.apps
+    render :rabl, :apps, :format => "json"
+  end
+
+  get "/api/v2/environments" do
+    @environments = @user.accounts.environments
+    render :rabl, :environments, :format => "json"
+  end
+
+  get "/api/v2/environments/resolve" do
+    @resolver = EY::Resolver.environment_resolver(@user, params['constraints'])
+    render :rabl, :resolve_environments, :format => "json"
+  end
+
+  get "/api/v2/app_environments/resolve" do
+    @resolver = EY::Resolver.app_env_resolver(@user, params['constraints'])
+    render :rabl, :resolve_app_environments, :format => "json"
+  end
+
+  get "/api/v2/environments/:env_id/instances" do
+    environment = @user.accounts.environments.get(params['env_id'])
+    @instances = environment.instances
+    render :rabl, :instances, :format => "json"
+  end
+
+  get "/api/v2/environments/:env_id/logs" do
+    {
+      "logs" => [
+        {
+          "id" => params['env_id'].to_i,
+          "role" => "app_master",
+          "main" => "MAIN LOG OUTPUT",
+          "custom" => "CUSTOM LOG OUTPUT"
+        }
+      ]
+    }.to_json
+  end
+
+  get "/api/v2/environments/:env_id/recipes" do
+    redirect '/fakes3/recipe'
+  end
+
+  get "/fakes3/recipe" do
+    content_type "binary/octet-stream"
+    status(200)
+
+    tempdir = File.join(Dir.tmpdir, "ey_test_cmds_#{Time.now.tv_sec}#{Time.now.tv_usec}_#{$$}")
+    Dir.mkdir(tempdir)
+    Dir.mkdir("#{tempdir}/cookbooks")
+    File.open("#{tempdir}/cookbooks/README", 'w') do |f|
+      f.write "Remove this file to clone an upstream git repository of cookbooks\n"
+    end
+
+    Dir.chdir(tempdir) { `tar czf - cookbooks` }
+  end
+
+  post "/api/v2/environments/:env_id/recipes" do
+    if params[:file][:tempfile]
+      files = `tar --list -z -f "#{params[:file][:tempfile].path}"`.split(/\n/)
+      if files.empty?
+        status(400)
+        "No files in uploaded tarball"
+      else
+        status(204)
+        ""
+      end
+    else
+      status(400)
+      "Recipe file not uploaded"
+    end
+  end
+
+  put "/api/v2/environments/:env_id/update_instances" do
+    status(202)
+    ""
+  end
+
+  put "/api/v2/environments/:env_id/run_custom_recipes" do
+    status(202)
+    ""
+  end
+
+  get "/api/v2/apps/:app_id/environments/:environment_id/deployments/last" do
+    {
+      "deployment" => {
+        "id" => 3,
+        "ref" => "HEAD",
+        "resolved_ref" => "HEAD",
+        "commit" => 'a'*40,
+        "user_name" => "User Name",
+        "migrate_command" => "rake db:migrate --trace",
+        "created_at" => Time.now.utc - 3600,
+        "finished_at" => Time.now.utc - 3400,
+        "successful" => true,
+      }
+    }.to_json
+  end
+
+  post "/api/v2/apps/:app_id/environments/:environment_id/deployments" do
+    {"deployment" => params[:deployment].merge({"id" => 2, "commit" => 'a'*40, "resolved_ref" => "resolved-#{params[:deployment][:ref]}"})}.to_json
+  end
+
+  put "/api/v2/apps/:app_id/environments/:environment_id/deployments/:deployment_id/finished" do
+    {"deployment" => params[:deployment].merge({"id" => 2, "finished_at" => Time.now})}.to_json
+  end
+
+  post "/api/v2/authenticate" do
+    user = User.first(:email => params[:email], :password => params[:password])
+    if user
+      {"api_token" => user.api_token, "ok" => true}.to_json
+    else
+      status(401)
+      {"ok" => false}.to_json
+    end
+  end
+
+end
+
+run FakeAwsm.new

data/lib/engineyard-cloud-client/test/fake_awsm/models.rb

@@ -0,0 +1,9 @@
+Dir[File.expand_path('../models/*.rb', __FILE__)].each do |model|
+  require model
+end
+
+require 'dm-migrations'
+require 'dm-aggregates'
+DataMapper.setup(:default, "sqlite::memory:")
+DataMapper.finalize
+DataMapper.auto_migrate!

data/lib/engineyard-cloud-client/test/fake_awsm/models/account.rb

@@ -0,0 +1,13 @@
+require 'dm-core'
+
+class Account
+  include DataMapper::Resource
+
+  property :id, Serial
+  property :name, String
+
+  belongs_to :user
+  has n, :environments
+  has n, :apps
+
+end

data/lib/engineyard-cloud-client/test/fake_awsm/models/app.rb

@@ -0,0 +1,24 @@
+require 'gitable'
+require 'dm-core'
+
+class App
+  include DataMapper::Resource
+
+  property :id, Serial
+  property :name, String
+  property :repository_uri, String
+  property :app_type_id, String
+
+  belongs_to :account
+  has n, :app_environments
+  has n, :environments, :through => :app_environments
+
+  def gitable_uri
+    Gitable::URI.parse(repository_uri)
+  end
+
+  def inspect
+    "#<App name:#{name} account:#{account.name}>"
+  end
+
+end

data/lib/engineyard-cloud-client/test/fake_awsm/models/app_environment.rb

@@ -0,0 +1,19 @@
+require 'dm-core'
+
+class AppEnvironment
+  include DataMapper::Resource
+
+  property :id, Serial
+  property :app_id, Integer
+  property :environment_id, Integer
+
+  belongs_to :app
+  belongs_to :environment
+
+  has n, :deployments
+
+  def inspect
+    "#<AppEnvironment app:#{app.name} env:#{environment.name}>"
+  end
+
+end

data/lib/engineyard-cloud-client/test/fake_awsm/models/deployments.rb

@@ -0,0 +1,15 @@
+require 'dm-core'
+
+class Deployment
+  include DataMapper::Resource
+
+  property :id, Serial
+  property :app_environment_id, Integer
+
+  belongs_to :app_environment
+
+  def inspect
+    "#<Deployment app_environment:#{app_environment.inspect}>"
+  end
+
+end

data/lib/engineyard-cloud-client/test/fake_awsm/models/environment.rb

@@ -0,0 +1,25 @@
+require 'dm-core'
+
+class Environment
+  include DataMapper::Resource
+
+  property :id, Serial
+  property :name, String
+  property :ssh_username, String
+  property :app_server_stack_name, String
+  property :load_balancer_ip_address, String
+  property :framework_env, String
+
+  belongs_to :account
+  has n, :app_environments
+  has n, :apps, :through => :app_environments
+  has n, :instances
+
+  def inspect
+    "#<Environment name:#{name} account:#{account.name}>"
+  end
+
+  def instances_count
+    instances.size
+  end
+end

data/lib/engineyard-cloud-client/test/fake_awsm/models/instance.rb

@@ -0,0 +1,23 @@
+require 'dm-core'
+
+class Instance
+  include DataMapper::Resource
+
+  property :id,              Serial
+  property :name,            String
+  property :role,            String
+  property :status,          String
+  property :amazon_id,       String
+  property :public_hostname, String
+
+  belongs_to :environment
+
+  def inspect
+    "#<Instance environment:#{environment.name} role:#{role} name:#{name}>"
+  end
+
+  def bridge
+    %w[app_master solo].include?(role)
+  end
+
+end

data/lib/engineyard-cloud-client/test/fake_awsm/models/user.rb

@@ -0,0 +1,15 @@
+require 'dm-core'
+
+class User
+  include DataMapper::Resource
+
+  property :id, Serial
+  property :name, String
+  property :email, String, :default => 'test@test.test'
+  property :password, String, :default => 'test'
+  property :api_token, String,
+    :default => lambda { |r, p| File.open('/dev/urandom', 'r') { |fh| fh.read(16).unpack('H*').first }}
+
+  has n, :accounts
+
+end

data/lib/engineyard-cloud-client/test/fake_awsm/scenarios.rb

@@ -0,0 +1,325 @@
+require 'engineyard-cloud-client/test/fake_awsm'
+
+module Scenario
+  class Base
+    attr_accessor :git_remote
+    attr_reader :user
+
+    def initialize(name = "User Name", email = "test@test.test", pass = "test")
+      @git_remote = "user@git.host:path/to/repo.git"
+      @user = User.create(:name => name, :email => email, :password => pass)
+      @account = @user.accounts.create({"name" => "main"})
+    end
+  end
+
+  class LinkedApp < Base
+    def initialize(name = 'Linked App', email = 'linked.app@test.local', pass = 'linked')
+      super
+      @app = @account.apps.create("name" => "rails232app", "repository_uri" => git_remote)
+      @env = @account.environments.create({
+        "name" => "giblets",
+        "ssh_username" => "turkey",
+        "app_server_stack_name" => "nginx_mongrel",
+        "load_balancer_ip_address" => '127.0.0.0',
+        "framework_env" => "production"
+      })
+
+      _instances.each do |inst|
+        @env.instances.create(inst)
+      end
+      @app_env = @app.app_environments.create(:environment => @env)
+    end
+
+    private
+
+    def _instances
+      [{
+        "role" => "app_master",
+        "name" => nil,
+        "status" => "running",
+        "amazon_id" => 'i-ddbbdd92',
+        "public_hostname" => "app_master_hostname.compute-1.amazonaws.com",
+      }, {
+        "name" => nil,
+        "role" => "db_master",
+        "status" => "running",
+        "amazon_id" => "i-d4cdddbf",
+        "public_hostname" => "db_master_hostname.compute-1.amazonaws.com",
+      }, {
+        "name" => nil,
+        "role" => "db_slave",
+        "status" => "running",
+        "amazon_id" => "i-asdfasdfaj",
+        "public_hostname" => "db_slave_1_hostname.compute-1.amazonaws.com",
+      }, {
+        "name" => nil,
+        "role" => "db_slave",
+        "status" => "running",
+        "amazon_id" => "i-asdfasdfaj",
+        "public_hostname" => "db_slave_2_hostname.compute-1.amazonaws.com",
+      }, {
+        "role" => "app",
+        "name" => nil,
+        "status" => "building",
+        "amazon_id" => "i-d2e3f1b9",
+        "public_hostname" => "app_hostname.compute-1.amazonaws.com",
+      }, {
+        "role" => "util",
+        "name" => "fluffy",
+        "status" => "running",
+        "amazon_id" => "i-80e3f1eb",
+        "public_hostname" => "util_fluffy_hostname.compute-1.amazonaws.com",
+      }, {
+        "role" => "util",
+        "name" => "rocky",
+        "status" => "running",
+        "amazon_id" => "i-80etf1eb",
+        "public_hostname" => "util_rocky_hostname.compute-1.amazonaws.com",
+      }]
+    end
+  end  # LinkedApp
+
+  class MultipleAmbiguousAccounts < LinkedApp
+    def initialize(name = 'Multiple Ambiguous Accounts', email = 'multiple.ambiguous.accounts@test.local', pass = 'multi')
+      super
+      @account2 = @user.accounts.create("name" => "account_2")
+      @app2 = @account2.apps.create("name" => "rails232app", "repository_uri" => git_remote)
+      @env2 = @account2.environments.create({
+        "name" => "giblets",
+        "ssh_username" => "turkey",
+        "app_server_stack_name" => "nginx_mongrel",
+        "load_balancer_ip_address" => '127.0.0.0',
+        "framework_env" => "production"
+      })
+
+      _instances.each do |inst|
+        @env2.instances.create(inst)
+      end
+      @app_env2 = @app2.app_environments.create(:environment => @env2)
+    end
+  end
+
+  class AppWithoutEnv < Base
+    def initialize(name = 'App Without Env', email = 'app.without.env@test.local', pass = 'without')
+      super
+
+      @app = @account.apps.create({
+        "name" => "rails232app",
+        "repository_uri" => git_remote
+      })
+    end
+  end # AppWithoutEnv
+
+
+  class UnlinkedApp < Base
+    def initialize(name = 'Unlinked App', email = 'unlinked.app@test.local', pass = 'unlinked')
+      super
+
+      @app = @account.apps.create({
+        "name" => "rails232app",
+        "repository_uri" => git_remote
+      })
+
+      @other = @account.environments.create({
+        "name" => "chickenwings",
+        "ssh_username" => "ham",
+        "app_server_stack_name" => "nginx_mongrel",
+        "load_balancer_ip_address" => '127.0.0.0',
+        "framework_env" => "production"
+      })
+      @app_env = @app.app_environments.create(:environment => @other)
+
+      @env = @account.environments.create({
+        "name" => "giblets",
+        "ssh_username" => "turkey",
+        "app_server_stack_name" => "nginx_mongrel",
+        "load_balancer_ip_address" => '127.0.0.0',
+        "framework_env" => "production"
+      })
+
+      @env.instances.create({
+        "status" => "running",
+        "amazon_id" => 'i-ddbbdd92',
+        "role" => "solo",
+        "public_hostname" => "ec2-174-129-198-124.compute-1.amazonaws.com"
+      })
+    end
+  end # UnlinkedApp
+
+  class LinkedAppNotRunning < Base
+    def initialize(name = 'Linked App Not Running', email = 'linked.app.not.running@test.local', pass = 'linked.stopped')
+      super
+      @app = @account.apps.create({
+        "name" => "rails232app",
+        "repository_uri" => git_remote
+      })
+
+      @env = @account.environments.create({
+        "ssh_username" => "turkey",
+        "instances" => [],
+        "name" => "giblets",
+        "app_server_stack_name" => "nginx_mongrel",
+        "load_balancer_ip_address" => '127.0.0.0',
+        "framework_env" => "production"
+      })
+
+      @app.app_environments.create(:environment => @env)
+    end
+  end # LinkedAppNotRunning
+
+  class LinkedAppRedMaster < LinkedApp
+    def initialize(name = 'Linked App Red Master', email = 'linked.app.red.master@test.local', pass = 'linked.red')
+      super
+      @env.instances.first.update(:status => "error")
+    end
+  end
+
+  class OneAppManyEnvs < Base
+    def initialize(name = 'One App Many Envs', email = 'one.app.many.envs@test.local', pass = '1app2cups')
+      super
+      @app = @account.apps.create({
+        "name" => "rails232app",
+        "repository_uri" => git_remote
+      })
+
+      @env1 = @account.environments.create({
+        "ssh_username" => "turkey",
+        "name" => "giblets",
+        "app_server_stack_name" => "nginx_mongrel",
+        "load_balancer_ip_address" => '127.0.0.0',
+        "framework_env" => "production",
+      })
+      @env1.app_environments.create(:app => @app)
+
+      @env1.instances.create({
+        "status" => "running",
+        "amazon_id" => 'i-ddbbdd92',
+        "role" => "solo",
+        "public_hostname" => "app_master_hostname.compute-1.amazonaws.com"
+      })
+      @env2 = @account.environments.create({
+        "ssh_username" => "ham",
+        "instances" => [],
+        "name" => "bakon",
+        "app_server_stack_name" => "nginx_passenger",
+        "load_balancer_ip_address" => '127.0.0.0',
+      })
+      @env2.app_environments.create(:app => @app)
+
+      @env3 = @account.environments.create({
+        "ssh_username" => "hamburger",
+        "instances" => [],
+        "name" => "beef",
+        "app_server_stack_name" => "nginx_passenger",
+        "load_balancer_ip_address" => '127.0.0.0',
+      })
+    end
+  end # OneAppTwoEnvs
+
+  class TwoApps < Base
+    def initialize(name = 'Two Apps', email = 'two.apps@test.local', pass = '2apps')
+      super
+      @env1 = @account.environments.create({
+          "name" => "giblets",
+          "framework_env" => "staging",
+          "ssh_username" => "turkey",
+          "app_server_stack_name" => "nginx_unicorn",
+          "load_balancer_ip_address" => '127.0.0.0',
+      })
+      @app1 = @account.apps.create({
+        "name" => "rails232app",
+        "repository_uri" => "git://github.com/smerritt/rails232app.git"
+      })
+      @env1.app_environments.create(:app => @app1)
+      @env1.instances.create({
+        "status" => "running",
+        "name" => nil,
+        "role" => "solo",
+        "public_hostname" => "ec2-174-129-7-113.compute-1.amazonaws.com",
+        "amazon_id" => "i-0911f063",
+      })
+
+      @env2 = @account.environments.create({
+        "framework_env" => "production",
+        "name" => "keycollector_production",
+        "ssh_username" => "deploy",
+        "load_balancer_ip_address" => '127.0.0.0',
+        "app_server_stack_name" => "nginx_mongrel",
+      })
+      @app2 = @account.apps.create({
+        "name" => "keycollector",
+        "repository_uri" => "git@github.com:smerritt/keycollector.git",
+      })
+      @env2.app_environments.create(:app => @app2)
+      @env2.instances.create({
+        "status" => "running",
+        "name" => nil,
+        "role" => "solo",
+        "public_hostname" => "app_master_hostname.compute-1.amazonaws.com",
+        "amazon_id" => "i-051195b9",
+      })
+    end
+  end # TwoApps
+
+  class TwoAppsSameGitUri < TwoApps
+    def initialize(name = 'Two Apps Same Git URI', email = 'two.apps.same.git.uri@test.local', pass = '2apps1repo')
+      super
+      @app1.update(:repository_uri => "git://github.com/engineyard/dup.git")
+      @app2.update(:repository_uri => "git://github.com/engineyard/dup.git")
+    end
+  end # TwoAppsSameGitUri
+
+  class OneAppManySimilarlyNamedEnvs < Base
+    def initialize(name = 'One App Similarly Named Envs', email = 'one.app.similarly.named.envs@test.local', pass = '1apptwinrepos')
+      super
+      @app = @account.apps.create({
+        "name" => "rails232app",
+        "repository_uri" => git_remote
+      })
+
+      @env1 = @account.environments.create({
+        "ssh_username" => "turkey",
+        "name" => "railsapp_production",
+        "load_balancer_ip_address" => '127.0.0.0',
+        "app_server_stack_name" => "nginx_mongrel",
+        "framework_env" => "production",
+      })
+      @env1.instances.create({
+        "status" => "running",
+        "amazon_id" => 'i-ddbbdd92',
+        "role" => "solo",
+        "public_hostname" => "app_master_hostname.compute-1.amazonaws.com"
+      })
+
+      @env2 = @account.environments.create({
+        "ssh_username" => "ham",
+        "name" => "railsapp_staging",
+        "load_balancer_ip_address" => '127.3.2.1',
+        "app_server_stack_name" => "nginx_passenger",
+        "framework_env" => "production",
+      })
+
+      @env2.instances.create({
+        "public_hostname" => '127.3.2.1',
+        "status" => "running",
+        "role" => "solo",
+      })
+      @env3 = @account.environments.create({
+        "ssh_username" => "ham",
+        "name" => "railsapp_staging_2",
+        "app_server_stack_name" => "nginx_passenger",
+        "load_balancer_ip_address" => '127.0.0.2',
+        "framework_env" => "production",
+      })
+      @env3.instances.create({
+        "status" => "running",
+        "role" => "solo",
+        "public_hostname" => "ec2-174-129-198-124.compute-1.amazonaws.com",
+      })
+
+      @app.app_environments.create(:environment => @env1)
+      @app.app_environments.create(:environment => @env2)
+      @app.app_environments.create(:environment => @env3)
+    end
+  end  # OneAppManySimilarlyNamedEnvs
+end