4 files changed, 199 insertions, 42 deletions

diff --git a/spec/unit/indirector/code/configuration.rb b/spec/unit/indirector/code/configuration.rb
index bc54f4e1c..0038a038e 100755
--- a/spec/unit/indirector/code/configuration.rb
+++ b/spec/unit/indirector/code/configuration.rb
@@ -143,7 +143,9 @@ describe Puppet::Indirector::Code::Configuration, " when creating configurations
 
     it "should return the results of compiling as the configuration" do
         config = mock 'config'
-        @compiler.interpreter.expects(:compile).with(@node).returns(:configuration)
+        result = mock 'result', :to_transportable => :configuration
+
+        @compiler.interpreter.expects(:compile).with(@node).returns(result)
         @compiler.find(@name).should == :configuration
     end
 
diff --git a/spec/unit/indirector/indirection.rb b/spec/unit/indirector/indirection.rb
index 4311c88bf..7a1c4531c 100755
--- a/spec/unit/indirector/indirection.rb
+++ b/spec/unit/indirector/indirection.rb
@@ -7,28 +7,62 @@ require 'puppet/indirector'
 describe Puppet::Indirector::Indirection do
     before do
         @indirection = Puppet::Indirector::Indirection.new(mock('model'), :test)
-        @terminus = mock 'terminus'
+        @terminus = stub 'terminus', :has_most_recent? => false
         @indirection.stubs(:terminus).returns(@terminus)
+        @instance = stub 'instance', :version => nil, :version= => nil, :name => "whatever"
+        @name = :mything
+    end
+  
+    it "should not attempt to set a timestamp if the terminus cannot find the instance" do
+        @terminus.expects(:find).with(@name).returns(nil)
+        proc { @indirection.find(@name) }.should_not raise_error
     end
   
     it "should handle lookups of a model instance by letting the appropriate terminus perform the lookup" do
-        @terminus.expects(:find).with(:mything).returns(:whev)
-        @indirection.find(:mything).should == :whev
+        @terminus.expects(:find).with(@name).returns(@instance)
+        @indirection.find(@name).should == @instance
     end
 
     it "should handle removing model instances from a terminus letting the appropriate terminus remove the instance" do
-        @terminus.expects(:destroy).with(:mything).returns(:whev)
-        @indirection.destroy(:mything).should == :whev
+        @terminus.expects(:destroy).with(@name).returns(@instance)
+        @indirection.destroy(@name).should == @instance
     end
   
     it "should handle searching for model instances by letting the appropriate terminus find the matching instances" do
-        @terminus.expects(:search).with(:mything).returns(:whev)
-        @indirection.search(:mything).should == :whev
+        @terminus.expects(:search).with(@name).returns(@instance)
+        @indirection.search(@name).should == @instance
     end
   
     it "should handle storing a model instance by letting the appropriate terminus store the instance" do
-        @terminus.expects(:save).with(:mything).returns(:whev)
-        @indirection.save(:mything).should == :whev
+        @terminus.expects(:save).with(@instance).returns(@instance)
+        @indirection.save(@instance).should == @instance
+    end
+
+    it "should add versions to found instances that do not already have them" do
+        @terminus.expects(:find).with(@name).returns(@instance)
+        time = mock 'time'
+        time.expects(:utc).returns(:mystamp)
+        Time.expects(:now).returns(time)
+        @instance.expects(:version=).with(:mystamp)
+        @indirection.find(@name)
+    end
+
+    it "should add versions to saved instances that do not already have them" do
+        time = mock 'time'
+        time.expects(:utc).returns(:mystamp)
+        Time.expects(:now).returns(time)
+        @instance.expects(:version=).with(:mystamp)
+        @terminus.stubs(:save)
+        @indirection.save(@instance)
+    end
+
+    # We've already tested this, basically, but...
+    it "should use the current time in UTC for versions" do
+        @instance.expects(:version=).with do |time|
+            time.utc?
+        end
+        @terminus.stubs(:save)
+        @indirection.save(@instance)
     end
 
     after do
@@ -189,33 +223,14 @@ describe Puppet::Indirector::Indirection, " when deciding whether to cache" do
         proc { @indirection.cache_class = :foo }.should raise_error(ArgumentError)
     end
 
-    it "should not use a cache if there no cache setting" do
-        @indirection.expects(:cache).never
-        @terminus.stubs(:save)
-        @indirection.save(:whev)
-    end
-
-    it "should use a cache if a cache was configured" do
-        cache = mock 'cache'
-        cache.expects(:save).with(:whev)
-
-        cache_class = mock 'cache class'
-        cache_class.expects(:new).returns(cache)
-        Puppet::Indirector::Terminus.stubs(:terminus_class).with(:mycache, :test).returns(cache_class)
-
-        @indirection.cache_class = :mycache
-        @terminus.stubs(:save)
-        @indirection.save(:whev)
-    end
-
     after do
         @indirection.delete
         Puppet::Indirector::Indirection.clear_cache
     end
 end
 
-describe Puppet::Indirector::Indirection, " when using a cache" do
-    before do
+module IndirectionCaching
+    def setup
         Puppet.settings.stubs(:value).with("test_terminus").returns("test_terminus")
         @terminus_class = mock 'terminus_class'
         @terminus = mock 'terminus'
@@ -228,13 +243,14 @@ describe Puppet::Indirector::Indirection, " when using a cache" do
         @indirection.terminus_class = :test_terminus
     end
 
-    it "should copy all writing indirection calls to the cache terminus" do
-        @cache_class.expects(:new).returns(@cache)
-        @indirection.cache_class = :cache_terminus
-        @cache.expects(:save).with(:whev)
-        @terminus.stubs(:save)
-        @indirection.save(:whev)
+    def teardown
+        @indirection.delete
+        Puppet::Indirector::Indirection.clear_cache
     end
+end
+
+describe Puppet::Indirector::Indirection, " when managing the cache terminus" do
+    include IndirectionCaching
 
     it "should not create a cache terminus at initialization" do
         # This is weird, because all of the code is in the setup.  If we got
@@ -257,9 +273,77 @@ describe Puppet::Indirector::Indirection, " when using a cache" do
         @indirection.clear_cache
         @indirection.cache.should equal(cache2)
     end
+end
 
-    after do
-        @indirection.delete
-        Puppet::Indirector::Indirection.clear_cache
+describe Puppet::Indirector::Indirection, " when saving and using a cache" do
+    include IndirectionCaching
+
+    before do
+        @indirection.cache_class = :cache_terminus
+        @cache_class.expects(:new).returns(@cache)
+        @name = "testing"
+        @instance = stub 'instance', :version => 5, :name => @name
+    end
+
+    it "should not update the cache or terminus if the new object is not different" do
+        @cache.expects(:has_most_recent?).with(@name, 5).returns(true)
+        @indirection.save(@instance)
+    end
+
+    it "should update the original and the cache if the cached object is different" do
+        @cache.expects(:has_most_recent?).with(@name, 5).returns(false)
+        @terminus.expects(:save).with(@instance)
+        @cache.expects(:save).with(@instance)
+        @indirection.save(@instance)
+    end
+end
+
+describe Puppet::Indirector::Indirection, " when finding and using a cache" do
+    include IndirectionCaching
+
+    before do
+        @indirection.cache_class = :cache_terminus
+        @cache_class.expects(:new).returns(@cache)
+    end
+
+    it "should return the cached object if the cache is up to date" do
+        cached = mock 'cached object'
+
+        name = "myobject"
+
+        @terminus.expects(:version).with(name).returns(1)
+        @cache.expects(:has_most_recent?).with(name, 1).returns(true)
+
+        @cache.expects(:find).with(name).returns(cached)
+
+        @indirection.find(name).should equal(cached)
+    end
+
+    it "should return the original object if the cache is not up to date" do
+        real = stub 'real object', :version => 1
+
+        name = "myobject"
+
+        @cache.stubs(:save)
+        @cache.expects(:has_most_recent?).with(name, 1).returns(false)
+        @terminus.expects(:version).with(name).returns(1)
+
+        @terminus.expects(:find).with(name).returns(real)
+
+        @indirection.find(name).should equal(real)
+    end
+
+    it "should cache any newly returned objects" do
+        real = stub 'real object', :version => 1
+
+        name = "myobject"
+
+        @terminus.expects(:version).with(name).returns(1)
+        @cache.expects(:has_most_recent?).with(name, 1).returns(false)
+
+        @terminus.expects(:find).with(name).returns(real)
+        @cache.expects(:save).with(real)
+
+        @indirection.find(name).should equal(real)
     end
 end
diff --git a/spec/unit/indirector/indirector.rb b/spec/unit/indirector/indirector.rb
index 390907ca2..78c8c614a 100755
--- a/spec/unit/indirector/indirector.rb
+++ b/spec/unit/indirector/indirector.rb
@@ -64,6 +64,16 @@ describe Puppet::Indirector, " when redirecting a model" do
         @indirection = @thingie.send(:indirects, :test)
     end
 
+    it "should give the model the ability set a version" do
+        thing = @thingie.new
+        thing.should respond_to(:version=)
+    end
+
+    it "should give the model the ability retrieve a version" do
+        thing = @thingie.new
+        thing.should respond_to(:version)
+    end
+
     it "should give the model the ability to lookup a model instance by letting the indirection perform the lookup" do
         @indirection.expects(:find)
         @thingie.find
diff --git a/spec/unit/indirector/terminus.rb b/spec/unit/indirector/terminus.rb
index 44180cf4b..3361bfeeb 100755
--- a/spec/unit/indirector/terminus.rb
+++ b/spec/unit/indirector/terminus.rb
@@ -1,3 +1,5 @@
+#!/usr/bin/env ruby
+
 require File.dirname(__FILE__) + '/../../spec_helper'
 require 'puppet/defaults'
 require 'puppet/indirector'
@@ -200,8 +202,8 @@ describe Puppet::Indirector::Terminus, " when creating terminus classes" do
     end
 end
 
-describe Puppet::Indirector::Terminus, " when a terminus instance" do
-    before do
+module TerminusInstanceTesting
+    def setup
         Puppet::Indirector::Terminus.stubs(:register_terminus_class)
         @indirection = stub 'indirection', :name => :myyaml, :register_terminus_type => nil
         Puppet::Indirector::Indirection.stubs(:instance).with(:my_stuff).returns(@indirection)
@@ -218,6 +220,10 @@ describe Puppet::Indirector::Terminus, " when a terminus instance" do
         @terminus_class.name = :test
         @terminus = @terminus_class.new
     end
+end
+
+describe Puppet::Indirector::Terminus, " when a terminus instance" do
+    include TerminusInstanceTesting
 
     it "should return the class's name as its name" do
         @terminus.name.should == :test
@@ -236,3 +242,58 @@ describe Puppet::Indirector::Terminus, " when a terminus instance" do
         @terminus.model.should == :yay
     end
 end
+
+describe Puppet::Indirector::Terminus, " when managing indirected instances" do
+    include TerminusInstanceTesting
+
+    it "should support comparing an instance's version with the terminus's version using just the instance's key" do
+        @terminus.should respond_to(:has_most_recent?)
+    end
+
+    it "should fail if the :version method has not been overridden and no :find method is available" do
+        proc { @terminus.version('yay') }.should raise_error(Puppet::DevError)
+    end
+
+    it "should use a found instance's version by default" do
+        name = 'instance'
+        instance = stub name, :version => 2
+        @terminus.expects(:find).with(name).returns(instance)
+        @terminus.version(name).should == 2
+    end
+
+    it "should return nil as the version if no instance can be found" do
+        name = 'instance'
+        @terminus.expects(:find).with(name).returns(nil)
+        @terminus.version(name).should be_nil
+    end
+
+    it "should consider an instance fresh if its version is more recent than the version provided" do
+        name = "yay"
+        @terminus.expects(:version).with(name).returns(5)
+        @terminus.has_most_recent?(name, 4).should be_true
+    end
+
+    it "should consider an instance fresh if its version is equal to the version provided" do
+        name = "yay"
+        @terminus.expects(:version).with(name).returns(5)
+        @terminus.has_most_recent?(name, 5).should be_true
+    end
+
+    it "should consider an instance not fresh if the provided version is more recent than its version" do
+        name = "yay"
+        @terminus.expects(:version).with(name).returns(4)
+        @terminus.has_most_recent?(name, 5).should be_false
+    end
+
+    # Times annoyingly can't be compared directly to numbers, and our
+    # default version is 0.
+    it "should convert versions to floats when checking for freshness" do
+        existing = mock 'existing version'
+        new = mock 'new version'
+        existing.expects(:to_f).returns(1.0)
+        new.expects(:to_f).returns(1.0)
+        name = "yay"
+        @terminus.expects(:version).with(name).returns(existing)
+        @terminus.has_most_recent?(name, new)
+    end
+end