shakespeare_analyzer.rb

require 'nokogiri'
require 'net/http'

class ShakespeareAnalyzer
  def initialize(url)
    @url = url
  end

  def analyze
    speech_elements.inject({}) do |result, element|
      character = character_from(element)
      count = count_from(element)
      previous_count = result[character] || 0
      result.merge(character => previous_count + count)
    end
  end

  private

  def speech_elements
    document.css('SPEECH')
  end

  def character_from(element)
    element.at('SPEAKER').text
  end

  def count_from(element)
    element.css('LINE').length
  end

  def document
    Nokogiri::XML.parse(xml)
  end

  def xml
    Net::HTTP.get(host, path)
  end

  def host
    uri.host
  end

  def path
    uri.path
  end

  def uri
    URI.parse(@url)
  end
end

shakespeare_analyzer_spec.rb

require 'rspec'
require_relative '../shakespeare_analyzer'

describe ShakespeareAnalyzer do
  context '#analyze' do
    it 'counts lines for each speaker' do
      xml = <<-XML
        <PLAY>
          <SPEECH>
            <SPEAKER>Macbeth</SPEAKER>
            <LINE>One</LINE>
            <LINE>Two</LINE>
          </SPEECH>
          <SPEECH>
            <SPEAKER>Other Guy</SPEAKER>
            <LINE>One</LINE>
            <LINE>Two</LINE>
            <LINE>Three</LINE>
          </SPEECH>
        </PLAY>
      XML
      analyze(xml).should eq({
        'Macbeth' => 2,
        'Other Guy' => 3
      })
    end

    it 'combines counts for multiple speech blocks per speaker' do
      xml = <<-XML
        <PLAY>
          <SPEECH>
            <SPEAKER>Macbeth</SPEAKER>
            <LINE>One</LINE>
            <LINE>Two</LINE>
          </SPEECH>
          <SPEECH>
            <SPEAKER>Macbeth</SPEAKER>
            <LINE>Three</LINE>
          </SPEECH>
        </PLAY>
      XML

      analyze(xml).should eq({
        'Macbeth' => 3
      })
    end

    def analyze(xml)
      url = 'http://example.com/somecoolxml.xml'
      Net::HTTP.
        stub(:get).
        with('example.com', '/somecoolxml.xml').
        and_return(xml)
      ShakespeareAnalyzer.new(url).analyze
    end
  end
end