rouster 0.5

data/test/functional/test_restart.rb

@@ -0,0 +1,44 @@
+  require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+  require 'rouster'
+  require 'rouster/puppet'
+  require 'test/unit'
+
+  class TestRestart < Test::Unit::TestCase
+
+    def setup
+      assert_nothing_raised do
+        @app = Rouster.new(:name => 'app', :verbosity => 4)
+      end
+    end
+
+    def test_happy_path
+      @app.up()
+
+      assert_equal(true, @app.is_available_via_ssh?)
+      sleep 10
+      original_uptime = @app.facter()['uptime_seconds'].to_i
+
+      assert_nothing_raised do
+        @app.restart()
+      end
+
+      count = 0
+      until @app.is_available_via_ssh?()
+        count += 1
+        break if count > 12 # wait up to 2 minutes
+        sleep 10
+      end
+
+      new_uptime = @app.facter(false)['uptime_seconds'].to_i
+
+      assert_not_equal(original_uptime, new_uptime)
+      assert(original_uptime > new_uptime)
+
+    end
+
+    def teardown
+      # noop
+    end
+
+  end

data/test/functional/test_run.rb

@@ -0,0 +1,77 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'test/unit'
+
+class TestRun < Test::Unit::TestCase
+
+  def setup
+    @app         = Rouster.new({:name => 'app', :verbose => 4})
+    @app_no_sudo = Rouster.new({:name => 'app', :verbose => 4, :sudo => false})
+
+    @app.up()
+    @app_no_sudo.up()
+
+    assert_equal(@app.is_available_via_ssh?, true, 'app is available via SSH')
+    assert_equal(@app.is_available_via_ssh?, true, 'app_no_sudo is available via SSH')
+  end
+
+  def test_happy_path
+
+    assert_nothing_raised do
+      @app.run('ls -l')
+    end
+
+    assert_equal(0, @app.exitcode, 'got expected exit code')
+    assert_not_nil(@app.get_output(), 'output is populated')
+    assert_match(/^total\s\d/, @app.get_output(), 'output matches expectations')
+  end
+
+  def test_bad_exit_codes
+
+    assert_raise Rouster::RemoteExecutionError do
+      @app.run('fizzbang')
+    end
+
+    assert_not_equal(0, @app.exitcode, 'got expected non-0 exit code')
+    assert_not_nil(@app.get_output(), 'output is populated')
+    assert_match(/fizzbang/, @app.get_output(), 'output matches expectations')
+  end
+
+  def test_bad_custom_exit_codes
+
+    assert_raise Rouster::RemoteExecutionError do
+      @app.run('ls -l', 2)
+    end
+
+    assert_equal(0, @app.exitcode)
+    assert_not_nil(@app.get_output())
+    assert_match(/total/, @app.get_output())
+  end
+
+  def test_sudo_enabled
+
+    assert_nothing_raised do
+      @app.run('ls -l /root')
+    end
+
+    assert_equal(0, @app.exitcode, 'got expected exit code')
+    assert_no_match(/Permission denied/i, @app.get_output(), 'output matches expectations 1of2')
+    assert_match(/^total\s\d/, @app.get_output(), 'output matches expectations 2of2')
+  end
+
+  def test_sudo_disabled
+
+    assert_raise Rouster::RemoteExecutionError do
+      @app_no_sudo.run('ls -l /root')
+    end
+
+    assert_not_equal(0, @app_no_sudo.exitcode, 'got expected non-0 exit code')
+    assert_match(/Permission denied/i, @app_no_sudo.get_output(), 'output matches expectations')
+  end
+
+  def teardown
+    @app.suspend()
+    @app_no_sudo.suspend()
+  end
+end

data/test/functional/test_status.rb

@@ -0,0 +1,38 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'test/unit'
+
+class TestStatus < Test::Unit::TestCase
+
+  def setup
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app')
+    end
+
+  end
+
+  def test_1_destroyed
+    @app.destroy()
+
+    assert_equal('not created', @app.status())
+  end
+
+  def test_2_upped
+    @app.up()
+
+    assert_equal('running', @app.status())
+  end
+
+  def test_3_suspended
+    @app.up unless @app.status.eql?('running')
+    @app.suspend()
+
+    assert_equal('saved', @app.status())
+  end
+
+  def teardown
+    # noop
+  end
+
+end

data/test/functional/test_suspend.rb

@@ -0,0 +1,31 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'test/unit'
+
+class TestSuspend < Test::Unit::TestCase
+
+  def setup
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app')
+    end
+
+  end
+
+  def test_happy_path
+    @app.up()
+
+    assert_equal(true, @app.is_available_via_ssh?)
+
+    assert_nothing_raised do
+      @app.suspend()
+    end
+
+    assert_equal(false, @app.is_available_via_ssh?)
+  end
+
+  def teardown
+    # noop
+  end
+
+end

data/test/functional/test_up.rb

@@ -0,0 +1,27 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'test/unit'
+
+class TestUp < Test::Unit::TestCase
+
+  def setup
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app')
+    end
+
+  end
+
+  def test_happy_path
+    assert_nothing_raised do
+      @app.up()
+    end
+
+    assert_equal(true, @app.is_available_via_ssh?)
+  end
+
+  def teardown
+    # noop
+  end
+
+end

data/test/functional/test_validate_file.rb

@@ -0,0 +1,30 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/puppet'
+require 'rouster/testing'
+require 'test/unit'
+
+class TestValidateFileFunctional < Test::Unit::TestCase
+
+  def setup
+    # expose private methods
+    Rouster.send(:public, *Rouster.private_instance_methods)
+    Rouster.send(:public, *Rouster.protected_instance_methods)
+
+    @app = Rouster.new(:name => 'app')
+  end
+
+  def test_negative_functional_fallthrough
+
+    assert_equal(false, @app.validate_file('/foo', {}, false, true))
+    assert_equal(false, @app.validate_file('/fizzy', { :ensure => 'directory' }, false, true))
+    assert_equal(false, @app.validate_file('/bang', { :ensure => 'file' }, false, true))
+
+  end
+
+  def teardown
+    # noop
+  end
+
+end

data/test/puppet/manifests/default.pp

@@ -0,0 +1,9 @@
+## common entry-point into Puppet config.
+
+if ($::virtual == 'virtualbox') {
+    $data_center = 'vagrant'
+} else {
+    $data_center = 'amazon'
+}
+
+include role::ui

data/test/puppet/manifests/hiera.yaml

@@ -0,0 +1,12 @@
+---
+:hierarchy:
+    - %{::data_center}
+    - common
+
+:backends:
+    - json
+
+:json:
+    # @todo replace with absolute path for non-Vagrant use
+    # alternatively set a top-level variable in default.pp and do %{::var_name}
+    :datadir: 'hieradata'

data/test/puppet/manifests/hieradata/common.json

@@ -0,0 +1,3 @@
+{
+    "role::ui::public_hostname": "%{::ipaddress_eth1}"
+}

data/test/puppet/manifests/hieradata/vagrant.json

@@ -0,0 +1,3 @@
+{
+    "role::ui::public_hostname": "%{::ipaddress_eth1}-vagrant"
+}

data/test/puppet/manifests/manifest.pp

@@ -0,0 +1,78 @@
+# == manifest.pp
+
+node default {
+  include baseclass
+}
+node 'app.hsd1.ca.comcast.net' {
+  include app_role
+}
+
+node 'db' {
+  include db_role
+}
+
+notify { 'test':
+  message => 'this is a test',
+}
+
+class baseclass {
+
+  file { '/etc/passwd':
+    ensure => file,
+    owner  => 'root',
+    group  => 'root',
+    mode   => '0644',
+  }
+
+  file { '/tmp/foo':
+    ensure => directory,
+    owner  => 'root',
+    group  => 'root',
+  }
+
+  package { ['puppet', 'facter'] :
+    ensure => installed,
+  }
+
+}
+
+class app_role {
+
+  package { 'rsync':
+    ensure => installed,
+  }
+
+  user { 'foo':
+    ensure  => present,
+    groups  => 'bar',
+  }
+
+  group { 'bar':
+    ensure => present,
+    before => User['foo'],
+  }
+
+  service { 'snmpd':
+    ensure => stopped,
+  }
+
+}
+
+class db_role {
+
+  package { 'httpd':
+    ensure => installed,
+  }
+
+  file { '/tmp/fizzy':
+    ensure   => file,
+    content  => 'this is a test',
+    owner    => 'vagrant',
+    group    => 'vagrant',
+    mode     => '0444',
+  }
+
+  service { 'httpd':
+    ensure => running,
+  }
+}

data/test/puppet/modules/role/manifests/ui.pp

@@ -0,0 +1,5 @@
+class role::ui {
+    notify {'role::ui::notify':
+        message => "role::ui configured for ${::data_center}",
+    }
+}

data/test/puppet/test_apply.rb

@@ -0,0 +1,149 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/puppet'
+require 'rouster/testing'
+
+require 'test/unit'
+
+class TestPuppetApply < Test::Unit::TestCase
+
+  def setup
+    @app = Rouster.new(:name => 'app', :verbose => 1)
+
+    ## TODO teach put() how to use -R (directories)
+    required_files = {
+      'test/puppet/manifests/default.pp'             => 'manifests/default.pp',
+      'test/puppet/manifests/hiera.yaml'             => 'manifests/hiera.yaml',
+      'test/puppet/manifests/manifest.pp'            => 'manifests/manifest.pp',
+      'test/puppet/manifests/hieradata/common.json'  => 'manifests/hieradata/common.json',
+      'test/puppet/manifests/hieradata/vagrant.json' => 'manifests/hieradata/vagrant.json',
+      'test/puppet/modules/role/manifests/ui.pp'     => 'modules/role/manifests/ui.pp',
+    }
+
+    ## TODO figure out a better pattern here -- scp tunnel is under 'vagrant' context, but dirs created with 'root'
+    @app.sudo = false
+    @app.run('mkdir -p manifests/hieradata')
+    @app.run('mkdir -p modules/role/manifests')
+    @app.sudo = true
+
+    required_files.each_pair do |source,dest|
+      @app.put(source, dest)
+    end
+
+    assert_nothing_raised do
+      @app.run_puppet('masterless', {
+        :expected_exitcode => [0,2],
+        :hiera_config      => 'manifests/hiera.yaml',
+        :manifest_file     => 'manifests/manifest.pp',
+        :module_dir        => 'modules/'
+      })
+    end
+
+    assert_match(/this is a test/, @app.get_output())
+    assert_match(/Finished catalog run in/, @app.get_output())
+
+    # define base here
+    @expected_packages = {
+      'puppet' => { :ensure => true },
+      'facter' => { :ensure => 'present' }
+    }
+
+    @expected_files = {
+      '/etc/passwd' => {
+        :contains => [ 'vagrant', 'root'],
+        :ensure   => 'file',
+        :group    => 'root',
+        :mode     => '0644',
+        :owner    => 'root'
+      },
+
+      '/tmp' => {
+        :ensure => 'directory',
+        :group  => 'root',
+        :owner  => 'root',
+      }
+    }
+
+    @expected_groups   = {
+      'root' => { :ensure => 'true' }
+    }
+
+    @expected_services = Hash.new()
+    @expected_users    = {
+      'root' => {
+        :ensure => 'present',
+        :group  => 'root',
+      }
+    }
+
+  end
+
+  def test_app
+    app_expected_packages = {
+      'rsync'    => { :ensure => 'present' }
+    }.merge(@expected_packages)
+
+    app_expected_files = {
+      '/etc/hosts' => {
+        :contains => [ 'localhost', 'app' ],
+        :ensure   => 'present',
+        :group    => 'root',
+        :owner    => 'root',
+      },
+    }.merge(@expected_files)
+
+    app_expected_groups   = {
+      'vagrant' => {
+        :ensure => 'present',
+      }
+    }.merge(@expected_groups)
+
+    app_expected_services = {}.merge(@expected_services)
+
+    app_expected_users    = {
+      'vagrant' => {
+        :ensure => 'present',
+      },
+    }.merge(@expected_users)
+
+    assert_nothing_raised do
+      @app.run_puppet('masterless', {
+        :expected_exitcode => [0,2],
+        :hiera_config      => 'manifests/hiera.yaml',
+        :manifest_dir      => 'manifests/',
+        :module_dir        => 'modules/'
+      })
+
+    end
+
+    assert_match(/Finished catalog run in/, @app.get_output()) # this only examines the last manifest run
+
+    # manually specified testing
+    app_expected_files.each_pair do |f,e|
+      assert_equal(true, @app.validate_file(f,e), "file[#{f}] expectation[#{e}]")
+    end
+
+    app_expected_groups.each_pair do |g,e|
+      assert_equal(true, @app.validate_group(g,e), "group[#{g}] expectation[#{e}]")
+    end
+
+    app_expected_packages.each_pair do |p,e|
+      assert_equal(true, @app.validate_package(p, e), "package[#{p}] expectation[#{e}]")
+    end
+
+    app_expected_services.each_pair do |s,e|
+      assert_equal(true, @app.validate_service(s,e), "service[#{s}] expectation[#{e}]")
+    end
+
+    app_expected_users.each_pair do |u,e|
+      assert_equal(true, @app.validate_user(u,e), "user[#{u}] expectation[#{e}]")
+    end
+
+  end
+
+  def teardown
+    # noop
+  end
+
+end