machines 0.5.1

data/spec/lib/packages/webapps_spec.rb

@@ -0,0 +1,70 @@
+require 'spec_helper'
+
+describe 'packages/webapps' do
+  before(:each) do
+    load_package('webapps')
+
+    $conf.from_hash(:user => {:name => 'username', :home => 'home_dir'}, :ruby => {:version => 'ruby_version'})
+    $conf.webapps = {'application' =>
+      AppBuilder.new(
+        'scm' => 'github.com/project',
+        'branch' => 'master',
+        'name' => 'application',
+        'url' => 'github url',
+        'path' => '/home/users/application',
+        'db_password' => 'pa55',
+        'server_name' => 'app.dev'
+      )
+    }
+    $conf.from_hash(:awstats => {:path => 'stats_path'})
+    $conf.from_hash(:webserver => {:name => 'nginx', :path => 'nginx_path', :servers_dir => 'servers'})
+    $conf.from_hash(:db_server => {:address => 'db_master'})
+    FileUtils.mkdir_p 'nginx'
+    File.open('nginx/app_server.conf.erb', 'w') {|f| f.puts 'the template' }
+  end
+
+  it 'adds the following commands' do
+    $conf.environment = 'production'
+    eval_package
+    $conf.commands.map(&:info).join("\n").must_equal [
+      "TASK   webapps - Sets up Web apps in config/webapps.yml using app_server.conf.erb. Copies SSL certs.",
+      "SUDO   mkdir -p nginx_path/servers",
+      "RUN    mkdir -p /home/users/application/releases",
+      "RUN    mkdir -p /home/users/application/shared/config",
+      "RUN    mkdir -p /home/users/application/shared/system",
+      "RUN    mkdir -p /home/users/application/shared/log",
+      "UPLOAD buffer from nginx/app_server.conf.erb to /tmp/application.conf",
+      "SUDO   cp -rf /tmp/application.conf nginx_path/servers/application.conf",
+      "RUN    rm -rf /tmp/application.conf",
+      "SUDO   mkdir -p /var/log/nginx",
+      'SUDO   grep "127.0.0.1 app.dev" /etc/hosts || echo "127.0.0.1 app.dev" >> /etc/hosts'
+    ].join("\n")
+  end
+
+  it 'creates a database.yml when requested' do
+    $conf.environment = 'production'
+    $conf.webapps['application'].write_yml = true
+    eval_package
+    $conf.commands.map(&:info).join("\n").must_match 'UPLOAD buffer from database.yml to /home/users/application/shared/config/database.yml'
+  end
+
+  it "doesn't make app structure when target is a development machine" do
+    $conf.from_hash(:ruby => {:gems_path => '.rbenv'})
+    $conf.environment = 'development'
+    eval_package
+    commandline = $conf.commands.map(&:info).join("\n")
+    commandline.must_equal [
+      "TASK   webapps - Sets up Web apps in config/webapps.yml using app_server.conf.erb. Copies SSL certs.",
+      "SUDO   mkdir -p nginx_path/servers",
+      "RUN    git clone --quiet --branch master github.com/project /home/users/application",
+      "RUN    cd /home/users/application && $HOME/.rbenv/bin/rbenv exec bundle",
+      "RUN    cd /home/users/application && $HOME/.rbenv/bin/rbenv exec bundle --binstubs=.bin",
+      "UPLOAD buffer from nginx/app_server.conf.erb to /tmp/application.conf",
+      "SUDO   cp -rf /tmp/application.conf nginx_path/servers/application.conf",
+      "RUN    rm -rf /tmp/application.conf",
+      "SUDO   mkdir -p /var/log/nginx",
+      'SUDO   grep "127.0.0.1 app.dev" /etc/hosts || echo "127.0.0.1 app.dev" >> /etc/hosts'
+    ].join("\n")
+  end
+end
+

data/spec/spec_helper.rb

@@ -0,0 +1,103 @@
+$LOAD_PATH << 'lib'
+
+require 'bundler/setup'
+require 'minitest/spec'
+require 'mocha'
+require 'stringio'
+require 'fakefs/safe'
+
+Dir['spec/support/*.rb'].each {|file| require File.join('./spec', 'support', File.basename(file)) }
+require 'machines'
+application_dir = $conf.application_dir
+
+include Machines::AppSettings
+
+# This is done so that Machines::Core.run doesn't collide with MiniTest::Unit::TestCase.run when included
+module Machines::Core
+  alias :run_command :run
+  remove_method :run
+end
+MiniTest::Spec.add_setup_hook do |klass|
+  klass.instance_eval do
+    alias :run :run_command if respond_to?(:run_command)
+  end
+end
+
+MiniTest::Spec.add_setup_hook do
+  include Machines::Checks
+
+  $conf.clear
+  $conf.passwords = []
+  $conf.commands = []
+  $conf.tasks = {}
+  $conf.application_dir = application_dir
+
+  $input = StringIO.new
+  $output = StringIO.new
+  $terminal = HighLine.new($input, $output)
+
+  $debug = FakeOut.new
+  $file = FakeOut.new
+  $console = FakeOut.new
+  Machines::Command.debug = Machines::Logger.new $debug
+  Machines::Command.file = Machines::Logger.new $file
+  Machines::Command.console = Machines::Logger.new $console, :truncate => true
+
+  FakeFS.activate!
+  FileUtils.mkdir_p 'tmp'
+end
+
+MiniTest::Spec.add_teardown_hook do
+  FakeFS.deactivate!
+  FakeFS::FileSystem.clear
+end
+
+module MiniTest
+  module Assertions
+    def capture_io
+      yield
+      return $output.string
+    end
+  end
+end
+
+class MiniTest::Spec::Package < MiniTest::Spec
+  include Machines
+  include AppSettings, CloudMachine, Commandline, Configuration, Core, Database
+  include FileOperations, Installation, Machinesfile, Questions, Services
+end
+
+MiniTest::Spec.register_spec_type(/packages\//, MiniTest::Spec::Package)
+
+def RealFS &block
+  FakeFS.deactivate!
+  yield block
+ensure
+  FakeFS.activate!
+end
+
+def load_package name
+  RealFS do
+    @package_path = File.join($conf.application_dir, File.join('packages', "#{name}.rb"))
+    @package = File.read(@package_path)
+  end
+end
+
+def eval_package
+  eval @package, nil, @package_path
+end
+
+def save_yaml contents, path
+  File.open(path, 'w') do |f|
+    YAML.dump(contents, f)
+  end
+end
+
+def colored string, color
+  ending = string.scan(/(\n|\r)$/).flatten.first
+  string.sub!(/(\n|\r)$/, '')
+  $terminal.color(string, color.to_sym) + ending.to_s
+end
+
+require 'minitest/autorun'
+

data/spec/support/coverage.rb

@@ -0,0 +1,8 @@
+if ENV['COVERAGE']
+  require 'simplecov'
+  SimpleCov.start do
+    add_filter '/spec/'
+    add_filter '/bin/'
+  end
+end
+

data/spec/support/fake_out.rb

@@ -0,0 +1,22 @@
+class FakeOut
+  def initialize
+    @buffer = []
+  end
+
+  def inspect
+    @buffer.join("\n")
+  end
+
+  def print message
+    @buffer << message
+  end
+
+  def next
+    @buffer.delete_at(0)
+  end
+
+
+  def flush
+  end
+end
+

data/spec/support/fakefs_additions.rb

@@ -0,0 +1,10 @@
+module FakeFS
+  class File
+    class Stat
+      def mode
+        File.directory?(@file) ? 0755 : 0644
+      end
+    end
+  end
+end
+

data/spec/support/minitest.rb

@@ -0,0 +1,69 @@
+# Standard minitest diff calls the OS's diff command. FakeFS breaks diff because
+# diff is looking for the files on disk but the files are written by Tempfile
+# while FakeFS is still active. The following modules provide a pure ruby diff
+# and override MiniTest::Assertions::diff to utilitize it.
+
+module Diff
+  attr_accessor :diffs, :sequences
+
+  def self.diffs(old, new)
+    new, old, @diffs = new.lines.to_a, old.lines.to_a, []
+    build_sequences(old, new)
+    find_diffs(old, new, old.size, new.size)
+  end
+
+  private
+
+  def self.find_diffs(old, new, row, col)
+    if row > 0 && col > 0 && old[row-1] == new[col-1]
+      find_diffs(old, new, row-1, col-1)
+      @diffs << { :line => row, :change => :same, :string => old[row-1].strip }
+    elsif col > 0 && (row == 0 || @sequences[row][col-1] >= @sequences[row-1][col])
+      find_diffs(old, new, row, col-1)
+      @diffs << { :line => col, :change => :add, :string => new[col-1].strip }
+    elsif row > 0 && (col == 0 || @sequences[row][col-1] < @sequences[row-1][col])
+      find_diffs(old, new, row-1, col)
+      @diffs << { :line => row, :change => :delete, :string => old[row-1].strip }
+    end
+  end
+
+  def self.build_sequences(old, new)
+    rows, cols = old.size+1, new.size+1
+    @sequences = rows.times.map{|x|[0]*(x+1)}
+    rows.times.each do |row|
+      cols.times.each do |col|
+        @sequences[row][col] = if old[row-1] == new[col-1]
+          @sequences[row-1][col-1] + 1
+        else
+          [@sequences[row][col-1].to_i, @sequences[row-1][col].to_i].max
+        end
+      end
+    end
+  end
+end
+
+module MiniTest
+  module Assertions
+    SYMBOLS = {:add => '+', :delete => '-', :same => ' '}
+
+    def diff exp, act
+      expect = mu_pp_for_diff exp
+      butwas = mu_pp_for_diff act
+      result = nil
+
+      need_to_diff =
+        (expect.include?("\n")    ||
+         butwas.include?("\n")    ||
+         expect.size > 30         ||
+         butwas.size > 30         ||
+         expect == butwas)
+
+      return "Expected: #{mu_pp exp}\n  Actual: #{mu_pp act}" unless need_to_diff
+
+      Diff.diffs(expect, butwas).map do |diff|
+        SYMBOLS[diff[:change]] + diff[:string]
+      end.join("\n")
+    end
+  end
+end
+

data/spec/support/vm_control.rb

@@ -0,0 +1,54 @@
+require 'net/ssh'
+
+class VmControl
+  def initialize
+    @name = 'machinesvm'
+  end
+
+  def start
+    vboxmanage "startvm #{@name} --type headless"
+  end
+
+  def stop
+    vboxmanage "controlvm #{@name} savestate"
+  end
+
+  def kill
+    return unless state =~ /running/
+    vboxmanage "controlvm #{@name} poweroff"
+  end
+
+  def restore
+    vboxmanage "snapshot #{@name} restorecurrent"
+  end
+
+  def state
+    string = "showvminfo #{@name}"
+    output = vboxmanage string
+    output.scan(/State:\s+(.*)/).first.first
+  end
+
+  def vboxmanage command
+    send File.read('.vmcontrol').gsub("\n", ''), command
+  rescue Errno::ENOENT
+    raise '.vmcontrol does not exist. Add `local` to test using VMs on a Linux host or add `remote` to test on a VM running on Windows.'
+  end
+
+private
+  def local command
+    output = `VBoxManage #{command}`
+  end
+
+  def remote command
+    output = ''
+    begin
+      Net::SSH.start('winhost', 'vbuser', :timeout => 10) do |ssh|
+        output = ssh.exec!("C:\\Program Files\\Oracle\\VirtualBox\\VBoxManage #{command}")
+      end
+    rescue Timeout::Error => e
+      raise 'Connection timed out. Did you start your SSH server on the host?'
+    end
+    output
+  end
+end
+