2 files changed, 19 insertions, 2 deletions
diff --git a/spec/integration/util/rdoc/parser.rb b/spec/integration/util/rdoc/parser.rb
index 542660998..1dd36c699 100755
--- a/spec/integration/util/rdoc/parser.rb
+++ b/spec/integration/util/rdoc/parser.rb
@@ -35,8 +35,25 @@ describe RDoc::Parser do
         File.unlink(@parsedfile)
     end
 
+    def get_test_class(toplevel)
+        # toplevel -> main -> test
+        toplevel.classes[0].classes[0]
+    end
+
     it "should parse to RDoc data structure" do
         @parser.expects(:document_class).with { |n,k,c| n == "::test" and k.is_a?(Puppet::Parser::ResourceType) }
         @parser.scan
     end
+
+    it "should get a PuppetClass for the main class" do
+        @parser.scan.classes[0].should be_a RDoc::PuppetClass
+    end
+
+    it "should produce a PuppetClass whose name is test" do
+        get_test_class(@parser.scan).name.should == "test"
+    end
+
+    it "should produce a PuppetClass whose comment is 'comment'" do
+        get_test_class(@parser.scan).comment.should == "comment\n"
+    end
 end
diff --git a/spec/unit/util/rdoc/parser.rb b/spec/unit/util/rdoc/parser.rb
index 19c91bb9a..ce776da9b 100755
--- a/spec/unit/util/rdoc/parser.rb
+++ b/spec/unit/util/rdoc/parser.rb
@@ -252,7 +252,7 @@ describe RDoc::Parser do
     describe "when documenting nodes" do
         before :each do
             @code = stub_everything 'code'
-            @node = stub_everything 'node', :doc => "mydoc", :parentclass => "parent", :code => @code, :file => "file", :line => 42
+            @node = stub_everything 'node', :doc => "mydoc", :parent => "parent", :code => @code, :file => "file", :line => 42
             @rdoc_node = stub_everything 'rdocnode'
 
             @class = stub_everything 'class'
@@ -295,7 +295,7 @@ describe RDoc::Parser do
     describe "when documenting classes" do
         before :each do
             @code = stub_everything 'code'
-            @class = stub_everything 'class', :doc => "mydoc", :parentclass => "parent", :code => @code, :file => "file", :line => 42
+            @class = stub_everything 'class', :doc => "mydoc", :parent => "parent", :code => @code, :file => "file", :line => 42
             @rdoc_class = stub_everything 'rdoc-class'
 
             @module = stub_everything 'class'