1 files changed, 106 insertions, 0 deletions
diff --git a/spec/unit/parser/loaded_code.rb b/spec/unit/parser/loaded_code.rb
new file mode 100644
index 000000000..50f6b930c
--- /dev/null
+++ b/spec/unit/parser/loaded_code.rb
@@ -0,0 +1,106 @@
+#!/usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../../spec_helper'
+
+require 'puppet/parser/loaded_code'
+
+describe Puppet::Parser::LoadedCode do
+    %w{hostclass node definition}.each do |data|
+        it "should have a method for adding a #{data}" do
+            Puppet::Parser::LoadedCode.new.should respond_to("add_" + data)
+        end
+
+        it "should be able to retrieve #{data} by name" do
+            loader = Puppet::Parser::LoadedCode.new
+            loader.send("add_" + data, "foo", "bar")
+            loader.send(data, "foo").should == "bar"
+        end
+
+        it "should retrieve #{data} insensitive to case" do
+            loader = Puppet::Parser::LoadedCode.new
+            loader.send("add_" + data, "Foo", "bar")
+            loader.send(data, "fOo").should == "bar"
+        end
+
+        it "should return nil when asked for a #{data} that has not been added" do
+            Puppet::Parser::LoadedCode.new.send(data, "foo").should be_nil
+        end
+    end
+
+    describe "when finding a qualified instance" do
+        it "should return any found instance if the instance name is fully qualified" do
+            loader = Puppet::Parser::LoadedCode.new
+            loader.add_node "foo::bar", "yay"
+            loader.find("namespace", "::foo::bar", :node).should == "yay"
+        end
+
+        it "should return nil if the instance name is fully qualified and no such instance exists" do
+            loader = Puppet::Parser::LoadedCode.new
+            loader.find("namespace", "::foo::bar", :node).should be_nil
+        end
+
+        it "should return the partially qualified object if it exists in the provided namespace" do
+            loader = Puppet::Parser::LoadedCode.new
+            loader.add_node "foo::bar::baz", "yay"
+            loader.find("foo", "bar::baz", :node).should == "yay"
+        end
+
+        it "should return the unqualified object if it exists in the provided namespace" do
+            loader = Puppet::Parser::LoadedCode.new
+            loader.add_node "foo::bar", "yay"
+            loader.find("foo", "bar", :node).should == "yay"
+        end
+
+        it "should return the unqualified object if it exists in the parent namespace" do
+            loader = Puppet::Parser::LoadedCode.new
+            loader.add_node "foo::bar", "yay"
+            loader.find("foo::bar::baz", "bar", :node).should == "yay"
+        end
+
+        it "should should return the partially qualified object if it exists in the parent namespace" do
+            loader = Puppet::Parser::LoadedCode.new
+            loader.add_node "foo::bar::baz", "yay"
+            loader.find("foo::bar", "bar::baz", :node).should == "yay"
+        end
+
+        it "should return the qualified object if it exists in the root namespace" do
+            loader = Puppet::Parser::LoadedCode.new
+            loader.add_node "foo::bar::baz", "yay"
+            loader.find("foo::bar", "foo::bar::baz", :node).should == "yay"
+        end
+
+        it "should return nil if the object cannot be found" do
+            loader = Puppet::Parser::LoadedCode.new
+            loader.add_node "foo::bar::baz", "yay"
+            loader.find("foo::bar", "eh", :node).should be_nil
+        end
+    end
+
+    it "should use the generic 'find' method with an empty namespace to find nodes" do
+        loader = Puppet::Parser::LoadedCode.new
+        loader.expects(:find).with("", "bar", :node)
+        loader.find_node("bar")
+    end
+
+    it "should use the generic 'find' method to find hostclasses" do
+        loader = Puppet::Parser::LoadedCode.new
+        loader.expects(:find).with("foo", "bar", :hostclass)
+        loader.find_hostclass("foo", "bar")
+    end
+
+    it "should use the generic 'find' method to find definitions" do
+        loader = Puppet::Parser::LoadedCode.new
+        loader.expects(:find).with("foo", "bar", :definition)
+        loader.find_definition("foo", "bar")
+    end
+
+    it "should indicate whether any nodes are defined" do
+        loader = Puppet::Parser::LoadedCode.new
+        loader.add_node("foo", "bar")
+        loader.should be_nodes
+    end
+
+    it "should indicate whether no nodes are defined" do
+        Puppet::Parser::LoadedCode.new.should_not be_nodes
+    end
+end