wmata 0.1.0

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,21 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Jeremy McAnally
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,17 @@
+= wmata
+
+Description goes here.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 Jeremy McAnally. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,52 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "wmata"
+    gem.summary = %Q{A gem for the WMATA API}
+    gem.description = %Q{A gem for accessing the WMATA API}
+    gem.email = "jeremymcanally@gmail.com"
+    gem.homepage = "http://github.com/jm/wmata"
+    gem.authors = ["Jeremy McAnally"]
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/test_*.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "wmata #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/lib/resource.rb

@@ -0,0 +1,37 @@
+module WMATA
+  class Resource
+    class <<self
+      def get_all(params={})
+        url = WMATA.base_url % [service, endpoint, to_query_string(params)]
+        HTTParty.get(url).first.last.map {|values| new(values) }
+      end
+      
+      alias_method :find_all, :get_all
+    
+      def service(value=nil)
+        @service = value if value
+        @service || "#{self.name.capitalize}s"
+      end
+      
+      def endpoint(value=nil)
+        @endpoint = value if value
+        @endpoint || "#{self.name.capitalize}s"
+      end
+      
+      def to_query_string(params)
+        "&" + params.map {|k, v| "#{k.to_s}=#{v}"}.join("&")
+      end
+    end
+
+    attr_reader :attrs
+
+    def initialize(attrs={})
+      @attrs = attrs
+    end
+
+    def method_missing(m, *args)
+      camel_cased = m.to_s.gsub(/\/(.?)/) { "::#{$1.upcase}" }.gsub(/(?:^|_)(.)/) { $1.upcase }
+      @attrs[m.to_s] or @attrs[camel_cased] or super
+    end
+  end
+end

data/lib/resources/elevator_incident.rb

@@ -0,0 +1,25 @@
+module WMATA
+  class ElevatorIncident < Resource
+    service "Incidents"
+    endpoint "ElevatorIncidents"
+    
+    def self.get_by_station(affected_station)
+      @incidents ||= get_all
+      @incidents.select {|i| i.station_code == affected_station.to_s }.pop
+    end
+    
+    def affected_station
+      Station.get(@attrs['StationCode'])
+    end
+
+    alias_method :station, :affected_station
+    
+    def date_out_of_service
+      Time.parse(@attrs['DateOutOfServ'])
+    end
+    
+    def date_updated
+      Time.parse(@attrs['DateUpdated'])
+    end
+  end
+end

data/lib/resources/line.rb

@@ -0,0 +1,54 @@
+module WMATA
+  class Line < Resource
+    service "Rail"
+    endpoint "JLines"
+    
+    class <<self
+      alias_method :get_all_without_memoize, :get_all
+        
+      # Memoize this since (a) there's no way to ask for just one line and
+      # (b) they're unlikely to change while we're doing a request.
+      def get_all(params)
+        @lines ||= get_all_without_memoize(params)
+      end
+    end
+    
+    def start_station
+      @start_station ||= Station.get(@attrs['StartStationCode'])
+    end
+    
+    def end_station
+      @end_station ||= Station.get(@attrs['EndStationCode'])
+    end
+    
+    def internal_destinations
+      [@attrs['InternalDestination1'], @attrs['InternalDestination2']].compact.map do |s| 
+        Station.get(s)
+      end
+    end
+    
+    def rail_incidents
+      @incidents ||= RailIncident.get_by_line(self)
+    end
+    
+    alias_method :incidents, :rail_incidents
+    
+    def route
+      Station.get_on_line(code)
+    end
+    
+    alias_method :stations, :route
+    
+    def get(code)
+      get_all.select {|l| l.code == code}.pop
+    end
+    
+    def code
+      @attrs['LineCode']
+    end
+    
+    def to_s
+      code
+    end
+  end
+end

data/lib/resources/path_segment.rb

@@ -0,0 +1,18 @@
+module WMATA
+  class PathSegment < Resource
+    service "Rail"
+    endpoint "JPath"
+    
+    def station
+      @station ||= Station.get(@attrs['StationCode'])
+    end
+    
+    def line
+      @line ||= Line.get(@attrs['LineCode'])
+    end
+    
+    def index
+      @attrs['SeqNum']
+    end
+  end
+end

data/lib/resources/prediction.rb

@@ -0,0 +1,28 @@
+module WMATA
+  class Prediction < Resource
+    service "StationPrediction"
+    
+    def self.predict_for(station_code)
+      url = WMATA.base_url % [service, "GetPrediction/#{station_code}", ""]
+      HTTParty.get(url).first.last.map {|values| new(values) }
+    end
+    
+    def location
+      @location ||= Station.get(@attrs['LocationCode'])
+    end
+    
+    alias_method :station, :location
+    
+    def destination
+      @destination ||= Station.get(@attrs['DestinationCode'])
+    end
+    
+    def line_code
+      @attrs['Line']
+    end
+    
+    def line
+      @line ||= Line.get(@attrs['Line'])
+    end
+  end
+end

data/lib/resources/rail_incident.rb

@@ -0,0 +1,23 @@
+module WMATA
+  class RailIncident < Resource
+    service "Incidents"
+    endpoint "Incidents"
+
+    def self.get_by_line(line)
+      @incidents ||= get_all
+      @incidents.select {|i| i.line_codes_affected.include?(line.to_s)}
+    end
+    
+    def date_updated
+      Time.parse(@attrs['DateUpdated'])
+    end
+    
+    def line_codes_affected
+      @attrs['LinesAffected'].split(";").reject {|s| s.empty? || s.nil?}
+    end
+    
+    def lines_affected
+      @lines_affected = line_codes_affected.map {|l| Line.get(l)}
+    end
+  end
+end

data/lib/resources/station.rb

@@ -0,0 +1,48 @@
+module WMATA
+  class Station < Resource
+    service "Rail"
+    endpoint "JStations"
+    
+    SYMBOL_TO_LINES_MAP = {
+      :red => "RD",
+      :blue => "BL",
+      :orange => "OR", 
+      :green => "GR", 
+      :yellow => "YE"
+    }
+    
+    def self.get_on_line(line)
+      line = SYMBOL_TO_LINES_MAP[line] if line.is_a?(Symbol)
+      get_all("LineCode" => line.to_s)
+    end
+    
+    def self.get(code)
+      url = WMATA.base_url % [service, "JStationInfo", to_query_string("StationCode" => code)]
+      new(HTTParty.get(url))
+    end
+    
+    def codes
+      [@attrs['Code'], @attrs['StationTogether1'], @attrs['StationTogether2']].compact
+    end
+    
+    def line_codes
+      [@attrs['LineCode1'], @attrs['LineCode2'], @attrs['LineCode3'], @attrs['LineCode4']].compact
+    end
+    
+    def lines
+      @lines ||= line_codes.map {|l| Line.get(l)}
+    end
+    
+    def predictions
+      @predictions ||= Prediction.predict_for(self)
+    end
+    
+    def elevator_incidents
+      @elevator_incidents ||= ElevatorIncident.get_by_station(self)
+    end
+    
+    def to_s
+      @attrs['Code']
+    end
+  end
+end

data/lib/resources/station_entrance.rb

@@ -0,0 +1,14 @@
+module WMATA
+  class StationEntrance < Resource
+    service "Rail"
+    endpoint "JStationEntrances"
+    
+    def station_codes
+      [@attrs['StationCode1'], @attrs['StationCode2']].compact
+    end
+    
+    def station
+      @station ||= Station.get(station_codes.first)
+    end
+  end
+end

data/lib/wmata.rb

@@ -0,0 +1,68 @@
+$:.unshift(File.dirname(__FILE__))
+
+require 'rubygems'
+require 'httparty'
+require 'forwardable'
+
+require 'resource'
+require 'resources/line'
+require 'resources/station'
+require 'resources/rail_incident'
+require 'resources/elevator_incident'
+require 'resources/prediction'
+require 'resources/path_segment'
+require 'resources/station_entrance'
+
+module WMATA
+  BASE_URL = "http://api.wmata.com/%s.svc/json/%s?api_key=%s%s"
+  
+  class <<self
+    attr_accessor :api_key
+        
+    def base_url
+      BASE_URL.dup % ["%s", "%s", @api_key, "%s"]    
+    end
+    
+    def lines
+      Line.get_all
+    end
+    
+    def stations
+      Station.get_all
+    end
+    
+    def station(code)
+      Station.get(code)
+    end
+    
+    def stations_on_line(code)
+      Station.get_on_line(code)
+    end
+    
+    def predict_for(station="All")
+      Prediction.predict_for(station)
+    end
+    
+    def rail_incidents
+      RailIncident.get_all
+    end
+    
+    alias_method :incidents, :rail_incidents
+    
+    def elevator_incidents
+      ElevatorIncident.get_all
+    end
+    
+    def build_path(from, to)
+      PathSegment.get_all("FromStationCode" => from, "ToStationCode" => to)
+    end
+    
+    def entrances(from={})
+      params = {:lat => 0, :lon => 0, :radius => 500}.merge(from)
+      StationEntrance.get_all(params)
+    end
+    
+    alias_method :station_entrances, :entrances
+    alias_method :entrances_near, :entrances
+  end
+end

data/test/helper.rb

@@ -0,0 +1,14 @@
+require 'rubygems'
+require 'test/unit'
+require 'flexmock'
+require 'flexmock/test_unit'
+require 'ostruct'
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'wmata'
+
+class Test::Unit::TestCase
+  
+end

data/test/test_elevator_incident.rb

@@ -0,0 +1,28 @@
+require 'helper'
+
+class TestElevatorIncident < Test::Unit::TestCase
+  def test_get_by_station
+    fake_incidents = []
+    5.times {|i| fake_incidents << OpenStruct.new(:station_code => "A#{i}")}
+    flexmock(WMATA::ElevatorIncident).should_receive(:get_all).and_return(fake_incidents)
+    
+    assert_equal "A2", WMATA::ElevatorIncident.get_by_station("A2").station_code
+  end
+  
+  def test_affected_station
+    flexmock(WMATA::Station).should_receive(:get).and_return(OpenStruct.new(:name => "Winner"))
+    incident = WMATA::ElevatorIncident.new("StationCode" => "A4")
+    
+    assert_equal "Winner", incident.affected_station.name
+  end
+  
+  def test_date_out_of_service
+    incident = WMATA::ElevatorIncident.new("DateOutOfServ" => "2010-07-27T00:00:00")
+    assert_equal Time.parse("07/27/2010"), incident.date_out_of_service
+  end
+  
+  def test_date_updated
+    incident = WMATA::ElevatorIncident.new("DateUpdated" => "2010-07-27T00:00:00")
+    assert_equal Time.parse("07/27/2010"), incident.date_updated    
+  end
+end

data/test/test_line.rb

@@ -0,0 +1,61 @@
+class TestLine < Test::Unit::TestCase
+  def test_start_station
+    flexmock(WMATA::Station).should_receive(:get).and_return(OpenStruct.new(:name => "Winner"))
+    line = WMATA::Line.new("StartStationCode" => "A4")
+    
+    assert_equal "Winner", line.start_station.name
+  end
+  
+  def test_end_station
+    flexmock(WMATA::Station).should_receive(:get).and_return(OpenStruct.new(:name => "Winner"))
+    line = WMATA::Line.new("EndStationCode" => "A4")
+    
+    assert_equal "Winner", line.end_station.name    
+  end
+  
+  def test_internal_destinations
+    flexmock(WMATA::Station).should_receive(:get).with("A1").and_return(OpenStruct.new(:name => "Winner"))
+    flexmock(WMATA::Station).should_receive(:get).with("A2").and_return(OpenStruct.new(:name => "Failure"))
+    line = WMATA::Line.new("InternalDestination1" => "A1", "InternalDestination2" => "A2")
+    
+    assert_equal ["Failure", "Winner"], line.internal_destinations.map {|d| d.name}.sort
+  end
+  
+  def test_internal_destinations_with_partial_set
+    flexmock(WMATA::Station).should_receive(:get).with("A1").and_return(OpenStruct.new(:name => "Winner"))
+    line = WMATA::Line.new("InternalDestination1" => "A1")
+    
+    assert_equal ["Winner"], line.internal_destinations.map {|d| d.name}.sort    
+  end
+  
+  def test_rail_incidents
+    flexmock(WMATA::RailIncident).should_receive(:get_by_line).and_return([OpenStruct.new(:name => "Winner")])
+    line = WMATA::Line.new("LineCode" => "RD")
+    
+    assert_equal ["Winner"], line.rail_incidents.map(&:name)
+  end
+  
+  def test_route
+    flexmock(WMATA::Station).should_receive(:get_on_line).and_return([OpenStruct.new(:name => "Winner")])
+    line = WMATA::Line.new("LineCode" => "RD")
+    
+    assert_equal ["Winner"], line.route.map(&:name)
+  end
+  
+  def test_get
+    flexmock(WMATA::Station).should_receive(:get_on_line).and_return([OpenStruct.new(:name => "Winner")])
+    line = WMATA::Line.new("LineCode" => "RD")
+    
+    assert_equal ["Winner"], line.route.map(&:name)    
+  end
+  
+  def test_code
+    line = WMATA::Line.new("LineCode" => "RD")
+    assert_equal "RD", line.code
+  end
+  
+  def test_to_s
+    line = WMATA::Line.new("LineCode" => "RD")
+    assert_equal "RD", line.to_s    
+  end
+end

data/test/test_path_segment.rb

@@ -0,0 +1,22 @@
+require 'helper'
+
+class TestPathSegment < Test::Unit::TestCase
+  def test_station
+    flexmock(WMATA::Station).should_receive(:get).and_return(OpenStruct.new(:name => "Winner"))
+    segment = WMATA::PathSegment.new("StationCode" => "A4")
+    
+    assert_equal "Winner", segment.station.name
+  end
+  
+  def test_line
+    flexmock(WMATA::Line).should_receive(:get).and_return(OpenStruct.new(:name => "Winner"))
+    segment = WMATA::PathSegment.new("LineCode" => "A4")
+    
+    assert_equal "Winner", segment.line.name
+  end
+  
+  def test_index
+    segment = WMATA::PathSegment.new("SeqNum" => "1")
+    assert_equal "1", segment.index
+  end
+end

data/test/test_prediction.rb

@@ -0,0 +1,36 @@
+require 'helper'
+
+class TestPrediction< Test::Unit::TestCase
+  def setup
+    WMATA.api_key = "1234"
+    @fake = OpenStruct.new(:name => "Winner")
+    @prediction = WMATA::Prediction.new("LocationCode" => "A4", "DestinationCode" => "A3", "Line" => "RD")
+  end
+  
+  def test_predict_for
+    flexmock(HTTParty).should_receive(:get).with("http://api.wmata.com/StationPrediction.svc/json/GetPrediction/A4?api_key=1234").and_return([[{"Things" => 1234}]])
+    prediction = WMATA::Prediction.predict_for("A4")
+  end
+  
+  def test_location
+    flexmock(WMATA::Station).should_receive(:get).and_return(@fake)
+    
+    assert_equal @fake, @prediction.location
+  end
+  
+  def test_destination
+    flexmock(WMATA::Station).should_receive(:get).and_return(@fake)
+    
+    assert_equal @fake, @prediction.destination 
+  end
+  
+  def test_line_code
+    assert_equal "RD", @prediction.line_code 
+  end
+  
+  def test_line
+    flexmock(WMATA::Line).should_receive(:get).and_return(@fake)
+    
+    assert_equal @fake, @prediction.line
+  end
+end

data/test/test_rail_incident.rb

@@ -0,0 +1,28 @@
+require 'helper'
+
+class TestRailIncident < Test::Unit::TestCase
+  def test_get_by_station
+    fake_incidents = []
+    5.times {|i| fake_incidents << OpenStruct.new(:line_codes_affected => ["RD", "BL#{i}"])}
+    flexmock(WMATA::RailIncident).should_receive(:get_all).and_return(fake_incidents)
+    
+    assert_equal ["BL2", "RD"], WMATA::RailIncident.get_by_line("RD")[2].line_codes_affected.sort
+  end
+  
+  def test_line_codes_affected
+    incident = WMATA::RailIncident.new("LinesAffected" => "RD;BL;OR")
+    assert_equal ["BL", "OR", "RD"], incident.line_codes_affected.sort
+  end
+  
+  def test_lines_affected
+    flexmock(WMATA::Line).should_receive(:get).and_return(WMATA::Line.new("Code" => "RD"))
+    incident = WMATA::RailIncident.new("LinesAffected" => "RD;BL;OR")
+    
+    assert_equal 3, incident.lines_affected.length
+  end
+  
+  def test_date_updated
+    incident = WMATA::RailIncident.new("DateUpdated" => "2010-07-27T00:00:00")
+    assert_equal Time.parse("07/27/2010"), incident.date_updated    
+  end
+end

data/test/test_resource.rb

@@ -0,0 +1,42 @@
+require 'helper'
+
+class TestResource < Test::Unit::TestCase
+  class Faker < WMATA::Resource
+  end
+  
+  def setup
+    @fake = Faker.new("method1" => "win", "MethodTwo" => "epic")
+  end
+  
+  def test_method_mapping
+    assert_nothing_raised do
+      assert_equal "win", @fake.method1
+    end
+  end
+  
+  def test_method_mapping_with_camel_case
+    assert_nothing_raised do
+      assert_equal "epic", @fake.method_two
+    end
+  end
+  
+  def test_method_missing_fails_if_missing
+    assert_raises(NoMethodError) do
+      @fake.whateva
+    end
+  end
+  
+  def test_service_set
+    Faker.service "Rail"
+    assert_equal "Rail", Faker.service
+  end
+  
+  def test_endpoint_set
+    Faker.endpoint "Rail"
+    assert_equal "Rail", Faker.endpoint
+  end
+  
+  def test_to_query_string
+    assert_equal "&things=yes&whatever=no", Faker.to_query_string(:things => "yes", :whatever => "no")
+  end
+end

data/test/test_station.rb

@@ -0,0 +1,66 @@
+require 'helper'
+
+class TestStation < Test::Unit::TestCase
+  def test_get_on_line
+    flexmock(WMATA::Station).should_receive(:get_all).with("LineCode" => "RD")
+    WMATA::Station.get_on_line("RD")
+  end
+  
+  def test_get_on_line_with_symbol
+    flexmock(WMATA::Station).should_receive(:get_all).with("LineCode" => "RD")
+    WMATA::Station.get_on_line(:red)    
+  end
+  
+  def test_get
+    flexmock(HTTParty).should_receive(:get).with("http://api.wmata.com/Rail.svc/json/JStationInfo?api_key=1234&StationCode=A4").and_return({"Things" => "1234"})
+    station = WMATA::Station.get("A4")
+    
+    assert_equal "1234", station.things
+  end
+  
+  def test_codes
+    station = WMATA::Station.new("Code" => "A1", "StationTogether1" => "C2", "StationTogether2" => "D1")
+    assert_equal ["A1", "C2", "D1"], station.codes.sort
+  end
+  
+  def test_codes_with_partial_set
+    station = WMATA::Station.new("Code" => "A1", "StationTogether2" => "D1")
+    assert_equal ["A1", "D1"], station.codes.sort    
+  end
+  
+  def test_line_codes
+    station = WMATA::Station.new("LineCode1" => "RD", "LineCode2" => "BL", "LineCode3" => "GR", "LineCode4" => "OR")
+    assert_equal ["BL", "GR", "OR", "RD"], station.line_codes.sort
+  end
+  
+  def test_line_codes_with_partial_set
+    station = WMATA::Station.new("LineCode1" => "RD", "LineCode3" => "GR", "LineCode4" => "OR")
+    assert_equal ["GR", "OR", "RD"], station.line_codes.sort    
+  end
+  
+  def test_lines
+    flexmock(WMATA::Line).should_receive(:get).with("RD").and_return(OpenStruct.new("code" => "RD"))
+    station = WMATA::Station.new("LineCode1" => "RD")
+    
+    assert_equal "RD", station.lines.first.code
+  end
+  
+  def test_predictions
+    flexmock(WMATA::Prediction).should_receive(:predict_for).and_return([])
+    station = WMATA::Station.new("Code" => "A1")
+    
+    assert_equal [], station.predictions
+  end
+  
+  def test_elevator_incidents
+    flexmock(WMATA::ElevatorIncident).should_receive(:get_by_station).and_return([])
+    station = WMATA::Station.new("Code" => "A1")
+    
+    assert_equal [], station.elevator_incidents  
+  end
+  
+  def test_to_s
+    station = WMATA::Station.new("Code" => "A1")
+    assert_equal "A1", station.to_s
+  end
+end

data/test/test_station_entrance.rb

@@ -0,0 +1,13 @@
+require 'helper'
+
+class TestStationEntrance < Test::Unit::TestCase
+  def test_station_codes
+    station = WMATA::StationEntrance.new("StationCode1" => "A1", "StationCode2" => "A2")
+    assert_equal ["A1","A2"], station.station_codes.sort
+  end
+  
+  def test_station_codes_with_partial_set
+    station = WMATA::StationEntrance.new("StationCode2" => "A2")
+    assert_equal ["A2"], station.station_codes.sort 
+  end
+end

data/test/test_wmata.rb

@@ -0,0 +1,66 @@
+require 'helper'
+
+class TestWmata < Test::Unit::TestCase
+  def setup
+    flexmock(WMATA).should_receive(:api_key).and_return("1234")
+  end
+  
+  def test_lines
+    mock_resource(WMATA::Line)
+    WMATA.lines
+  end
+  
+  def test_stations
+    mock_resource(WMATA::Station)
+    WMATA.stations
+  end
+  
+  def test_station
+    flexmock(WMATA::Station).should_receive(:get).with("A2")
+    WMATA.station("A2")
+  end
+  
+  def test_stations_on_line
+    flexmock(WMATA::Station).should_receive(:get_on_line).with(:red)
+    WMATA.stations_on_line(:red)
+  end
+  
+  def test_predict_for
+    flexmock(WMATA::Prediction).should_receive(:predict_for).with("A7")
+    WMATA.predict_for("A7")
+  end
+  
+  def test_predict_for_with_default
+    flexmock(WMATA::Prediction).should_receive(:predict_for).with("All")
+    WMATA.predict_for
+  end
+  
+  def test_rail_incidents
+    mock_resource(WMATA::RailIncident)
+    WMATA.rail_incidents
+  end
+  
+  def test_elevator_incidents
+    mock_resource(WMATA::ElevatorIncident)
+    WMATA.elevator_incidents    
+  end
+  
+  def test_build_path
+    flexmock(WMATA::PathSegment).should_receive(:get_all).with("FromStationCode" => "37.80", "ToStationCode" => "88.7")
+    WMATA.build_path("37.80", "88.7")
+  end
+  
+  def test_entrances
+    flexmock(WMATA::StationEntrance).should_receive(:get_all).with(:lat => 0, :lon => 0, :radius => 500)
+    WMATA.entrances
+  end
+  
+  def test_entrances_with_specifics
+    flexmock(WMATA::StationEntrance).should_receive(:get_all).with(:lat => 39.0, :lon => 0, :radius => 2000)
+    WMATA.entrances(:lat => 39.0, :radius => 2000)
+  end
+  
+  def mock_resource(resource_class)
+    flexmock(resource_class).should_receive(:get_all)
+  end
+end

metadata

@@ -0,0 +1,96 @@
+--- !ruby/object:Gem::Specification 
+name: wmata
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Jeremy McAnally
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-08-16 00:00:00 -04:00
+default_executable: 
+dependencies: []
+
+description: A gem for accessing the WMATA API
+email: jeremymcanally@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- lib/resource.rb
+- lib/resources/elevator_incident.rb
+- lib/resources/line.rb
+- lib/resources/path_segment.rb
+- lib/resources/prediction.rb
+- lib/resources/rail_incident.rb
+- lib/resources/station.rb
+- lib/resources/station_entrance.rb
+- lib/wmata.rb
+- test/helper.rb
+- test/test_elevator_incident.rb
+- test/test_line.rb
+- test/test_path_segment.rb
+- test/test_prediction.rb
+- test/test_rail_incident.rb
+- test/test_resource.rb
+- test/test_station.rb
+- test/test_station_entrance.rb
+- test/test_wmata.rb
+has_rdoc: true
+homepage: http://github.com/jm/wmata
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: A gem for the WMATA API
+test_files: 
+- test/helper.rb
+- test/test_elevator_incident.rb
+- test/test_line.rb
+- test/test_path_segment.rb
+- test/test_prediction.rb
+- test/test_rail_incident.rb
+- test/test_resource.rb
+- test/test_station.rb
+- test/test_station_entrance.rb
+- test/test_wmata.rb