rouster 0.5

data/path_helper.rb

@@ -0,0 +1,21 @@
+# heavily influenced by https://github.com/puppetlabs/hiera/blob/master/spec/spec_helper.rb
+
+# this gets us Rouster, still need to figure out how to find vagrant
+$LOAD_PATH << File.join([File.dirname(__FILE__), 'lib'])
+
+require 'rubygems'
+
+# debugging help
+begin
+  require 'debugger'
+rescue LoadError
+end
+
+class Object
+  def my_methods
+    # Cookbook implementation
+    # my_super = self.class.superclass
+    # my_super ? methods - my_super.methods : methods
+    (methods - Object.methods).sort
+  end
+end

data/rouster.gemspec

@@ -0,0 +1,30 @@
+## heavily modified/copy-paste-replaced from vagrant.gemspec
+
+require './path_helper'
+require 'rouster'
+
+Gem::Specification.new do |s|
+  s.name          = 'rouster'
+  s.version       = Rouster::VERSION
+  s.platform      = Gem::Platform::RUBY
+  s.author        = 'Conor Horan-Kates'
+  s.email         = 'conor.code@gmail.com'
+  s.homepage      = 'http://github.com/chorankates/rouster'
+  s.summary       = 'Rouster is an abstraction layer for Vagrant'
+  s.description   = 'Rouster allows you to programmatically control and interact with your existing Vagrant virtual machines'
+  s.license       = 'BSD 3-Clause'
+
+  s.required_rubygems_version = '>= 1.3.6'
+  s.rubyforge_project         = 'Rouster'
+
+  s.add_dependency 'json'
+  s.add_dependency 'log4r',   '~> 1.1.9'
+  s.add_dependency 'net-scp'
+  s.add_dependency 'net-ssh'
+  s.add_dependency 'rake'
+
+  s.add_development_dependency 'test-unit', '~> 2.0'
+  s.add_development_dependency 'reek'
+
+  s.files = `git ls-files`.split("\n")
+end

data/test/basic.rb

@@ -0,0 +1,10 @@
+require sprintf('%s/../%s', File.dirname(File.expand_path(__FILE__)), 'path_helper')
+
+require 'rouster'
+require 'rouster/puppet'
+require 'rouster/testing'
+require 'rouster/tests'
+
+p = Rouster.new(:name => 'app', :verbosity => 0)
+
+p 'DBGZ' if nil?

data/test/functional/deltas/test_get_crontab.rb

@@ -0,0 +1,99 @@
+require sprintf('%s/../../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/deltas'
+require 'test/unit'
+
+class TestDeltasGetCrontab < Test::Unit::TestCase
+
+  def setup
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app')
+    end
+
+    @app.up()
+
+    ## setup some cronjobs so we have something to look at - and yes, this is hacktastic
+    ['root', 'puppet'].each do | user|
+      tmp = sprintf('/tmp/rouster.tmp.crontab.%s.%s.%s', user, Time.now.to_i, $$)
+      @app.run("echo '0 0 * * * echo #{user}' > #{tmp}")
+      #@app.run("crontab -u #{user} -f #{tmp}") # rhel
+      @app.run("crontab -u #{user} #{tmp}") # centos
+    end
+
+  end
+
+  def test_happy_path
+
+    res = nil
+
+    assert_nothing_raised do
+      res = @app.get_crontab()
+    end
+
+    assert_equal(Hash, res.class)
+    assert_equal(res, @app.deltas[:crontab]['root'])
+    assert_not_nil(@app.deltas[:crontab]['root'])
+
+  end
+
+  def test_happy_path_specified_user
+
+    res = nil
+
+    assert_nothing_raised do
+      res = @app.get_crontab('puppet')
+    end
+
+    assert_equal(Hash, res.class)
+    assert_equal(res, @app.deltas[:crontab]['puppet'])
+    assert_not_nil(@app.deltas[:crontab]['puppet'])
+
+  end
+
+  def test_happy_path_specified_star
+
+    res = nil
+
+    assert_nothing_raised do
+      res = @app.get_crontab('*')
+    end
+
+    assert_equal(Hash, res.class)
+    assert_equal(res, @app.deltas[:crontab])
+    assert_not_nil(@app.deltas[:crontab]['root'])
+    assert_not_nil(@app.deltas[:crontab]['puppet'])
+
+  end
+
+  def test_unhappy_path_non_existent_user
+
+    res = nil
+
+    assert_nothing_raised do
+      res = @app.get_crontab('fizzybang')
+    end
+
+    assert_equal(Hash, res.class)
+    assert_equal(0, res.keys.size)
+
+  end
+
+  def test_happy_path_no_cache
+
+    res = nil
+
+    assert_nothing_raised do
+      res = @app.get_crontab('root', false)
+    end
+
+    assert_equal(Hash, res.class)
+    assert_nil(@app.deltas[:crontab])
+
+  end
+
+  def teardown
+    @app = nil
+  end
+
+end

data/test/functional/deltas/test_get_groups.rb

@@ -0,0 +1,48 @@
+require sprintf('%s/../../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/deltas'
+require 'test/unit'
+
+class TestDeltasGetGroups < Test::Unit::TestCase
+
+  def setup
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app')
+    end
+
+    @app.up()
+  end
+
+  def test_happy_path
+    res = nil
+
+    assert_nothing_raised do
+      res = @app.get_groups()
+    end
+
+    assert_equal(Hash, res.class)
+    assert_not_nil(@app.deltas[:groups])
+
+    res.each_key do |k|
+      assert_not_nil(res[k][:users])
+      assert_equal(res[k][:users].class, Array)
+      assert_not_nil(res[k][:gid])
+    end
+
+    ## only working on *nix right now, check some specific accounts
+    expected = %w[root vagrant]
+
+    expected.each do |e|
+      assert_not_nil(res[e])
+    end
+
+  end
+
+  # TODO add some non-caching tests
+
+  def teardown
+    @app = nil
+  end
+
+end

data/test/functional/deltas/test_get_packages.rb

@@ -0,0 +1,71 @@
+require sprintf('%s/../../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/deltas'
+require 'test/unit'
+
+# TODO need to figure out how to add package strings on our own for better testing (i.e. sfdc-razorpolicy-rhel-6.2-batch-1.0-17.noarch)
+
+class TestDeltasGetPackages < Test::Unit::TestCase
+
+  def setup
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app')
+    end
+
+    @app.up()
+  end
+
+  def test_happy_path
+    res = nil
+
+    assert_equal(false, @app.deltas.has_key?(:packages))
+
+    assert_nothing_raised do
+      res = @app.get_packages()
+    end
+
+    assert_equal(Hash, res.class)
+    assert_not_nil(@app.deltas[:packages])
+
+    res.each_key do |k|
+      assert_not_nil(res[k])
+      assert_match(/^\d+/, res[k]) # start with a number
+    end
+
+  end
+
+  def test_caching_negative
+    res = nil
+
+    assert_equal(false, @app.deltas.has_key?(:packages))
+
+    assert_nothing_raised do
+      res = @app.get_packages(false)
+    end
+
+    assert_equal(Hash, res.class)
+    assert_equal(false, @app.deltas.has_key?(:packages))
+  end
+
+  def test_without_deep_inspection
+    res = nil
+
+    assert_nothing_raised do
+      res = @app.get_packages(true, false)
+    end
+
+    res.each_key do |k|
+      assert_not_nil(res[k])
+
+      #assert_match(/\d*\..*/, res[k]) # testing the regular expression used in deltas.rb itself
+      assert_match(/\?/, res[k])
+    end
+
+  end
+
+  def teardown
+    @app = nil
+  end
+
+end

data/test/functional/deltas/test_get_ports.rb

@@ -0,0 +1,119 @@
+require sprintf('%s/../../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/deltas'
+require 'test/unit'
+
+class TestDeltasGetPorts < Test::Unit::TestCase
+
+  def setup
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app')
+    end
+
+    @app.up()
+  end
+
+  def test_happy_path
+    res = nil
+
+    assert_nil(@app.deltas[:ports])
+
+    assert_nothing_raised do
+      res = @app.get_ports()
+    end
+
+    assert_equal(Hash, res.class)
+
+    res.each_key do |proto|
+      assert_not_nil(res[proto])
+
+      res[proto].each_key do |port|
+        assert_not_nil(res[proto][port])
+      end
+
+    end
+
+    assert_nil(@app.deltas[:ports])
+
+  end
+
+  def test_happy_path_caching
+
+    assert_nil(@app.deltas[:ports])
+
+    assert_nothing_raised do
+      @app.get_ports(true)
+    end
+
+    assert_equal(Hash, @app.deltas[:ports].class)
+
+  end
+
+  # TODO this probably isn't the right placefor this test..
+  def test_functional_tests
+
+    # TODO this is still wrong, since the data structure may change further
+    stock = {
+        "udp"=>
+          {"123"=>
+            {:address=>
+              {"fe80::a00:27ff:fe30:a00b"=>"you_might_not_get_it",
+               "::1"=>"you_might_not_get_it",
+               "0.0.0.0"=>"you_might_not_get_it",
+               "fe80::a00:27ff:fe42:f532"=>"you_might_not_get_it",
+               "127.0.0.1"=>"you_might_not_get_it",
+               "::"=>"you_might_not_get_it",
+               "10.0.2.15"=>"you_might_not_get_it",
+               "10.0.1.104"=>"you_might_not_get_it",
+               "192.168.1.161"=>"you_might_not_get_it"
+              }
+            },
+             "68"=>
+                {:address=>
+                  {"0.0.0.0"=>"you_might_not_get_it"}
+                },
+             "111"=>
+               {:address=>
+                 {"0.0.0.0"=>"you_might_not_get_it", "::"=>"you_might_not_get_it"}
+               },
+             "840"=>
+               {:address=>
+                 {"0.0.0.0"=>"you_might_not_get_it", "::"=>"you_might_not_get_it"}
+               },
+             "161"=>
+               {:address=>
+                 {"0.0.0.0"=>"you_might_not_get_it"}
+               }
+            },
+        "tcp"=>
+          {"111"=>{:address=>{"0.0.0.0"=>"LISTEN", "::"=>"LISTEN"}},
+           "199"=>{:address=>{"127.0.0.1"=>"LISTEN"}},
+           "25"=>{:address=>{"127.0.0.1"=>"LISTEN"}},
+           "22"=>{:address=>{"0.0.0.0"=>"LISTEN", "::"=>"LISTEN"}}
+          }
+    }
+
+    @app.deltas[:ports] = stock
+
+    assert_equal(true, @app.is_port_open?(1234, 'tcp', true))
+    assert_equal(true, @app.is_port_active?(22, 'tcp', true))
+
+    assert_equal(true, @app.is_port_open?(303, 'udp', true))
+    assert_equal(true, @app.is_port_active?(123, 'udp', true))
+
+    # expected to return false
+    assert_equal(false, @app.is_port_open?(22, 'tcp', true))
+    assert_equal(false, @app.is_port_active?(1234, 'tcp', true))
+
+    # caching/argument default validation -- can't currently do this, don't know what ports will be open on others systems
+    #assert_equal(true, @app.is_port_active?(22))
+    #assert_equal(true, @app.is_port_open?(1234))
+
+  end
+
+  def teardown
+    @app = nil
+  end
+
+end

data/test/functional/deltas/test_get_services.rb

@@ -0,0 +1,43 @@
+require sprintf('%s/../../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/deltas'
+require 'test/unit'
+
+class TestDeltasGetServices < Test::Unit::TestCase
+
+  def setup
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app')
+    end
+
+    @app.up()
+
+    # eventually this should be standardized (and symbolized?)
+    @allowed_states = %w(exists operational running stopped)
+  end
+
+  def test_happy_path
+    res = nil
+
+    assert_nothing_raised do
+      res = @app.get_services()
+    end
+
+    assert_equal(Hash, res.class)
+    assert_not_nil(@app.deltas[:services])
+
+    res.each_key do |k|
+      assert_not_nil(res[k])
+      #assert(@allowed_states.member?(res[k]))
+    end
+
+  end
+
+  # TODO add some caching tests
+
+  def teardown
+    @app = nil
+  end
+
+end