automateit 0.71102 → 0.71103

data/lib/ext/shell_escape.rb

@@ -0,0 +1,7 @@
+# Suraj Kurapati -- http://blade.nagaokaut.ac.jp/cgi-bin/scat.rb/ruby/ruby-talk/261459
+class String
+  # Transforms this string into an escaped POSIX shell argument.
+  def shell_escape
+    inspect.gsub(/\\(\d{3})/) {$1.to_i(8).chr}
+  end
+end

data/lib/nitpick.rb

@@ -0,0 +1,29 @@
+module Nitpick
+  module ClassMethods
+    # Use to manage nitpick message for debugging AutomateIt internals.
+    #
+    # Arguments:
+    # * nil -- Returns boolean of whether nitpick messages will be displayed.
+    # * Boolean -- Sets nitpick state.
+    # * String or Symbol -- Displays nitpick message if state is on.
+    #
+    # Example:
+    #   nitpick true
+    #   nitpick "I'm nitpicking"
+    def nitpick(value=nil)
+      case value
+      when NilClass: @nitpick
+      when TrueClass, FalseClass: @nitpick = value
+      when String, Symbol: puts "%% #{value}" if @nitpick
+      else raise TypeError.new("Unknown nitpick type: #{value.class}")
+      end
+    end
+  end
+
+	def self.included(receiver)
+		receiver.extend(ClassMethods)
+	end
+
+  include ClassMethods
+  extend ClassMethods
+end

data/spec/integration/account_manager_nscd_spec.rb

@@ -0,0 +1,37 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "/../spec_helper.rb")
+
+if not INTERPRETER.euid?
+  puts "NOTE: Can't check 'euid' on this platform, #{__FILE__}"
+elsif not INTERPRETER.superuser?
+  puts "NOTE: Must be root to check #{__FILE__}"
+elsif not INTERPRETER.account_manager.available?(:invalidate)
+  puts "NOTE: Can't check AccountManager::NSCD on this platform, #{__FILE__}"
+else
+  describe AutomateIt::AccountManager::NSCD do
+    before(:all) do
+      @a = AutomateIt.new(:verbosity => Logger::WARN)
+      @m = @a.account_manager
+      @d = @m[:nscd]
+    end
+
+    it "should know what NSCD databases are associated with passwd" do
+      for query in %w(user users passwd password)
+        @d.database_for(query).should == :passwd
+      end
+    end
+
+    it "should know what NSCD databases are associated with group" do
+      for query in %w(group groups)
+        @d.database_for(query).should == :group
+      end
+    end
+
+    it "should fail with invalid queries" do
+      lambda { @d.database_for(:stuff) }.should raise_error(ArgumentError)
+    end
+
+    it "should invalidate NSCD databases" do
+      @m.invalidate(:passwd).should be_true
+    end
+  end
+end

data/spec/integration/account_manager_spec.rb

@@ -20,6 +20,7 @@ else
       # Some OSes are limited to 8 character names :(
       @username =  "aitestus"
       @groupname = "aitestgr"
+      @password = "automateit"
 
       begin
         raise "User named '#{@username}' found. If this isn't a real user, delete it so that the test can contineu. If this is a real user, change the spec to test with a user that shouldn't exist." if @m.users[@username]
@@ -68,6 +69,12 @@ else
       return @m.add_group(@groupname, :members => @username)
     end
 
+    unless INTERPRETER.tagged?("windows | darwin")
+      it "should have Etc support on Unix-like platforms" do
+        @m.driver_for(:users).should have_etc
+      end
+    end
+
     it "should find root user" do
       entry = @m.users["root"]
       entry.should_not be_nil
@@ -78,6 +85,10 @@ else
       @m.has_user?(@username).should be_false
     end
 
+    it "should fail to change the password for a non-existent user" do
+      lambda{ @m.passwd(@username, "foo") }.should raise_error(ArgumentError)
+    end
+
     it "should create a user" do
       entry = add_user
 
@@ -253,9 +264,8 @@ else
       end
     end
 
-    it "should change password" do
+    def change_password_with(object)
       add_user if @independent
-      pass = "automateit"
 
       # TODO This isn't portable
       def extract_pwent(username)
@@ -266,9 +276,34 @@ else
       end
 
       before = extract_pwent(@username)
-      @m.passwd(@username, pass, :quiet => @quiet).should be_true
+      object.passwd(@username, @password, :quiet => @quiet).should be_true
       after = extract_pwent(@username)
-      before.should_not eql(after)
+
+      before.should_not == after
+    end
+
+    it "should change password with default driver" do
+      change_password_with(@m)
+    end
+
+    if INTERPRETER.account_manager[:passwd_pty].available?
+      it "should change password with PTY driver" do
+        change_password_with(@m[:passwd_pty])
+      end
+    else
+      puts "NOTE: Can't check AccountManager::PasswdPTY on this platform, #{__FILE__}"
+    end
+
+    if INTERPRETER.account_manager[:passwd_expect].available?
+      it "should change password with Expect driver" do
+        change_password_with(@m[:passwd_expect])
+      end
+    else
+      puts "NOTE: Can't check AccountManager::PasswdExpect on this platform, #{__FILE__}"
+    end
+
+    it "should fail to change the password when given invalid arguents" do
+      lambda{ @m.passwd(Hash.new, "foo") }.should raise_error(TypeError)
     end
 
     it "should remove a user" do

data/spec/integration/shell_manager_spec.rb

@@ -304,7 +304,7 @@ describe AutomateIt::ShellManager, " in general" do
       @a.render(:text => "Hello", :to => source_file2)
       @a.mkdir(target_dir)
 
-      @a.cp(source_dir, target_dir).should == source_dir
+      @a.cp_R(source_dir, target_dir).should == source_dir
       File.exists?("meh/feh/file1").should be_true
       File.exists?("meh/feh/file2").should be_true
 
@@ -312,6 +312,22 @@ describe AutomateIt::ShellManager, " in general" do
     end
   end
 
+  it "should copy files whose contents changed (cp)" do
+    @m.mktempdircd do
+      source = "file1"
+      target = "file2"
+
+      @a.render(:text => "Hello", :to => source)
+      @a.render(:text => "olleH", :to => target)
+
+      source_atime = File.atime(source)
+      source_mtime = File.mtime(source)
+      File.utime(source_atime, source_mtime, target)
+
+      @a.cp(source, target).should == source
+    end
+  end
+
   it "should move files (mv)" do
     @m.mktempdircd do
       file1 = "foo"
@@ -357,6 +373,19 @@ describe AutomateIt::ShellManager, " in general" do
     end
   end
 
+  it "should delete forcefully (rm_f)" do
+    @m.mktempdircd do
+      target = "file1"
+
+      @m.touch(target)
+      @m.chmod(0000, target)
+
+      @m.rm(target, :force => true).should == [target]
+
+      File.exists?(target).should be_false
+    end
+  end
+
   it "should delete recursively and forcefully (rm_rf)" do
     @m.mktempdircd do
       dir = "foo/bar"
@@ -505,9 +534,85 @@ describe AutomateIt::ShellManager, " when managing modes" do
       end
     end
 
-    #it "should set the default mask (umask)" # TODO
-  end
+    it "should copy files whose mode changed (cp)" do
+      @m.mktempdircd do
+        source = "file1"
+        target = "file2"
+
+        @a.render(:text => "Hello", :to => source)
+        @a.cp(source, target)
+
+        File.chmod(0754, target)
+        source_atime = File.atime(source)
+        source_mtime = File.mtime(source)
+        File.utime(source_atime, source_mtime, target)
+
+        @a.cp(source, target, :preserve => true).should == source
+      end
+    end
+
+    it "should return umask value (umask)" do
+      @m.umask.should == File::umask
+    end
+
+    it "should set the default mask (umask)" do
+      @m.mktempdircd do
+        source = "file1"
+        previous_umask = File::umask
+
+        begin
+          @m.umask(0000)
+          File::umask.should_not == previous_umask
 
+          @a.touch(source)
+          File.stat(source).mode.should == 0100666
+        ensure
+          File::umask(previous_umask)
+        end
+      end
+    end
+
+    it "should set the default mask within block (umask)" do
+      @m.mktempdircd do
+        source = "file1"
+        previous_umask = File::umask
+
+        begin
+          @m.umask(0000) do
+            File::umask.should_not == previous_umask
+
+            @a.touch(source)
+            File.stat(source).mode.should == 0100666
+          end
+
+          File::umask.should == previous_umask
+        ensure
+          File::umask(previous_umask)
+        end
+      end
+    end
+
+    it "should reset umask correctly after an exception within a block (umask)" do
+      @m.mktempdircd do
+        previous_umask = File::umask
+
+        begin
+          begin
+            @m.umask(0000) do
+              File::umask.should_not == previous_umask
+              raise ArgumentError.new("OMFG")
+            end
+          rescue ArgumentError
+            # Error was expected
+          end
+
+          File::umask.should == previous_umask
+        ensure
+          File::umask(previous_umask)
+        end
+      end
+    end
+  end
 end
 
 describe AutomateIt::ShellManager, " when managing permissions" do
@@ -602,6 +707,25 @@ describe AutomateIt::ShellManager, " when managing permissions" do
         (stat.gid == @gid).should be_true
       end
     end
+
+    it "should copy files whose owner changed (cp)" do
+      @m.mktempdircd do
+        source = "file1"
+        target = "file2"
+
+        @a.render(:text => "Hello", :to => source)
+        @a.cp(source, target)
+
+        # TODO not portable
+        File.chown(2, 2, target)
+        source_atime = File.atime(source)
+        source_mtime = File.mtime(source)
+        File.utime(source_atime, source_mtime, target)
+
+        @a.cp(source, target, :preserve => true).should == source
+      end
+    end
+
   else
     puts "NOTE: Must be root to check 'chown' in #{__FILE__}"
   end
@@ -613,6 +737,10 @@ describe AutomateIt::ShellManager, " when managing hard links" do
   else
     it_should_behave_like "AutomateIt::ShellManager"
 
+    it "should provide links" do
+      @m.provides_link?.should be_true
+    end
+
     it "should create hard links when needed (ln)" do
       @m.mktempdircd do
         source = "foo"
@@ -636,6 +764,10 @@ describe AutomateIt::ShellManager, " when managing symbolic links" do
   else
     it_should_behave_like "AutomateIt::ShellManager"
 
+    it "should provide symlinks" do
+      @m.provides_symlink?.should be_true
+    end
+
     it "should create symlinks when needed (ln_s)" do
       @m.mktempdircd do
         source = "foo"

data/spec/unit/address_manager_spec.rb

@@ -0,0 +1,49 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "/../spec_helper.rb")
+
+describe AutomateIt::AddressManager do
+  before(:all) do
+    @a = AutomateIt.new(:verbosity => Logger::WARN)
+    @m = @a.address_manager
+    @d = @a.address_manager.driver_for(:hostnames_for)
+  end
+
+  it "should be able to convert netmasks to CIDR addresses" do
+    @d.send(:mask_to_cidr, "255.255.255.0").should == 24
+  end
+
+  it "should be able to convert CIDR addresses to netmasks" do
+    @d.send(:cidr_to_mask, 24).should == "255.255.255.0"
+  end
+
+  it "should be able to convert decimals to binaries" do
+    @d.send(:dec2bin, 255).should == "11111111"
+  end
+
+  it "should be able to convert binaries to decimals" do
+    @d.send(:bin2dec, "11111111").should == 255
+  end
+
+  it "should be able to derive interface and label from a device" do
+    @d.send(:_interface_and_label, :device => "eth0").should == "eth0"
+  end
+
+  it "should be able to derive interface and label from a device and alias" do
+    @d.send(:_interface_and_label, :device => "eth0", :label => "1").should == "eth0:1"
+  end
+
+  it "should fail to derive interface and lable from inadequate options" do
+    lambda { @d.send(:_interface_and_label, :label => "1") }.should raise_error(ArgumentError)
+  end
+
+  it "should be able to prepend arguments to ifconfig commands" do
+    @d.send(:_ifconfig_helper, :del, 
+      {:address => "127.0.0.1", :device => "eth0", :label => "1"}, 
+      {:prepend => "inet"}).should == "ifconfig eth0:1 inet 127.0.0.1 down"
+  end
+
+  it "should be able to append arguments to ifconfig commands" do
+    @d.send(:_ifconfig_helper, :del, 
+      {:address => "127.0.0.1", :device => "eth0", :label => "1"}, 
+      {:append => "unplumb"}).should == "ifconfig eth0:1 127.0.0.1 down unplumb"
+  end
+end

data/spec/unit/interpreter_spec.rb

@@ -2,13 +2,23 @@ require File.join(File.dirname(File.expand_path(__FILE__)), "/../spec_helper.rb"
 
 describe AutomateIt::Interpreter do
   before(:all) do
-    @a = AutomateIt::Interpreter.new(:verbosity => Logger::WARN)
+    @verbosity = Logger::WARN
+    @a = AutomateIt::Interpreter.new(:verbosity => @verbosity)
   end
 
   it "should have a logger" do
     @a.log.should be_a_kind_of(Logger)
   end
 
+  it "should be able to set a logger" do
+    @old_logger = @a.log
+
+    @a.log(QueuedLogger.new($stdout))
+    @a.log.level = @verbosity
+
+    @a.log.should_not == @old_logger
+  end
+
   it "should provide a preview mode" do
     @a.preview = true
     @a.preview?.should be_true
@@ -21,6 +31,27 @@ describe AutomateIt::Interpreter do
     @a.preview_for("answer"){42}.should == 42
     @a.noop?.should be_false
     @a.writing?.should be_true
+
+    @a.preview true
+    @a.preview?.should be_true
+
+    @a.noop false
+    @a.preview?.should be_false
+
+    @a.noop = true
+    @a.preview?.should be_true
+
+    @a.writing true
+    @a.preview?.should be_false
+
+    @a.writing = false
+    @a.preview?.should be_true
+  end
+
+  it "should get and set values in the interpreter" do
+    @a.set("meow", :MEOW).should == :MEOW
+    @a.get("meow").should == :MEOW
+    @a.meow.should == :MEOW
   end
 
   it "should eval commands within Interpreter's context" do

data/spec/unit/nested_error_spec.rb

@@ -0,0 +1,10 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "/../spec_helper.rb")
+
+describe NestedError do
+  it "should preserve the cause" do
+    e = NestedError.new("nested message", TypeError.new("cause message"))
+    e.message.should == "nested message"
+    e.cause.class.should == TypeError
+    e.cause.message.should == "cause message"
+  end
+end