rouster 0.5

data/test/functional/deltas/test_get_users.rb

@@ -0,0 +1,45 @@
+require sprintf('%s/../../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/deltas'
+require 'test/unit'
+
+class TestDeltasGetUsers < 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_users()
+    end
+
+    assert_equal(Hash, res.class)
+    assert_not_nil(@app.deltas[:users])
+
+    res.each_key do |k|
+      assert_not_nil(res[k][:shell])
+      assert_not_nil(res[k][:uid])
+      assert_match(/^\d+$/, res[k][:uid])
+      assert_not_nil(res[k][:gid])
+      assert_match(/^\d+$/, res[k][:gid])
+      assert_not_nil(res[k][:home])
+      assert_not_nil(res[k][:home_exists])
+    end
+
+  end
+
+  # TODO add some caching tests
+
+  def teardown
+    @app = nil
+  end
+
+end

data/test/functional/puppet/test_facter.rb

@@ -0,0 +1,59 @@
+require sprintf('%s/../../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/puppet'
+require 'test/unit'
+
+class TestFacter < Test::Unit::TestCase
+
+  def setup
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app')
+    end
+
+    @app.up()
+  end
+
+  def test_happy_path
+
+    facts = nil
+
+    assert_nothing_raised do
+      facts = @app.facter()
+    end
+
+    assert_equal(true, facts.class.eql?(Hash))
+    assert_equal(facts, @app.facts)
+  end
+
+  def test_negative_caching
+
+    facts = nil
+
+    assert_nothing_raised do
+      facts = @app.facter(false)
+    end
+
+    assert_equal(true, facts.class.eql?(Hash))
+    assert_equal(nil, @app.facts)
+  end
+
+  def test_custom_facts
+
+    facts = nil
+
+    assert_nothing_raised do
+      facts = @app.facter(false, false)
+    end
+
+    assert_equal(true, facts.class.eql?(Hash))
+
+    # need to come up with definitive test to not include custom facts, unsure how exactly we should do this
+
+  end
+
+  def teardown
+    # noop
+  end
+
+end

data/test/functional/test_caching.rb

@@ -0,0 +1,124 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'test/unit'
+
+class TestCaching < Test::Unit::TestCase
+
+  def setup
+    # expose private methods
+    Rouster.send(:public, *Rouster.protected_instance_methods)
+  end
+
+  def test_status_caching
+    timeout = 5
+    app     = Rouster.new(:name => 'app', :cache_timeout => timeout)
+
+    assert_equal(app.cache_timeout, timeout)
+
+    status_orig = app.status()
+
+    assert_not_nil(app.cache[:status])
+    status_orig_time = app.cache[:status][:time]
+
+    assert_equal(status_orig, app.cache[:status][:status])
+
+    status_new = app.status()
+    status_new_time = app.cache[:status][:time]
+
+    assert_equal(status_orig, status_new)
+    assert_equal(status_orig_time, status_new_time)
+
+    app.status()
+    status_orig_time = app.cache[:status][:time]
+    sleep(timeout + 1)
+
+    app.status()
+    status_new_time = app.cache[:status][:time]
+
+    assert_not_equal(status_orig_time, status_new_time)
+
+  end
+
+  def test_status_caching_functional
+    timeout = 5
+    app     = Rouster.new(:name => 'app', :cache_timeout => timeout)
+
+    status_orig = app.status()
+    status_orig_time = app.cache[:status][:time]
+
+    if status_orig.eql?('running')
+      app.suspend()
+    elsif status_orig.eql?('not created')
+      app.up()
+    else
+      # should just be suspended/saved
+      app.destroy()
+    end
+
+    sleep(timeout) # this is almost certainly unnecessary, given the amount of time the above operations take, but just being safe
+
+    status_new = app.status()
+    status_new_time = app.cache[:status][:time]
+
+    assert_not_equal(status_orig, status_new)
+    assert_not_equal(status_orig_time, status_new_time)
+
+  end
+
+  def test_status_caching_negative
+    app = Rouster.new(:name => 'app')
+
+    app.status()
+    assert_nil(app.cache[:status])
+  end
+
+  def test_ssh_caching
+
+    skip('see comments in rouster.rb line ~84')
+
+    timeout = 100
+    app     = Rouster.new(:name => 'app', :sshtunnel => true, :cache_timeout => timeout)
+    app.up()
+
+    assert_equal(app.cache_timeout, timeout)
+
+    avail_orig = app.is_available_via_ssh?
+    assert_not_nil(app.cache[:is_available_via_ssh?])
+
+    avail_orig_time = app.cache[:is_available_via_ssh?][:time]
+    assert_equal(avail_orig, app.cache[:is_available_via_ssh?][:status])
+
+    avail_new = app.is_available_via_ssh?
+    avail_new_time = app.cache[:is_available_via_ssh?][:time]
+
+    assert_equal(avail_orig, avail_new)
+    assert_equal(avail_new_time, avail_orig_time)
+
+    app.is_available_via_ssh?
+    avail_orig_time = app.cache[:is_available_via_ssh?][:time]
+
+    sleep(timeout + 1)
+
+    app.is_available_via_ssh?
+    avail_new_time = app.cache[:is_available_via_ssh?][:time]
+
+    assert_not_equal(avail_orig_time, avail_new_time)
+  end
+
+  def test_ssh_caching_functional
+    # noop
+  end
+
+  def test_ssh_caching_negative
+    app = Rouster.new(:name => 'app')
+
+    app.is_available_via_ssh?()
+    assert_nil(app.cache[:is_available_via_ssh?])
+  end
+
+  def teardown
+    # noop
+  end
+
+end

data/test/functional/test_destroy.rb

@@ -0,0 +1,51 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'test/unit'
+
+class TestDestroy < Test::Unit::TestCase
+
+  def setup
+    assert_nothing_raised do
+      @app = Rouster.new(:name => 'app')
+    end
+
+    @app.up()
+    assert_equal(@app.is_available_via_ssh?(), true)
+  end
+
+  def test_happy_path
+
+    assert_equal(@app.status(), 'running')
+
+    assert_nothing_raised do
+      @app.destroy()
+    end
+
+    assert_equal(false, @app.is_available_via_ssh?)
+    assert_equal('not created', @app.status())
+  end
+
+  def test_thats_what_we_call_overkill
+    assert_equal(@app.status(), 'running')
+
+    assert_nothing_raised do
+      @app.destroy()
+    end
+
+    assert_equal(false, @app.is_available_via_ssh?)
+    assert_equal('not created', @app.status())
+
+    assert_nothing_raised do
+      @app.destroy()
+    end
+
+    assert_equal('not created', @app.status())
+
+  end
+
+  def teardown
+    @app = nil
+  end
+
+end

data/test/functional/test_dirs.rb

@@ -0,0 +1,88 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/tests'
+require 'test/unit'
+
+class TestDirs < Test::Unit::TestCase
+
+  def setup
+    @app = Rouster.new(:name => 'app', :verbose => 1)
+    @app.up()
+
+    @dir = sprintf('/tmp/rouster.dirs.%s/', Time.now.to_i)
+  end
+
+  def test_happy
+    dirs_expected = ['foo', 'bar', 'baz']
+
+    dirs_expected.each do |dir|
+      @app.run(sprintf('mkdir -p %s/%s', @dir, dir))
+    end
+
+    dirs_actual = @app.dirs(@dir)
+
+    # could totally use an is_deeply here..
+    dirs_actual.each do |dir|
+      dir = dir.gsub(/#{@dir}/, '') # remove the path prefix
+      assert(dirs_expected.member?(dir))
+    end
+
+  end
+
+  def test_happy_filter
+
+    dirs_to_create = ['foo', 'bar', 'baz']
+    dirs_expected  = ['bar', 'baz']
+
+    dirs_to_create.each do |dir|
+      @app.run(sprintf('mkdir -p %s/%s', @dir, dir))
+    end
+
+    dirs_actual = @app.dirs(@dir, 'b*')
+
+    # would like to do some negative testing here
+    dirs_actual.each do |dir|
+      dir = dir.gsub(/#{@dir}/, '') # remove the path prefix
+      assert(dirs_expected.member?(dir))
+    end
+
+
+  end
+
+  def no_test_happy_recurse
+
+    raise NotImplementedError.new()
+
+  end
+
+  def test_case_sensitvity
+    dirs_to_create = ['Fizz', 'foo', 'baz']
+    dirs_expected_sensitive = ['foo']
+    dirs_expected_insensitive = ['Fizz', 'foo']
+
+    dirs_to_create.each do |dir|
+      @app.run(sprintf('mkdir -p %s/%s', @dir, dir))
+    end
+
+    dirs_actual_sensitive = @app.dirs(@dir, 'f*', false)
+
+    dirs_actual_sensitive.each do |dir|
+      dir = dir.gsub(/#{@dir}/, '')
+      assert(dirs_expected_sensitive.member?(dir))
+    end
+
+    dirs_actual_insensitive = @app.dirs(@dir, 'f*', true)
+
+    dirs_actual_insensitive.each do |dir|
+      dir = dir.gsub(/#{@dir}/, '')
+      assert(dirs_expected_insensitive.member?(dir))
+    end
+
+  end
+
+  def teardown
+    @app.run(sprintf('rm -rf %s', @dir))
+  end
+
+end

data/test/functional/test_files.rb

@@ -0,0 +1,64 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/tests'
+require 'test/unit'
+
+class TestFiles < Test::Unit::TestCase
+
+  def setup
+    @app = Rouster.new(:name => 'app')
+    @app.up()
+
+    @dir = sprintf('/tmp/rouster.files.%s/', Time.now.to_i)
+    @app.run(sprintf('mkdir %s', @dir))
+  end
+
+  def test_happy
+    files_expected = ['foo', 'bar', 'baz']
+
+    files_expected.each do |file|
+      @app.run(sprintf('touch %s/%s', @dir, file))
+    end
+
+    files_actual = @app.files(@dir)
+
+    # could totally use an is_deeply here..
+    files_actual.each do |file|
+      file = file.gsub(/#{@dir}/, '') # remove the path prefix
+      assert(files_expected.member?(file))
+    end
+
+  end
+
+  def test_happy_filter
+
+    files_to_create = ['foo', 'bar', 'baz']
+    files_expected  = ['bar', 'baz']
+
+    files_to_create.each do |file|
+      @app.run(sprintf('touch %s/%s', @dir, file))
+    end
+
+    files_actual = @app.files(@dir, 'b*')
+
+    # would like to do some negative testing here
+    files_actual.each do |file|
+      file = file.gsub(/#{@dir}/, '') # remove the path prefix
+      assert(files_expected.member?(file))
+    end
+
+  end
+
+  def no_test_happy_recurse
+
+    raise NotImplementedError.new()
+
+  end
+
+  def teardown
+    # noop
+    @app.run(sprintf('rm -r %s', @dir))
+  end
+
+end

data/test/functional/test_get.rb

@@ -0,0 +1,76 @@
+require sprintf('%s/../../path_helper', File.dirname(File.expand_path(__FILE__)))
+
+require 'rouster'
+require 'rouster/tests'
+require 'test/unit'
+
+class TestGet < Test::Unit::TestCase
+
+  def setup
+    @app = Rouster.new(:name => 'app')
+
+    @app.up()
+
+    @kg_local_location  = sprintf('/tmp/rouster-test_get_local.%s.%s', $$, Time.now.to_i)
+    @kg_local_location.freeze
+    @kg_remote_location = '/etc/hosts'
+    @kb_dne_location    = '/tmp/this-doesnt_exist/and/never/will.txt'
+
+    File.delete(@kg_local_location) if File.file?(@kg_local_location).true?
+
+    assert_equal(@app.is_available_via_ssh?, true, 'app is available via SSH')
+    assert_equal(File.file?(@kg_local_location), false, 'test KG file not present')
+  end
+
+  def test_happy_path
+
+    assert_nothing_raised do
+      @app.get(@kg_remote_location, @kg_local_location)
+    end
+
+    assert(File.file?(@kg_local_location), 'downloaded file exists')
+  end
+
+  def test_local_path_dne
+
+    assert_raise Rouster::FileTransferError do
+      @app.get(@kg_remote_location, @kb_dne_location)
+
+      # TODO how can we check the contents of the expception message?
+    end
+
+    assert_equal(false, File.file?(@kg_local_location), 'known bad local path DNE')
+  end
+
+  def test_remote_path_dne
+
+    assert_raise Rouster::FileTransferError do
+      res = @app.get(@kb_dne_location, @kg_local_location)
+      assert_equal(false, res)
+    end
+
+    assert_equal(false, File.file?(@kg_local_location), 'known bad remote file path DNE')
+  end
+
+  def test_with_suspended_machine
+    @app.suspend()
+
+    #assert_raise Rouster::SSHConnectionError do <-- this is what we want when the connection is bad, as opposed to permission/disk space issues
+    assert_raise Rouster::FileTransferError do
+      @app.get(@kg_remote_location, @kg_local_location)
+    end
+
+    assert_equal(false, File.file?(@kg_local_location), 'when machine is suspended, unable to get from it')
+  end
+
+  def teardown
+    File.delete(@kg_local_location) if File.file?(@kg_local_location).true?
+  end
+
+  def self.shutdown
+    # TODO we should suspend instead if any test failed for triage
+    #@app.suspend()
+    #@ppm.suspend()
+  end
+
+end