metra_schedule 0.1

data/test/test_cacher.rb

@@ -0,0 +1,82 @@
+require 'test/unit'
+require 'fileutils'
+
+require File.join(File.dirname(__FILE__), "../lib", "metra")
+
+class TestLine < Test::Unit::TestCase
+
+  def cleanup_dir
+    FileUtils.rmtree(MetraSchedule::Cacher.new.cache_dir)
+  end
+
+  def test_init
+    assert_nothing_raised do
+      MetraSchedule::Cacher.new
+    end
+  end
+
+  def test_check_for_and_create_metra_cache_dir
+    cleanup_dir
+    c = MetraSchedule::Cacher.new
+    assert_equal(false, c.check_for_metra_cache_dir)
+    c.create_metra_cache_dir
+    assert_equal(true, c.check_for_metra_cache_dir)
+    cleanup_dir
+  end
+
+  def test_create_cache_dir_if_not_exists
+    cleanup_dir
+    c = MetraSchedule::Cacher.new
+    assert_equal(false, c.check_for_metra_cache_dir)
+    assert_equal(true, c.create_cache_dir_if_not_exists)
+    cleanup_dir
+  end
+
+  def test_check_if_line_cache_file_exists
+    cleanup_dir
+    c = MetraSchedule::Cacher.new
+    l = Metra.new.line(:up_nw) 
+    c.create_cache_dir_if_not_exists
+    assert_equal(false, c.line_exists?(l))
+    cleanup_dir
+  end
+
+  def test_create_engine_cache
+    cleanup_dir
+    c = MetraSchedule::Cacher.new
+    line = Metra.new.line(:up_nw) 
+    c.create_cache_dir_if_not_exists
+
+    stop1 = MetraSchedule::Stop.new :station => :barrington, :time => Time.parse('12:30')
+    stop2 = MetraSchedule::Stop.new :station => :arlington_heights, :time => Time.parse('12:30')
+    stop3 = MetraSchedule::Stop.new :station => :ogilve, :time => Time.parse('13:30')
+    train1 = MetraSchedule::Train.new :stops => [stop1, stop2, stop3], :direction => :outbound, :schedule => :weekday
+    train2 = MetraSchedule::Train.new :stops => [stop1, stop3], :direction => :outbound, :schedule => :weekday
+    train3 = MetraSchedule::Train.new :stops => [stop2, stop3], :direction => :outbound, :schedule => :weekday
+    line.engines = [train1, train2, train3]
+
+    assert_equal(true, c.persist_line(line))
+    assert_equal(line.engines.count, c.retrieve_line(line).engines.count)
+    cleanup_dir
+  end
+
+  def test_load_cache
+    cleanup_dir
+    c = MetraSchedule::Cacher.new
+    line = Metra.new.line(:up_nw) 
+    c.create_cache_dir_if_not_exists
+
+    stop1 = MetraSchedule::Stop.new :station => :barrington, :time => Time.parse('12:30')
+    stop2 = MetraSchedule::Stop.new :station => :arlington_heights, :time => Time.parse('12:30')
+    stop3 = MetraSchedule::Stop.new :station => :ogilve, :time => Time.parse('13:30')
+    train1 = MetraSchedule::Train.new :stops => [stop1, stop2, stop3], :direction => :outbound, :schedule => :weekday
+    train2 = MetraSchedule::Train.new :stops => [stop1, stop3], :direction => :outbound, :schedule => :weekday
+    train3 = MetraSchedule::Train.new :stops => [stop2, stop3], :direction => :outbound, :schedule => :weekday
+    line.engines = [train1, train2, train3]
+
+    assert_equal(true, MetraSchedule::Cacher.store_to_cache(line))
+    assert_equal(c.retrieve_line(line).engines.count, MetraSchedule::Cacher.load_from_cache(line).engines.count)
+    cleanup_dir
+  end
+
+end

data/test/test_line.rb

@@ -0,0 +1,253 @@
+require 'test/unit'
+require File.join(File.dirname(__FILE__), "../lib", "metra")
+
+class TestLine < Test::Unit::TestCase
+
+# INTEGRATION:
+#  def test_update_schedule
+#    line = Metra.new.line(:up_nw).outbound
+#    line.update_schedule
+#    assert_not_nil(line.trains)
+#  end
+
+  def test_initialize_with_line_name
+    assert_nothing_raised do
+      Metra.new.line(:up_nw)
+    end
+    assert_raise ArgumentError do
+      Metra.new.line("blah")
+    end
+  end
+
+  def test_initialize_with_incorrect_line_symbol
+    assert_raise ArgumentError do
+      Metra.new.line(:blah)
+    end
+  end
+
+  def test_has_name_and_url
+    line = Metra.new.line(:up_nw)
+    assert_equal("Union Pacific Northwest", line.name)
+    assert_equal("http://metrarail.com/metra/en/home/maps_schedules/metra_system_map/up-nw/schedule.full.html", line.url)
+  end
+
+  def test_direction
+    line = Metra.new.line(:up_nw)
+    assert_nothing_raised do
+      line.direction :outbound
+    end
+    assert_equal(:outbound, line.dir)
+    assert_nothing_raised do
+      line.direction :inbound
+    end
+    assert_equal(:inbound, line.dir)
+    assert_equal(line.direction.class, Metra.new.line(:up_nw).class)
+  end
+
+  def test_direction_incorrect_argument
+    assert_raise ArgumentError do
+      line = Metra.new.line(:up_nw).direction(:blah)
+    end
+  end
+
+  def test_outbound
+    line = Metra.new.line(:up_nw)
+    assert_equal(line.direction(:outbound), line.outbound)
+  end
+
+  def test_inbound
+    line = Metra.new.line(:up_nw)
+    assert_equal(line.direction(:inbound), line.inbound)
+  end
+
+  def test_schedule
+    line = Metra.new.line(:up_nw)
+    assert_nothing_raised do
+      line.schedule :weekday
+    end
+    assert_equal(:weekday, line.sched)
+    assert_nothing_raised do
+      line.schedule :saturday
+    end
+    assert_equal(:saturday, line.sched)
+    assert_nothing_raised do
+      line.schedule :sunday
+    end
+    assert_equal(:sunday, line.sched)
+    assert_nothing_raised do
+      line.schedule :holiday
+    end
+    assert_equal(:holiday, line.sched)
+    assert_equal(line.schedule.class, Metra.new.line(:up_nw).class)
+  end
+
+  def test_weekday
+    line = Metra.new.line(:up_nw)
+    assert_equal(line.schedule(:weekday), line.weekday)
+  end
+
+  def test_saturday
+    line = Metra.new.line(:up_nw)
+    assert_equal(line.schedule(:saturday), line.saturday)
+  end
+
+  def test_sunday
+    line = Metra.new.line(:up_nw)
+    assert_equal(line.schedule(:sunday), line.sunday)
+  end
+
+  def test_holiday
+    line = Metra.new.line(:up_nw)
+    assert_equal(line.schedule(:holiday), line.holiday)
+  end
+
+  def test_set_station
+    line = Metra.new.line(:up_nw)
+    line.set_station(:start, :barrington)
+    assert_equal(:barrington, line.start)
+    line.set_station(:destination, :barrington)
+    assert_equal(:barrington, line.destination)
+  end
+
+  def test_set_station_invalid_station
+    line = Metra.new.line(:up_nw)
+    assert_raises ArgumentError do
+      line.set_station(:start, :bleepy_bleepy)
+    end
+  end
+
+  def test_from
+    line = Metra.new.line(:up_nw)
+    assert_equal(line.set_station(:start, :barrington), line.from(:barrington))
+  end
+
+  def test_to
+    line = Metra.new.line(:up_nw)
+    assert_equal(line.set_station(:destination, :ogilve), line.to(:ogilve))
+  end
+
+  def test_at
+    line = Metra.new.line(:up_nw)
+    test_time = "12:30PM"
+    assert_nothing_raised do
+      line.at(test_time)
+      line.at(Time.now)
+    end
+    assert_equal(line.time, Time.parse(test_time))
+    assert_raises ArgumentError do
+      line.at(:bloopybloop)
+    end
+  end
+
+  def test_deduce_direction
+    line = Metra.new.line(:up_nw)
+
+    assert_equal(:outbound, line.from(:ogilve).to(:barrington).deduce_direction)
+    assert_equal(:inbound, line.from(:barrington).to(:ogilve).deduce_direction)
+  end
+
+  def test_trains_filter_by_station
+    line = Metra.new.line(:up_nw)
+
+    stop1 = MetraSchedule::Stop.new :station => :barrington, :time => Time.parse('12:30')
+    stop2 = MetraSchedule::Stop.new :station => :arlington_heights, :time => Time.parse('12:30')
+    stop3 = MetraSchedule::Stop.new :station => :ogilve, :time => Time.parse('13:30')
+    train1 = MetraSchedule::Train.new :stops => [stop1, stop2, stop3], :direction => :outbound, :schedule => :weekday
+    train2 = MetraSchedule::Train.new :stops => [stop1, stop3], :direction => :outbound, :schedule => :weekday
+    train3 = MetraSchedule::Train.new :stops => [stop2, stop3], :direction => :outbound, :schedule => :weekday
+    line.engines = [train1, train2, train3]
+
+    valid_trains = line.to(:barrington).from(:ogilve).trains
+    assert_equal([train1, train2], valid_trains)
+  end
+
+  def test_trains_filter_by_start
+    line = Metra.new.line(:up_nw)
+
+    stop1 = MetraSchedule::Stop.new :station => :arlington_heights, :time => Time.parse('12:30')
+    stop2 = MetraSchedule::Stop.new :station => :arlington_heights, :time => Time.parse('12:40')
+    stop3 = MetraSchedule::Stop.new :station => :ogilve, :time => Time.parse('13:30')
+    train1 = MetraSchedule::Train.new :stops => [stop1, stop3], :direction => :inbound, :schedule => :weekday
+    train2 = MetraSchedule::Train.new :stops => [stop2, stop3], :direction => :inbound, :schedule => :weekday
+    line.engines = [train1, train2]
+
+    valid_trains = line.to(:ogilve).from(:arlington_heights).at('12:35').trains
+    assert_equal([train2], valid_trains)
+  end
+
+  def test_trains_just_direction_and_time
+    line = Metra.new.line(:up_nw)
+
+    stop1 = MetraSchedule::Stop.new :station => :barrington, :time => Time.parse('12:30')
+    stop2 = MetraSchedule::Stop.new :station => :arlington_heights, :time => Time.parse('12:30')
+    stop3 = MetraSchedule::Stop.new :station => :ogilve, :time => Time.parse('13:30')
+    train1 = MetraSchedule::Train.new :stops => [stop1, stop2, stop3], :direction => :outbound, :schedule => :weekday
+    train2 = MetraSchedule::Train.new :stops => [stop1, stop3], :direction => :outbound, :schedule => :weekday
+    train3 = MetraSchedule::Train.new :stops => [stop2, stop3], :direction => :outbound, :schedule => :weekday
+    line.engines = [train1, train2, train3]
+
+    valid_trains = line.outbound.at('12:00').trains
+    assert_equal([train1, train2, train3], valid_trains)
+  end
+
+  def test_filter_by_direction
+    line = Metra.new.line(:up_nw)
+
+    stop1 = MetraSchedule::Stop.new :station => :barrington, :time => Time.parse('12:30')
+    stop2 = MetraSchedule::Stop.new :station => :ogilve, :time => Time.parse('13:30')
+
+    train1 = MetraSchedule::Train.new :direction => :outbound, :stops => [stop1, stop2], :schedule => :weekday
+    train2 = MetraSchedule::Train.new :direction => :inbound, :stops => [stop1, stop2], :schedule => :weekday
+    train3 = MetraSchedule::Train.new :direction => :outbound, :stops => [stop1, stop2], :schedule => :weekday
+    line.engines = [train1, train2, train3]
+
+    valid_trains = line.to(:barrington).from(:ogilve).trains
+    assert_equal([train1, train3], valid_trains)
+  end
+
+  def test_filter_by_schedule
+    line = Metra.new.line(:up_nw)
+
+    stop1 = MetraSchedule::Stop.new :station => :barrington, :time => Time.parse('12:30')
+    stop2 = MetraSchedule::Stop.new :station => :ogilve, :time => Time.parse('13:30')
+
+    train1 = MetraSchedule::Train.new :direction => :outbound, :stops => [stop1, stop2], :schedule => :weekday
+    train2 = MetraSchedule::Train.new :direction => :outbound, :stops => [stop1, stop2], :schedule => :holiday
+    train3 = MetraSchedule::Train.new :direction => :outbound, :stops => [stop1, stop2], :schedule => :saturday
+    train4 = MetraSchedule::Train.new :direction => :outbound, :stops => [stop1, stop2], :schedule => :sunday
+    line.engines = [train1, train2, train3, train4]
+
+    valid_weekday = line.to(:barrington).from(:ogilve).weekday.trains
+    assert_equal([train1], valid_weekday)
+    valid_saturday = line.to(:barrington).from(:ogilve).saturday.trains
+    assert_equal([train3], valid_saturday)
+    valid_holiday = line.to(:barrington).from(:ogilve).holiday.trains
+    assert_equal([train2, train4], valid_holiday)
+    valid_sunday = line.to(:barrington).from(:ogilve).sunday.trains
+    assert_equal([train2, train4], valid_sunday)
+  end
+
+  def test_no_engines
+    line = Metra.new.line(:up_nw)
+
+    assert_nothing_raised do
+      line.outbound.trains
+    end
+  end
+
+  def test_all_trains_on_line
+    line = Metra.new.line(:up_nw)
+
+    stop1 = MetraSchedule::Stop.new :station => :barrington, :time => Time.parse('12:30')
+    stop2 = MetraSchedule::Stop.new :station => :ogilve, :time => Time.parse('13:30')
+
+    train1 = MetraSchedule::Train.new :direction => :inbound, :stops => [stop1, stop2], :schedule => :weekday
+    train2 = MetraSchedule::Train.new :direction => :outbound, :stops => [stop1, stop2], :schedule => :holiday
+    train3 = MetraSchedule::Train.new :direction => :inbound, :stops => [stop1, stop2], :schedule => :saturday
+    train4 = MetraSchedule::Train.new :direction => :outbound, :stops => [stop1, stop2], :schedule => :sunday
+    line.engines = [train1, train2, train3, train4]
+
+    assert_equal([train1, train2, train3, train4], line.trains)
+  end
+
+end

data/test/test_metra.rb

@@ -0,0 +1,10 @@
+require 'test/unit'
+require File.join(File.dirname(__FILE__), "../lib", "metra")
+
+class TestMetra < Test::Unit::TestCase
+  def test_metra_init
+    assert_nothing_raised do
+      @m = Metra.new
+    end
+  end
+end

data/test/test_parser.rb

@@ -0,0 +1,137 @@
+require 'test/unit'
+require File.join(File.dirname(__FILE__), "../lib", "metra")
+
+class TestLine < Test::Unit::TestCase
+  include MetraSchedule::TrainData
+
+  def up_nw_stub
+    f = File.open(File.join(File.dirname(__FILE__), 'fixture/UP_NW.html'), 'r')
+    parser = MetraSchedule::Parser.new f
+    parser.line = LINES[:up_nw]
+    parser.scrape
+    parser
+  end
+
+  def test_init
+    assert_nothing_raised do
+      @@p = MetraSchedule::Parser.new 'http://blake.com'
+    end
+    assert_not_nil(up_nw_stub.html_doc)
+    assert_not_nil(@@p.html_doc)
+  end
+
+  def test_seperate_tables
+    p = up_nw_stub
+    assert_equal(2, p.tables.find_all {|t| t[:schedule] == :weekday}.count)
+    assert_equal(2, p.tables.find_all {|t| t[:schedule] == :saturday}.count)
+    assert_equal(2, p.tables.find_all {|t| t[:schedule] == :sunday}.count)
+  end
+
+  def test_sanitize
+    p = up_nw_stub
+    list = [1, 2, 3]
+    assert_equal(2, p.sanitize(list).count)
+  end
+
+  def test_train_count
+    p = up_nw_stub
+    assert_equal(16, p.train_count(p.tables[0][:tables][0]))
+    assert_equal(16, p.train_count(p.tables[0][:tables][1]))
+    assert_equal(12, p.train_count(p.tables[1][:tables][0]))
+    assert_equal(7, p.train_count(p.tables[2][:tables][0]))
+    assert_equal(16, p.train_count(p.tables[3][:tables][0]))
+    assert_equal(16, p.train_count(p.tables[3][:tables][1]))
+    assert_equal(1, p.train_count(p.tables[3][:tables][2]))
+    assert_equal(12, p.train_count(p.tables[4][:tables][0]))
+  end
+
+  def test_stop_count
+    p = up_nw_stub
+    assert_equal(23, p.stop_count(p.tables[0][:tables][0]))
+  end
+
+  def test_make_stop_inbound
+    p = up_nw_stub
+
+    node = p.tables[0][:tables][0].xpath("tbody[4]/tr/td[1]")[0]
+    expected = MetraSchedule::Stop.new :station => :crystal_lake, :time => Time.parse("4:47AM")
+    assert_equal(expected.station, p.make_stop(node, 3, :inbound).station)
+    assert_equal(expected.time, p.make_stop(node, 3, :inbound).time)
+  end
+
+  def test_make_stop_outbound
+    p = up_nw_stub
+
+    node = p.tables[3][:tables][0].xpath("tbody[4]/tr/td[1]")[0]
+    expected = MetraSchedule::Stop.new :station => :jefferson_park, :time => Time.parse("6:13AM")
+    assert_equal(expected.station, p.make_stop(node, 3, :outbound).station)
+    assert_equal(expected.time, p.make_stop(node, 3, :outbound).time)
+  end
+
+  def test_make_stop_nil
+    p = up_nw_stub
+
+    node = p.tables[0][:tables][0].xpath("tbody[1]/tr/td[1]")[0]
+    assert_nil(p.make_stop(node, 0, :inbound))
+  end
+
+  def test_make_stop_am
+    p = up_nw_stub
+
+    node = p.tables[0][:tables][1].xpath("tbody[18]/tr/td[5]")[0]
+    expected = MetraSchedule::Stop.new :station => :norwood_park, :time => Time.parse("11:57AM")
+    assert_equal(expected.station, p.make_stop(node, 17, :inbound).station)
+    assert_equal(expected.time, p.make_stop(node, 17, :inbound).time)
+  end
+
+  def test_make_stop_pm
+    p = up_nw_stub
+
+    node = p.tables[0][:tables][1].xpath("tbody[20]/tr/td[5]")[0]
+    expected = MetraSchedule::Stop.new :station => :jefferson_park, :time => Time.parse("12:01PM")
+    assert_equal(expected.station, p.make_stop(node, 19, :inbound).station)
+    assert_equal(expected.time, p.make_stop(node, 19, :inbound).time)
+  end
+
+  def test_find_stops
+    p = up_nw_stub
+
+    table = p.tables[0][:tables][0]
+    assert_equal(19, p.find_stops(table, 1, :inbound).count)
+  end
+
+  def test_find_stops_express
+    p = up_nw_stub
+
+    table = p.tables[0][:tables][0]
+    stops = p.find_stops(table, 5, :inbound)
+    assert_equal(Time.parse("6:43AM").to_f , stops[6].time.to_f)
+    assert_equal(:arlington_park, stops[6].station)
+    assert_equal(10, stops.count)
+  end
+
+  def test_make_trains_count
+    p = up_nw_stub
+    trains = p.make_trains
+    assert_equal(104, p.make_trains.count)
+  end
+
+  def test_find_train_num
+    p = up_nw_stub
+    train_602 = p.make_trains.first
+    train_632 = p.make_trains[15]
+
+    assert_equal(602, train_602.train_num)
+    assert_equal(632, train_632.train_num)
+  end
+
+  def test_find_bike_count
+    p = up_nw_stub
+    train_602 = p.make_trains.first
+    train_640 = p.make_trains[18]
+
+    assert_equal(nil, train_602.bike_limit)
+    assert_equal(12, train_640.bike_limit)
+  end
+
+end