automateit 0.70923

data/spec/integration/template_manager_erb_spec.rb

@@ -0,0 +1,31 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "/../spec_helper.rb")
+
+describe "AutomateIt::TemplateManager::ERB" do
+  before(:all) do
+    @a = AutomateIt.new(:verbosity => Logger::WARN)
+    @m = @a.template_manager
+    @d = @m[:erb]
+    @d.setup(:default_check => :mtime)
+    @d.available?.should be_true
+  end
+
+  it "should set mode when rendering" do
+    @a.mktempdircd do
+      source = "foo"
+      target = "bar"
+      mode1 = 0646 if INTERPRETER.shell_manager.provides_mode?
+      mode2 = 0100646
+      File.open(source, "w+"){|h| h.write("<%=variable%>")}
+
+      opts = {:file => source, :to => target, :locals => {:variable => 42}}
+      opts[:mode] = mode1 if mode1
+      @a.render(opts).should be_true
+      File.read(target).should == "42"
+      if INTERPRETER.shell_manager.provides_mode?
+        File.stat(target).mode.should == mode2
+      else
+        puts "NOTE: Can't check permission modes on this platform, #{__FILE__}"
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,23 @@
+AutomateIt_Base = File.dirname(File.expand_path(__FILE__)) unless defined?(AutomateIt_Base)
+AutomateIt_Lib = File.join(AutomateIt_Base, "/../lib") unless defined?(AutomateIt_Lib)
+AutomateIt_Bin = File.join(AutomateIt_Base, "/../bin") unless defined?(AutomateIt_Bin)
+
+$LOAD_PATH.unshift(AutomateIt_Lib)
+
+# Remove environmental variables that will contaminate tests
+ENV.delete("AUTOMATEIT_PROJECT")
+
+require 'automateit'
+
+# Create a global instance, using WARN verbosity so that
+# INTERPRETER.mktempdircd calls don't generate output.
+unless defined?(INTERPRETER)
+  INTERPRETER = AutomateIt.new(:verbosity => Logger::WARN) 
+end
+
+# Inject matchers into interpreter, e.g., 'should be_true'
+INTERPRETER.class.send(:include, Spec::Matchers)
+
+def write_to(filename, text)
+  File.open(filename, "w+"){|h| h.write(text)}
+end

data/spec/unit/edit_manager_spec.rb

@@ -0,0 +1,162 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "/../spec_helper.rb")
+
+describe "AutomateIt::EditManager for strings" do
+  before(:all) do
+    @a = AutomateIt.new(:verbosity => Logger::WARN)
+  end
+
+  before(:each) do
+    @input = "This\nis\n\a\nstring."
+  end
+
+  it "should pass contents" do
+    @a.edit(:text => @input) do
+      contents.should == "This\nis\n\a\nstring."
+    end
+  end
+
+  it "should pass params" do
+    @a.edit(:text => @input, :params => {:hello => "world"}) do
+      params[:hello].should == "world"
+    end
+  end
+
+  it "should find contained lines" do
+    @a.edit(:text => @input) do
+      contains?(/This/).should == true
+    end
+  end
+
+  it "should prepend lines to the top" do
+    output = @a.edit(:text => @input) do
+      prepend "PREPEND"
+      prepend "PREPEND" # Duplicate line will be ignored.
+    end
+    output.should =~ /\APREPEND\nThis/s
+  end
+
+  it "should prepend lines to the top unless they match an expression" do
+    output = @a.edit(:text => @input) do
+      prepend "PREPEND", :unless => /PR.+ND/
+      prepend "PRETENDER", :unless => /PR.+ND/ # Regexp matches.
+    end
+    output.should =~ /\APREPEND\nThis/s
+  end
+
+  it "should append lines to the bottom" do
+    output = @a.edit(:text => @input) do
+      append "APPEND"
+      append "APPEND" # Duplicate line will be ignored.
+    end
+    output.should =~ /string\.\nAPPEND\Z/s
+  end
+
+  it "should append lines to the bottom unless they match an expression" do
+    output = @a.edit(:text => @input) do
+      append "APPEND", :unless => /^APP/
+      append "APPENDIX", :unless => /^APP/ # Regexp matches.
+    end
+    output.should =~ /^string\.\nAPPEND\Z/s
+  end
+
+  it "should delete lines" do
+    output = @a.edit(:text => @input) do
+      delete "This"
+    end
+    output.should_not =~ /This/
+  end
+
+  it "should comment lines" do
+    output = @a.edit(:text => @input) do
+      comment_style "<", ">"
+      comment "This"
+    end
+    output.should =~ /^<This>$/s
+  end
+
+  it "should uncomment lines" do
+    output = @a.edit(:text => @input) do
+      comment_style "T", "s"
+      uncomment "hi"
+    end
+    output.should =~ /^hi\nis\n/s
+  end
+
+  it "should replace strings" do
+    output = @a.edit(:text => @input) do
+      replace "This", "That"
+    end
+    output.should =~ /^That\nis\n/s
+  end
+
+  it "should manipulate contents" do
+    output = @a.edit(:text => @input) do
+      manipulate do |buffer|
+        buffer.gsub(/i/, "@")
+      end
+    end
+    output.should =~ /^Th@s\n@s\n/
+  end
+
+  it "should tell if content is different after an edit" do
+    @a.edit(:text => @input) do
+      different?.should == false
+      append "changing"
+      different?.should == true
+    end
+  end
+
+  it "should provide session with access to interpreter" do
+    @a.lookup["foo"] = "bar"
+    output = @a.edit(:text => @input) do
+      manipulate do |b|
+        lookup "foo"
+      end
+    end
+    output.should == "bar"
+  end
+
+  it "should raise exceptions for invalid methods" do
+    lambda { 
+      @a.edit(:text => @input) do
+        qwoiuerjzxiuo
+      end
+    }.should raise_error(NameError, /qwoiuerjzxiuo/)
+  end
+end
+
+describe "AutomateIt::EditManager for files" do
+  before(:all) do
+    @a = AutomateIt.new(:verbosity => Logger::WARN)
+    @filename = "input"
+  end
+
+  before(:each) do
+    @input = "This\nis\n\a\nstring."
+  end
+
+  it "should edit a file" do
+    File.should_receive(:read).with(@filename).and_return(@input)
+    File.should_receive(:open).with(@filename, "w+").and_return(true)
+    result = @a.edit(:file => @filename) do
+      append "APPEND"
+    end
+    result.should be_true
+  end
+
+  it "should not rewrite an unchanged file" do
+    File.should_receive(:read).with(@filename).and_return(@input)
+    result = @a.edit(:file => @filename) do
+      # Do nothing
+    end
+    result.should be_false
+  end
+
+  it "should default to editing a file" do
+    File.should_receive(:read).with(@filename).and_return(@input)
+    result = @a.edit(@filename) do
+      # Do nothing
+    end
+    result.should be_false
+  end
+end

data/spec/unit/field_manager_spec.rb

@@ -0,0 +1,79 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "/../spec_helper.rb")
+
+describe "AutomateIt::FieldManager", :shared => true do
+  before(:all) do
+    @a = AutomateIt.new
+    @a.tags << "magical_hosts"
+    @m = @a.field_manager
+  end
+
+  it "should be able to lookup entire hash" do
+    rv = @m.lookup("*")
+    rv.keys.should include("hash")
+    rv["hash"].should be_a_kind_of(Hash)
+    rv.keys.should include("key")
+    rv["key"].should == "value"
+  end
+
+  it "should lookup keys by string" do
+    @m.lookup("key").should == "value"
+  end
+
+  it "should lookup keys by symbol" do
+    @m.lookup(:key).should == "value"
+  end
+
+  it "should lookup hash" do
+    @m.lookup(:hash)["leafkey"].should == "leafvalue"
+  end
+
+  it "should lookup leaves" do
+    @m.lookup("hash#leafkey").should == "leafvalue"
+    @m.lookup("hash#branchkey#deepleafkey").should == "deepleafvalue"
+  end
+
+  it "should have aliased lookup into the interpreter" do
+    @m.should equal(@m.interpreter.field_manager)
+    @m[:yaml].should equal(@m.interpreter.field_manager[:yaml])
+  end
+
+  it "should be aliases into the interpreter" do
+    @a.lookup("hash#leafkey").should == "leafvalue"
+  end
+end
+
+describe AutomateIt::FieldManager::Struct do
+  it_should_behave_like "AutomateIt::FieldManager"
+
+  before(:all) do
+    @m.setup(:default => :struct, :struct => {
+      "key" => "value",
+      "hash" => {
+        "leafkey" => "leafvalue",
+        "branchkey" => {
+          "deepleafkey" => "deepleafvalue",
+        },
+      },
+    })
+  end
+end
+
+describe AutomateIt::FieldManager::YAML do
+  it_should_behave_like "AutomateIt::FieldManager"
+
+  before(:all) do
+    @m[:yaml].should_receive(:_read).with("demo.yml").and_return(<<-EOB)
+      <%="key"%>: value
+      hash:
+        leafkey: leafvalue
+        branchkey:
+          deepleafkey: deepleafvalue
+      magical: <%= tagged?("magical_hosts") ? true : false %>
+    EOB
+    @m.setup(:default => :yaml, :file => "demo.yml")
+  end
+
+  it "should expose the interpreter to ERB statements" do
+    @a.lookup("magical").should be_true
+  end
+end

data/spec/unit/hashcache_spec.rb

@@ -0,0 +1,28 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "/../spec_helper.rb")
+
+describe "HashCache" do
+  before(:each) do
+    @c = HashCache.new
+  end
+
+  it "should store and fetch values" do
+    @c.store("foo", "bar")
+    @c.fetch("foo").should == "bar"
+  end
+
+  it "should store values when needed if given a block" do
+    @c.fetch("foo"){"bar"}.should == "bar"
+    @c.fetch("foo").should == "bar"
+  end
+
+  it "should use HashCached values" do
+    @c.fetch("foo"){"bar"}.should == "bar"
+    @c.fetch("foo"){raise Exception.new("won't be raised")}.should == "bar"
+  end
+
+  it "should invalidate values" do
+    @c.store("foo", "bar")
+    @c.delete("foo")
+    @c.fetch("foo").should be_nil
+  end
+end

data/spec/unit/interpreter_spec.rb

@@ -0,0 +1,98 @@
+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)
+  end
+
+  it "should have a logger" do
+    @a.log.should be_a_kind_of(Logger)
+  end
+
+  it "should provide a preview mode" do
+    @a.preview = true
+    @a.preview?.should be_true
+    @a.preview_for("answer"){42}.should == :preview
+    @a.noop?.should be_true
+    @a.writing?.should be_false
+
+    @a.preview = false
+    @a.preview?.should be_false
+    @a.preview_for("answer"){42}.should == 42
+    @a.noop?.should be_false
+    @a.writing?.should be_true
+  end
+
+  it "should eval commands within Interpreter's context" do
+    @a.preview = true
+    @a.instance_eval{preview?}.should be_true
+    @a.instance_eval{self}.should == @a
+    @a.instance_eval("preview?").should be_true
+  end
+
+  it "should be able to include methods into a class" do
+    @a.preview = true
+    @a.instance_eval do
+      class MyInterpreterWrapper1
+        def initialize(interpreter)
+          interpreter.include_in(self)
+        end
+
+        def answer
+          42
+        end
+
+        # Method overridden by Interpreter#preview?
+        def preview?
+          raise ArgumentError.new("MyInterpreterWrapper#preview? wasn't overriden")
+        end
+      end
+      mc = MyInterpreterWrapper1.new(self)
+      mc.answer.should == 42      # Instance method
+      mc.preview?.should == true  # Interpreter method that overrides instance
+      lambda{ mc.not_a_method }.should raise_error(NoMethodError)
+    end
+  end
+
+  it "should be able to add method_missing to a class" do
+    @a.preview = true
+    @a.instance_eval do
+      class MyInterpreterWrapper2
+        def initialize(interpreter)
+          interpreter.add_method_missing_to(self)
+        end
+
+        def answer
+          42
+        end
+
+        # Method masking Interpreter#preview?
+        def preview?
+          42
+        end
+      end
+      mc = MyInterpreterWrapper2.new(self)
+      mc.answer.should == 42      # Instance method
+      mc.preview?.should == 42       # Instance method masking Interpreter
+      lambda{ mc.not_a_method }.should raise_error(NoMethodError)
+    end
+  end
+
+  it "should be able to add method_missing to a class with an existing method_missing" do
+    @a.preview = true
+    @a.instance_eval do
+      class MyInterpreterWrapper3
+        def initialize(interpreter)
+          interpreter.add_method_missing_to(self)
+        end
+
+        def method_missing(method, *args, &block)
+          42
+        end
+      end
+      mc = MyInterpreterWrapper3.new(self)
+      mc.answer.should == 42      # Instance#method_missing
+      mc.preview?.should == true  # Interpreter method
+    end
+  end
+end

data/spec/unit/platform_manager_spec.rb

@@ -0,0 +1,44 @@
+require File.join(File.dirname(File.expand_path(__FILE__)), "/../spec_helper.rb")
+
+describe "AutomateIt::PlatformManager" do
+  before(:all) do
+    @a = AutomateIt.new
+    @m = @a.platform_manager
+    @m.setup(:default => :struct, :struct => {
+      :os => "mizrahi",
+      :arch => "realian",
+      :distro => "momo",
+      :version => "s100",
+    })
+  end
+
+  it "should query key by symbol" do
+    @m.query(:os).should == "mizrahi"
+  end
+
+  it "should query key by string" do
+    @m.query("distro").should == "momo"
+  end
+
+  it "should query by two-key query" do
+    @m.query("os#arch").should == "mizrahi_realian"
+    @m.query("distro#version").should == "momo_s100"
+  end
+
+  it "should query by three-part query" do
+    @m.query("os#distro#version").should == "mizrahi_momo_s100"
+    @m.query("distro#version#arch").should == "momo_s100_realian"
+  end
+
+  it "should query by aliases" do
+    @m.query("release#version").should == "s100_s100"
+  end
+
+  it "should fail on invalid top-level keys" do
+    lambda { @m.query(:asdf) }.should raise_error(IndexError)
+  end
+
+  it "should fail on invalid subkeys" do
+    lambda { @m.query("os#asdf") }.should raise_error(IndexError)
+  end
+end