rouster 0.5

data/test/puppet/test_roles.rb

@@ -0,0 +1,186 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/puppet'
+require 'rouster/testing'
+
+require 'test/unit'
+
+class TestPuppetRoles < Test::Unit::TestCase
+
+  def setup
+
+    piab_vagrantfile = sprintf('%s/../../../piab/Vagrantfile', File.dirname(File.expand_path(__FILE__)))
+
+    unless File.file?(piab_vagrantfile)
+      skip(sprintf('missing SFDC specific Vagrantfile[%s], skipping', piab_vagrantfile))
+    end
+
+    assert_nothing_raised do
+      @ppm = Rouster.new(:name => 'ppm', :vagrantfile => piab_vagrantfile)
+      @ppm.up()
+      @ppm.remove_existing_certs('ppm')
+      #@ppm.rebuild() unless @ppm.status.eql?('running') # destroy / rebuild
+    end
+
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app', :vagrantfile => piab_vagrantfile)
+    end
+
+    assert_nothing_raised do
+		  @ppm.run_puppet('master', { :expected_exitcode => [0,2] })
+    end
+
+    assert_match(/Finished catalog run in/, @ppm.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',
+        }
+    }
+
+    # manually specified testing
+    @expected_files.each_pair do |f,e|
+      assert_equal(true, @ppm.validate_file(f,e))
+    end
+
+    @expected_groups.each_pair do |g,e|
+      assert_equal(true, @ppm.validate_group(g,e))
+    end
+
+    @expected_packages.each_pair do |p,e|
+      assert_equal(true, @ppm.validate_package(p, e))
+    end
+
+    @expected_services.each_pair do |s,e|
+      assert_equal(true, @ppm.validate_service(s,e))
+    end
+
+    @expected_users.each_pair do |u,e|
+      assert_equal(true, @ppm.validate_user(u,e))
+    end
+
+  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.up()
+      @app.run_puppet('master', { :expected_exitcode => [0, 2] })
+    end
+
+    assert_match(/Finished catalog run in/, @app.get_output())
+
+    # manually specified testing
+    app_expected_files.each_pair do |f,e|
+      assert_equal(true, @app.validate_file(f,e))
+    end
+
+    app_expected_groups.each_pair do |g,e|
+       assert_equal(true, @app.validate_group(g,e))
+    end
+
+    app_expected_packages.each_pair do |p,e|
+      assert_equal(true, @app.validate_package(p, e))
+    end
+
+    app_expected_services.each_pair do |s,e|
+      assert_equal(true, @app.validate_service(s,e))
+    end
+
+    app_expected_users.each_pair do |u,e|
+      assert_equal(true, @app.validate_user(u,e))
+    end
+
+  end
+
+  def dont_test_app_automated
+    catalog      = @app.get_catalog()
+    expectations = @app.parse_catalog(catalog)
+
+    assert_nothing_raised do
+      @app.up()
+      @app.run_puppet('master', { :expected_exitcode => 2 })
+    end
+
+    assert_match(/Finished catalog run in/, @app.get_output())
+
+    expectations.each_pair do |k,v|
+      res = nil
+      case v[:type]
+        when :dir, :file
+          res = @app.validate_file(k, v)
+        when :group
+          res = @app.validate_group(k, v)
+        when :package
+          res = @app.validate_package(k, v)
+        when :user
+          res = @app.validate_user(k, v)
+        when :service
+          res = @app.validate_service(k, v)
+      end
+
+      assert_equal(true, res, sprintf('failed[%s]: %s',v, res))
+    end
+
+  end
+
+  def teardown
+    # noop
+  end
+
+end

data/test/tunnel_vs_scp.rb

@@ -0,0 +1,41 @@
+require sprintf('%s/../%s', File.dirname(File.expand_path(__FILE__)), 'path_helper')
+require 'rouster'
+require 'rouster/tests'
+
+# 'performance' test to determine if we should implement the scp option
+# in is_in_file()? -- first run says yes, probably make it off by default
+
+w = Rouster.new(:name => 'app', :verbose => 3)
+w.up()
+
+file     = '/etc/hosts'
+look_for = [ 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']
+
+print w.is_available_via_ssh?
+
+## through the scp tunnel
+start = Time.now
+look_for.each do |element|
+  print w.is_in_file?(file, element)
+end
+
+finish = Time.now
+
+print "\nscp tunnel took #{finish - start}\n"
+local = '/tmp/test-ing'
+
+## get the file, then _run against it
+
+w.get(file, local)
+
+start = Time.now
+look_for.each do |element|
+  begin
+    print w._run(sprintf("grep -c '%s' %s", element, local))
+  rescue
+  end
+
+end
+finish = Time.now
+
+print "\nwith a get, took #{finish - start}\n"

data/test/unit/puppet/test_get_puppet_star.rb

@@ -0,0 +1,68 @@
+require sprintf('%s/../../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/puppet'
+require 'test/unit'
+
+# this is technically a unit test, no need for a real Rouster VM
+
+class TestGetPuppetStar < Test::Unit::TestCase
+
+  def setup
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app', :unittest => true)
+    end
+
+    # expose private methods
+    Rouster.send(:public, *Rouster.protected_instance_methods)
+  end
+
+  def test_happy_get_errors
+
+    output = "[1;35merr: Could not retrieve catalog from remote server: Error 400 on SERVER: no result for [api_vip] in Hiera (YAML/CMS)\nfizzybang\n[1;35merr: Could not retrieve catalog from remote server: Error 400 on SERVER: no result for [foobar] in Hiera (YAML/CMS)"
+    @app.output.push(output)
+
+    assert_not_nil(@app.get_puppet_errors())
+    assert_equal(2, @app.get_puppet_errors().size)
+    assert_equal(@app.get_puppet_errors(), @app.get_puppet_errors(output)) # could be split out to a different test
+
+    assert_nil(@app.get_puppet_notices())
+
+  end
+
+  def test_happy_get_notices
+
+    output = "[0;36mnotice: Not using cache on failed catalog\nfizzbang\n[0;36mnotice: Not using cache on failed catalog"
+    @app.output.push(output)
+
+    assert_not_nil(@app.get_puppet_notices())
+    assert_equal(2, @app.get_puppet_notices().size)
+    assert_equal(@app.get_puppet_notices(), @app.get_puppet_notices(output))
+
+    assert_nil(@app.get_puppet_errors())
+
+  end
+
+  def test_no_errors
+
+    output = 'there are no errors here'
+    @app.output.push(output)
+
+    assert_nil(@app.get_puppet_errors())
+  end
+
+  def test_no_notices
+
+    output = 'there are no notices here'
+    @app.output.push(output)
+
+    assert_nil(@app.get_puppet_notices())
+
+  end
+
+  def teardown
+    # noop
+  end
+
+end
+

data/test/unit/test_generate_unique_mac.rb

@@ -0,0 +1,43 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'test/unit'
+
+class TestGenerateUniqueMac < Test::Unit::TestCase
+
+  def setup
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app', :unittest => true)
+    end
+
+    def @app.exposed_generate_unique_mac(*args)
+      generate_unique_mac
+    end
+
+  end
+
+  def test_happy_path
+
+    assert_nothing_raised do
+      @app.exposed_generate_unique_mac
+    end
+
+  end
+
+  def test_uniqueness
+
+    # is this really a valid test?
+    (0..100).each do |i|
+      a = @app.exposed_generate_unique_mac
+      b = @app.exposed_generate_unique_mac
+
+      assert_not_equal(a, b)
+    end
+
+  end
+
+  def teardown
+    # noop
+  end
+
+end

data/test/unit/test_new.rb

@@ -0,0 +1,31 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'test/unit'
+
+# TODO add a bad perms test -- though it should be fixed automagically
+
+class TestNew < Test::Unit::TestCase
+
+  def setup
+    @app = nil
+  end
+
+  # TODO this is an awful pattern, do better
+
+  def test_1_good_basic_instantiation
+
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app', :unittest => true)
+    end
+
+    assert_equal('app', @app.name)
+    assert_equal(false, @app.is_passthrough?())
+    assert_equal(true, @app.uses_sudo?())
+  end
+
+  def teardown
+    # noop
+  end
+
+end

data/test/unit/test_parse_ls_string.rb

@@ -0,0 +1,334 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+# this is based on RHEL output, how cross compatible is this?
+
+require 'rouster'
+require 'rouster/tests'
+require 'test/unit'
+
+class TestParseLsString < Test::Unit::TestCase
+
+  def setup
+
+    @app = Rouster.new(:name => 'app', :unittest => true)
+
+    def @app.exposed_parse_ls_string(*args)
+      parse_ls_string(*args)
+    end
+
+  end
+
+  def test_readable_by_all
+    str = "-r--r--r-- 1 root root 199 May 27 22:51 readable\n"
+
+    expectation = {
+      :directory?  => false,
+      :file?       => true,
+      :mode        => '0444',
+      :name        => 'readable',
+      :owner       => 'root',
+      :group       => 'root',
+      :size        => '199',
+      :executable? => [false, false, false],
+      :readable?   => [true, true, true],
+      :writeable?  => [false, false, false]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_readable_by_u
+    str = "-r-------- 1 root root 199 May 27 22:51 readable\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0400',
+        :name        => 'readable',
+        :owner       => 'root',
+        :group       => 'root',
+        :size        => '199',
+        :executable? => [false, false, false],
+        :readable?   => [true,  false, false],
+        :writeable?  => [false, false, false]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_readable_by_g
+    str = "----r----- 1 root root 199 May 27 22:51 readable\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0040',
+        :name        => 'readable',
+        :owner       => 'root',
+        :group       => 'root',
+        :size        => '199',
+        :executable? => [false, false, false],
+        :readable?   => [false, true, false],
+        :writeable?  => [false, false, false]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_readable_by_o
+    str = "-------r-- 1 root root 199 May 27 22:51 readable\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0004',
+        :name        => 'readable',
+        :owner       => 'root',
+        :group       => 'root',
+        :size        => '199',
+        :executable? => [false, false, false],
+        :readable?   => [false, false, true],
+        :writeable?  => [false, false, false]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_executable_by_all
+    str = "---x--x--x 1 root root 199 May 27 22:51 executable\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0111',
+        :name        => 'executable',
+        :owner       => 'root',
+        :group       => 'root',
+        :size        => '199',
+        :executable? => [true, true, true],
+        :readable?   => [false, false, false],
+        :writeable?  => [false, false, false]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_executable_by_u
+    str = "---x------ 1 root root 199 May 27 22:51 executable\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0100',
+        :name        => 'executable',
+        :owner       => 'root',
+        :group       => 'root',
+        :size        => '199',
+        :executable? => [true, false, false],
+        :readable?   => [false, false, false],
+        :writeable?  => [false, false, false]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_executable_by_g
+    str = "------x--- 1 root root 199 May 27 22:51 executable\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0010',
+        :name        => 'executable',
+        :owner       => 'root',
+        :group       => 'root',
+        :size        => '199',
+        :executable? => [false, true, false],
+        :readable?   => [false, false, false],
+        :writeable?  => [false, false, false]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_executable_by_o
+    str = "---------x 1 root root 199 May 27 22:51 executable\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0001',
+        :name        => 'executable',
+        :owner       => 'root',
+        :group       => 'root',
+        :size        => '199',
+        :executable? => [false, false, true],
+        :readable?   => [false, false, false],
+        :writeable?  => [false, false, false]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_writeable_by_all
+    str = "--w--w--w- 1 root root 199 May 27 22:51 writeable\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0222',
+        :name        => 'writeable',
+        :owner       => 'root',
+        :group       => 'root',
+        :size        => '199',
+        :executable? => [false, false, false],
+        :readable?   => [false, false, false],
+        :writeable?  => [true, true, true]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_writeable_by_u
+    str = "--w------- 1 root root 199 May 27 22:51 writeable\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0200',
+        :name        => 'writeable',
+        :owner       => 'root',
+        :group       => 'root',
+        :size        => '199',
+        :executable? => [false, false, false],
+        :readable?   => [false, false, false],
+        :writeable?  => [true, false, false]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_writeable_by_g
+    str = "-----w---- 1 root root 199 May 27 22:51 writeable\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0020',
+        :name        => 'writeable',
+        :owner       => 'root',
+        :group       => 'root',
+        :size        => '199',
+        :executable? => [false, false, false],
+        :readable?   => [false, false, false],
+        :writeable?  => [false, true, false]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_writeable_by_o
+    str = "--------w- 1 root root 199 May 27 22:51 writeable\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0002',
+        :name        => 'writeable',
+        :owner       => 'root',
+        :group       => 'root',
+        :size        => '199',
+        :executable? => [false, false, false],
+        :readable?   => [false, false, false],
+        :writeable?  => [false, false, true]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_mix_and_match_1
+    str = "-------rwx 1 vagrant vagrant 1909 May 27 22:51 able\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0007',
+        :name        => 'able',
+        :owner       => 'vagrant',
+        :group       => 'vagrant',
+        :size        => '1909',
+        :executable? => [false, false, true],
+        :readable?   => [false, false, true],
+        :writeable?  => [false, false, true]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_mix_and_match_2
+    str = "-rw-r--r--  1 vagrant  root  0 Jun 13 09:35 foo\n"
+
+    expectation = {
+        :directory?  => false,
+        :file?       => true,
+        :mode        => '0644',
+        :name        => 'foo',
+        :owner       => 'vagrant',
+        :group       => 'root',
+        :size        => '0',
+        :executable? => [false, false, false],
+        :readable?   => [true, true, true],
+        :writeable?  => [true, false, false]
+    }
+
+    res = @app.exposed_parse_ls_string(str)
+
+    assert_equal(expectation, res)
+  end
+
+  def test_dir_detection
+    dir_str = "drwxrwxrwt 5 root root 4096 May 28 00:26 /tmp/\n"
+    file_str  = "-rw-r--r-- 1 root    root      906 Oct  2  2012 grub.conf\n"
+
+    dir  = @app.exposed_parse_ls_string(dir_str)
+    file = @app.exposed_parse_ls_string(file_str)
+
+    assert_equal(true,  dir[:directory?])
+    assert_equal(false, dir[:file?])
+
+    assert_equal(false, file[:directory?])
+    assert_equal(true,  file[:file?])
+
+  end
+
+  def teardown
+    # noop
+  end
+
+end