2 files changed, 60 insertions, 0 deletions
diff --git a/lib/puppet/indirector/envelope.rb b/lib/puppet/indirector/envelope.rb
new file mode 100644
index 000000000..ef7952ef6
--- /dev/null
+++ b/lib/puppet/indirector/envelope.rb
@@ -0,0 +1,13 @@
+require 'puppet/indirector'
+
+# Provide any attributes or functionality needed for indirected
+# instances.
+module Puppet::Indirector::Envelope
+    attr_accessor :expiration
+
+    def expired?
+        return false unless expiration
+        return false if expiration >= Time.now
+        return true
+    end
+end
diff --git a/spec/unit/indirector/envelope.rb b/spec/unit/indirector/envelope.rb
new file mode 100755
index 000000000..17c62023a
--- /dev/null
+++ b/spec/unit/indirector/envelope.rb
@@ -0,0 +1,47 @@
+#!/usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../../spec_helper'
+require 'puppet/indirector/envelope'
+
+describe Puppet::Indirector::Envelope do
+    before do
+        @instance = Object.new
+        @instance.extend(Puppet::Indirector::Envelope)
+    end
+
+    it "should have an expiration accessor" do
+        @instance.expiration = "testing"
+        @instance.expiration.should == "testing"
+    end
+
+    it "should have an expiration setter" do
+        @instance.should respond_to(:expiration=)
+    end
+
+    it "should have a means of testing whether it is expired" do
+        @instance.should respond_to(:expired?)
+    end
+
+    describe "when testing if it is expired" do
+        it "should return false if there is no expiration set" do
+            @instance.should_not be_expired
+        end
+
+        it "should return true if the current date is after the expiration date" do
+            @instance.expiration = Time.now - 10
+            @instance.should be_expired
+        end
+
+        it "should return false if the current date is prior to the expiration date" do
+            @instance.expiration = Time.now + 10
+            @instance.should_not be_expired
+        end
+
+        it "should return false if the current date is equal to the expiration date" do
+            now = Time.now
+            Time.stubs(:now).returns(now)
+            @instance.expiration = now
+            @instance.should_not be_expired
+        end
+    end
+end