0
@@ -49,7 +49,8 @@
0
     end
0
     
0
     def attr_on_child(node, child, attrib, value)
0
-      node.at('#'+child).set_attribute(attrib, value)
0
+      child = node.at('#'+child)
0
+      child.set_attribute(attrib, value) if child
0
       node
0
     end
0
     

0
@@ -128,6 +128,13 @@
0
       @f.run 'attr_on_child', @doc.at('div'), 'foo', 'class', 'active'
0
       @doc.at('#foo').classes.should == ['active']
0
     end
0
+    
0
+    it "handles a child not found" do
0
+      html = '<div><span id="bar">bar</span></div>'
0
+      @doc = Hpricot(html)
0
+      @f.run 'attr_on_child', @doc.at('div'), 'foo', 'class', 'active'
0
+      @doc.to_s.should == html
0
+    end
0
   end
0
   
0
 end