tenderlove · November 26, 2008 18:56
diff --git a/gistfile1.rb b/gistfile1.rb
 require 'rubygems'
 require 'nokogiri'

 class Nokogiri::XML::Node
  def method_missing name, *args, &block
    if args.empty?
      list = xpath("./#{name}")
    elsif args.first.is_a? Hash
      hash = args.first
      if hash[:css]
        list = css("#{name}#{hash[:css]}")
      elsif hash[:xpath]
        conds = Array(hash[:xpath]).collect{|j| "[#{j}]"}
        list = xpath("./#{name}#{conds}")
      end
    else
      list = css("#{name}#{args.first}")
    end
    list.length == 1 ? list.first : list
  end
 end

 HTML = <<-eohtml
 <html>
  <body>
    <ul>
      <li class='red'>one</li>
      <li class='blue'>two</li>
    </ul>
    <div>
      one
      <div>div two</div>
    </div>
  </body>
 </html>
 eohtml

 describe "sweetness" do
  attr_reader :doc
  
  before(:each) do
    @doc = Nokogiri(HTML.dup)
  end
  
  it "allows chaining of tag names" do
    doc.html.body.ul.li.first.text.should == "one"
  end
  
  it "allows string args for css selectors" do
    doc.html.body.ul.li(".blue").text.should == "two"
  end

  it "doesn't go too deep. that is what she said." do
    doc.html.body.div.div.text.should == "div two"
  end
  
  it "allows hash args for css selectors" do
    doc.html.body.ul.li(:css => ".blue").text.should == "two"
  end
  
  it "works with xpath stuff that I don't really get" do
    doc.html.body.ul.li(:xpath => "position()=2").text.should == "two"
    doc.html.body.ul.li(:xpath => ["contains(text(),'o')"]).first.text.should == "one"
    doc.html.body.ul.li(:xpath => ["contains(text(),'o')","contains(text(),'t')"]).text.should == "two"
  end
 end
	require 'rubygems'
	require 'nokogiri'

	class Nokogiri::XML::Node
	def method_missing name, *args, &block
	if args.empty?
	list = xpath("./#{name}")
	elsif args.first.is_a? Hash
	hash = args.first
	if hash[:css]
	list = css("#{name}#{hash[:css]}")
	elsif hash[:xpath]
	conds = Array(hash[:xpath]).collect{\|j\| "[#{j}]"}
	list = xpath("./#{name}#{conds}")
	end
	else
	list = css("#{name}#{args.first}")
	end
	list.length == 1 ? list.first : list
	end
	end

	HTML = <<-eohtml
	<html>
	<body>
	<ul>
	<li class='red'>one</li>
	<li class='blue'>two</li>
	</ul>
	<div>
	one
	<div>div two</div>
	</div>
	</body>
	</html>
	eohtml

	describe "sweetness" do
	attr_reader :doc

	before(:each) do
	@doc = Nokogiri(HTML.dup)
	end

	it "allows chaining of tag names" do
	doc.html.body.ul.li.first.text.should == "one"
	end

	it "allows string args for css selectors" do
	doc.html.body.ul.li(".blue").text.should == "two"
	end

	it "doesn't go too deep. that is what she said." do
	doc.html.body.div.div.text.should == "div two"
	end

	it "allows hash args for css selectors" do
	doc.html.body.ul.li(:css => ".blue").text.should == "two"
	end

	it "works with xpath stuff that I don't really get" do
	doc.html.body.ul.li(:xpath => "position()=2").text.should == "two"
	doc.html.body.ul.li(:xpath => ["contains(text(),'o')"]).first.text.should == "one"
	doc.html.body.ul.li(:xpath => ["contains(text(),'o')","contains(text(),'t')"]).text.should == "two"
	end
	end