Merge pull request #33 from kaapa/elements_by_class_name

Finding an element by a class name

Author: seanlinsley

lib/arbre/element.rb

@@ -108,7 +108,7 @@ def get_elements_by_class_name(class_name)
       elements = ElementCollection.new
       children.each do |child|
         elements << child if child.class_list.include?(class_name)
-        elements.concat(child.get_elements_by_tag_name(tag_name))
+        elements.concat(child.get_elements_by_class_name(class_name))
       end
       elements
     end
@@ -129,7 +129,7 @@ def indent_level
     def each(&block)
       [to_s].each(&block)
     end
-    
+
     def inspect
       to_s
     end

lib/arbre/html/text_node.rb

@@ -22,6 +22,10 @@ def build(string)
         @content = string
       end
 
+      def class_list
+        []
+      end
+
       def tag_name
         nil
       end

spec/arbre/unit/element_finder_methods_spec.rb

@@ -61,6 +61,12 @@
       }.get_elements_by_class_name("my_class").size).to eq(0)
     end
 
+    it "should allow text nodes on tree" do
+      expect(arbre {
+        text_node "text"
+      }.get_elements_by_class_name("my_class").size).to eq(0)
+    end
+
     it "should return a child element" do
       html = arbre do
         div :class => "some_class"
@@ -95,7 +101,16 @@
       expect(elements[0]).to be_instance_of(Arbre::HTML::Div)
     end
 
-    # TODO: find children's children by class name
+    it "should return a grandchild element" do
+      html = arbre do
+        div :class => "some_class" do
+          div :class => "my_class"
+        end
+      end
+      elements = html.get_elements_by_class_name("my_class")
+      expect(elements.size).to eq(1)
+      expect(elements[0]).to be_instance_of(Arbre::HTML::Div)
+    end
 
   end
 end