ticket_sales 0.1.1

data/test/ticket_sales/ticket/route_test.rb

@@ -0,0 +1,50 @@
+require 'test_helper'
+
+class TestRoute  < MiniTest::Unit::TestCase
+  def setup
+    @haskovo_sofia = TicketSales::Route.new 'Haskovo', 'Sofia', 230
+    @sofia_ruse = TicketSales::Route.new 'Sofia', 'Ruse', 400
+    @haskovo_ruse = TicketSales::Route.new 'Haskovo', 'Ruse', 630, ['Sofia']
+    @ruse_sofia = TicketSales::Route.new 'Ruse', 'Sofia', 400
+    @burgas_haskovo = TicketSales::Route.new 'Burgas', 'Haskovo', 240
+  end
+
+  # Tests
+
+  def test_make_a_direct_route
+    assert_equal 'Haskovo', @haskovo_sofia.from
+    assert_equal 'Sofia', @haskovo_sofia.to
+    assert_equal 230, @haskovo_sofia.distance
+  end
+
+  def test_make_a_indirect_route
+    assert_equal 'Haskovo', @haskovo_ruse.from
+    assert_equal 'Ruse', @haskovo_ruse.to
+    assert_equal 630, @haskovo_ruse.distance
+    assert @haskovo_ruse.stops.include?('Sofia')
+  end
+
+  def test_routes_can_be_combined
+    combined_route = @haskovo_sofia.combine @sofia_ruse
+
+    assert_equal 'Haskovo', combined_route.from
+    assert_equal 'Ruse', combined_route.to
+    assert_equal @haskovo_sofia.distance + @sofia_ruse.distance, combined_route.distance
+    assert combined_route.stops.include?('Sofia')
+  end
+
+  def test_combine_with_indirect_routes
+    combined_route = @burgas_haskovo.combine @haskovo_ruse
+
+    assert_equal 'Burgas', combined_route.from
+    assert_equal 'Ruse', combined_route.to
+    assert_equal @burgas_haskovo.distance + @haskovo_ruse.distance, combined_route.distance
+    assert ['Haskovo', 'Sofia'].all? { |stop| combined_route.stops.include?(stop) }
+  end
+
+  def test_combine_doesnt_work_with_incorrect_routes
+    assert_raises(RuntimeError) { @haskovo_sofia.combine @ruse_sofia }
+    assert_raises(RuntimeError) { @burgas_haskovo.combine @sofia_ruse }
+    assert_raises(RuntimeError) { @haskovo_ruse.combine @burgas_haskovo }
+  end
+end

data/test/ticket_sales/ticket/ticket_test.rb

@@ -0,0 +1,62 @@
+require 'test_helper'
+
+class TestTicket  < MiniTest::Unit::TestCase
+  def setup
+    @haskovo_sofia = TicketSales::Ticket.build 'Haskovo', 'Sofia', 230,Time.new(2013, 3, 3, 15),
+      Time.new(2013, 3, 3, 18), '20'.to_d, 'fast, class A, cheap'
+    @sofia_ruse = TicketSales::Ticket.build 'Sofia', 'Ruse', 400,Time.new(2013, 3, 3, 19),
+      Time.new(2013,3,4), '25'.to_d, 'long'
+    @ruse_burgas = TicketSales::Ticket.build 'Ruse', 'Burgas', 500, Time.new(2013, 3, 4, 5),
+      Time.new(2013, 3, 4, 13), '35'.to_d, ''
+    @burgas_moscow = TicketSales::Ticket.build 'Burgas', 'Moscow', 1900, Time.new(2013, 3, 4, 20),
+      Time.new(2013, 3, 4, 22), '120'.to_d, 'airplane'
+  end
+
+  # Tests
+
+  def test_ticket_build
+    london_moscow = TicketSales::Ticket.build 'London', 'Moscow', 1900, Time.new(2013, 3, 4, 20),
+      Time.new(2013, 3, 4, 22), '100'.to_d, 'airplane'
+
+    assert_equal 'London', london_moscow.from
+    assert_equal 'Moscow', london_moscow.to
+    assert_equal '100'.to_d, london_moscow.price
+    assert_equal 'airplane', london_moscow.features
+    assert_equal Time.new(2013, 3, 4, 20), london_moscow.departure_time
+    assert_equal Time.new(2013, 3, 4, 22), london_moscow.arrival_time
+    assert_equal '2'.to_d, london_moscow.travel_length
+    assert_equal 1900, london_moscow.distance
+  end
+
+  def test_combine_ticket
+    combined_ticket = @haskovo_sofia.combine @sofia_ruse
+
+    assert_equal @haskovo_sofia.from, combined_ticket.from
+    assert_equal @sofia_ruse.to, combined_ticket.to
+    assert_equal @haskovo_sofia.price + @sofia_ruse.price, combined_ticket.price
+    assert_equal @haskovo_sofia.features + ', ' + @sofia_ruse.features, combined_ticket.features
+    assert_equal @haskovo_sofia.departure_time, combined_ticket.departure_time
+    assert_equal @sofia_ruse.arrival_time, combined_ticket.arrival_time
+    assert_equal (@sofia_ruse.arrival_time - @haskovo_sofia.departure_time).to_d / 3600,
+      combined_ticket.travel_length
+
+    another = combined_ticket.combine @ruse_burgas
+
+    assert_equal @haskovo_sofia.from, another.from
+    assert_equal @ruse_burgas.to, another.to
+    assert_equal combined_ticket.price + @ruse_burgas.price, another.price
+    assert_equal combined_ticket.features + ', ' + @ruse_burgas.features, another.features
+    assert_equal @haskovo_sofia.departure_time, another.departure_time
+    assert_equal @ruse_burgas.arrival_time, another.arrival_time
+    assert_equal '22'.to_d, another.travel_length
+  end
+
+  def test_ticket_equality
+    assert @haskovo_sofia == TicketSales::Ticket.build('Haskovo', 'Sofia', 230,Time.new(2013, 3, 3, 15),
+      Time.new(2013, 3, 3, 18), '20'.to_d, 'fast, class A, cheap')
+    assert @sofia_ruse == TicketSales::Ticket.build('Sofia', 'Ruse', 400,Time.new(2013, 3, 3, 19),
+      Time.new(2013,3,4), '25'.to_d, 'long')
+    refute @ruse_burgas == TicketSales::Ticket.build('Ruse', 'Burgas', 500, Time.new(2013, 3, 4, 5),
+      Time.new(2013, 3, 4, 13), '37'.to_d, '')
+  end
+end

data/test/ticket_sales/ticket/travel_time_test.rb

@@ -0,0 +1,57 @@
+require 'test_helper'
+
+class TestTravelTime  < MiniTest::Unit::TestCase
+  def get_travel_time(departure, arrival)
+    TicketSales::TravelTime.new departure, arrival
+  end
+
+  # Tests
+
+  def test_time_interval
+    time = TicketSales::TimeInterval.new Time.new(2013, 1, 1, 14), Time.new(2013, 1, 1, 15)
+
+    assert_equal Time.new(2013, 1, 1, 14), time.start_time
+    assert_equal Time.new(2013, 1, 1, 15), time.end_time
+    assert_equal '1'.to_d, time.length
+  end
+
+  def test_time_interval_equality
+    a_hour = TicketSales::TimeInterval.new Time.new(2013, 1, 1, 14), Time.new(2013, 1, 1, 15)
+    equal_hour = TicketSales::TimeInterval.new Time.new(2013, 1, 1, 14), Time.new(2013, 1, 1, 15)
+
+    assert a_hour == equal_hour
+    refute a_hour == TicketSales::TimeInterval.new(Time.new(2013, 1, 1, 14), Time.new(2013, 1, 1, 15, 1))
+  end
+
+  def test_travel_time
+    travel = get_travel_time Time.new(2013, 1, 1, 14), Time.new(2013, 1, 1, 15)
+
+    assert_equal Time.new(2013, 1, 1, 14), travel.departure
+    assert_equal Time.new(2013, 1, 1, 15), travel.arrival
+    assert_equal '1'.to_d, get_travel_time(Time.new(2013, 1, 1, 14), Time.new(2013, 1, 1, 15)).length
+    assert_equal '0.5'.to_d, get_travel_time(Time.new(2013, 1, 1, 14), Time.new(2013, 1, 1, 14, 30)).length
+  end
+
+  def test_combined_time
+    travel = get_travel_time Time.new(2013, 2, 14, 8), Time.new(2013, 2, 14, 15)
+    next_travel = get_travel_time Time.new(2013, 2, 15, 15), Time.new(2013, 2, 15, 16)
+    combined = travel.combine next_travel
+
+    assert_equal travel.departure, combined.departure
+    assert_equal next_travel.arrival, combined.arrival
+    assert_equal (next_travel.arrival - travel.departure).to_d / 3600, combined.length
+    assert_equal '24'.to_d, combined.pauses_length
+    assert_equal Time.new(2013, 2, 14, 15), combined.pauses.first.start_time
+    assert_equal Time.new(2013, 2, 15, 15), combined.pauses.first.end_time
+  end
+
+  def test_travel_time_equality
+    travel = get_travel_time Time.new(2013, 2, 14, 8), Time.new(2013, 2, 14, 15)
+    second_travel = get_travel_time Time.new(2013, 2, 15, 15), Time.new(2013, 2, 15, 16)
+    third_travel = get_travel_time Time.new(2013, 2, 15, 17), Time.new(2013, 2, 15, 19)
+    combined = travel.combine(second_travel).combine(third_travel)
+    same_combined = travel.combine(second_travel.combine third_travel)
+
+    assert_equal combined, same_combined
+  end
+end

data/test/ticket_sales/ticket_center_test.rb

@@ -0,0 +1,219 @@
+require 'test_helper'
+
+class TestTicketProperty < MiniTest::Unit::TestCase
+  def setup
+    @property = TicketSales::TicketProperty.new 12, TicketSales::Promotion::NoPromotion.new
+    @another_property = TicketSales::TicketProperty.new 10, TicketSales::Promotion.build(percent: 10)
+  end
+
+  def test_properties_can_be_accessed
+    assert_equal 12, @property.count
+    assert_kind_of TicketSales::Promotion::NoPromotion, @property.promotion
+    assert_equal 0, @property.bought_count
+
+    assert_equal 10, @another_property.count
+    assert_kind_of TicketSales::Promotion::PercentOff, @another_property.promotion
+    assert_equal 0, @property.bought_count
+  end
+
+  def test_decreasing_count_of_property
+    @property.decrease_count 5
+    assert_equal 7, @property.count
+    assert_equal 5, @property.bought_count
+    @property.decrease_count 5
+    assert_equal 10, @property.bought_count
+
+    @another_property.decrease_count 7
+    assert_equal 3, @another_property.count
+    assert_equal 7, @another_property.bought_count
+    @another_property.decrease_count 2
+    assert_equal 9, @another_property.bought_count
+  end
+end
+
+class TestTicketCenter < MiniTest::Unit::TestCase
+  def setup
+    @haskovo_sofia = TicketSales::Ticket.build 'Haskovo', 'Sofia', 230, Time.new(2013, 3, 3, 15),
+      Time.new(2013, 3, 3, 18), '15'.to_d, 'fast, class A, cheap'
+    @sofia_ruse = TicketSales::Ticket.build 'Sofia', 'Ruse', 400, Time.new(2013, 3, 3, 19),
+      Time.new(2013, 3, 4), '15'.to_d, 'long'
+    @ruse_burgas = TicketSales::Ticket.build 'Ruse', 'Burgas', 500, Time.new(2013, 3, 4, 5),
+      Time.new(2013, 3, 4, 13), '35'.to_d, ''
+    @user = TicketSales::User.new 'user', 'user@gmail.com', '50000'.to_d
+    @ticket_center = TicketSales::TicketCenter.new
+  end
+
+  # Tests
+
+  def test_adding_tickets
+    @ticket_center.add @haskovo_sofia
+    @ticket_center.add @sofia_ruse, 2
+    @ticket_center.add @ruse_burgas, 30, TicketSales::Promotion::build(amount: 10)
+  end
+
+  def test_available
+    @ticket_center.add @haskovo_sofia, 20
+
+    assert @ticket_center.available? @haskovo_sofia
+    refute @ticket_center.available? @ruse_burgas
+    refute @ticket_center.available? @sofia_ruse
+  end
+
+  def test_available_with_more_complex_tickets
+    @ticket_center.add @haskovo_sofia, 20
+    @ticket_center.add @ruse_burgas, 50, TicketSales::Promotion.build(percent: 10)
+    @ticket_center.add @sofia_ruse, 33
+
+    assert @ticket_center.available? @haskovo_sofia
+    assert @ticket_center.available? @ruse_burgas
+    assert @ticket_center.available? @sofia_ruse.combine(@ruse_burgas)
+    assert @ticket_center.available? @haskovo_sofia.combine(@sofia_ruse)
+    assert @ticket_center.available? @haskovo_sofia.combine(@sofia_ruse).combine(@ruse_burgas)
+  end
+
+  def test_available_working_correct_for_any_count
+    @ticket_center.add @haskovo_sofia, 20
+    @ticket_center.add @ruse_burgas, 50
+    @ticket_center.add @sofia_ruse, 33
+
+    assert @ticket_center.available? @haskovo_sofia, 3
+    assert @ticket_center.available? @haskovo_sofia, 19
+    assert @ticket_center.available? @haskovo_sofia, 20
+    refute @ticket_center.available? @haskovo_sofia, 21
+
+    assert @ticket_center.available? @sofia_ruse, 33
+    refute @ticket_center.available? @sofia_ruse, 34
+
+    assert @ticket_center.available? @ruse_burgas, 45
+    refute @ticket_center.available? @ruse_burgas, 51
+
+    haskovo_ruse = @haskovo_sofia.combine @sofia_ruse
+    assert @ticket_center.available? haskovo_ruse, 3
+    assert @ticket_center.available? haskovo_ruse, 19
+    assert @ticket_center.available? haskovo_ruse, 20
+    refute @ticket_center.available? haskovo_ruse, 21
+    refute @ticket_center.available? haskovo_ruse, 30
+
+    haskovo_burgas = haskovo_ruse.combine @ruse_burgas
+    assert @ticket_center.available? haskovo_burgas, 3
+    assert @ticket_center.available? haskovo_burgas, 19
+    assert @ticket_center.available? haskovo_burgas, 20
+    refute @ticket_center.available? haskovo_burgas, 21
+    refute @ticket_center.available? haskovo_burgas, 30
+
+    sofia_burgas = @sofia_ruse.combine @ruse_burgas
+    assert @ticket_center.available? @sofia_ruse, 33
+    refute @ticket_center.available? @sofia_ruse, 34
+    refute @ticket_center.available? @sofia_ruse, 45
+  end
+
+  def test_selling_tickets
+    @ticket_center.add @haskovo_sofia, 20
+    @ticket_center.add @ruse_burgas, 50
+
+    @ticket_center.sell @user, @ruse_burgas
+    @ticket_center.sell @user, @haskovo_sofia, 2
+  end
+
+  def test_selling_tickets_decreases_counts
+    @ticket_center.add @haskovo_sofia, 20
+    @ticket_center.add @ruse_burgas, 50
+    @ticket_center.add @sofia_ruse, 33
+
+    @ticket_center.sell @user, @ruse_burgas, 25
+    refute @ticket_center.available? @ruse_burgas, 26
+
+    @ticket_center.sell @user, @haskovo_sofia, 5
+    refute @ticket_center.available? @haskovo_sofia, 16
+    assert @ticket_center.available? @haskovo_sofia, 15
+
+    @ticket_center.sell @user, @sofia_ruse, 10
+    refute @ticket_center.available? @sofia_ruse, 24
+    assert @ticket_center.available? @sofia_ruse, 23
+  end
+
+  def test_selling_a_combined_ticket_decreases_counts
+    @ticket_center.add @haskovo_sofia, 20
+    @ticket_center.add @sofia_ruse, 33
+    @ticket_center.add @ruse_burgas, 50
+
+    haskovo_ruse = @haskovo_sofia.combine @sofia_ruse
+    @ticket_center.sell @user, haskovo_ruse, 5
+    refute @ticket_center.available? @haskovo_sofia, 16
+    assert @ticket_center.available? @haskovo_sofia, 15
+    refute @ticket_center.available? @sofia_ruse, 29
+    assert @ticket_center.available? @sofia_ruse, 28
+
+    haskovo_burgas = haskovo_ruse.combine @ruse_burgas
+    @ticket_center.sell @user, haskovo_burgas, 10
+    refute @ticket_center.available? @haskovo_sofia, 6
+    assert @ticket_center.available? @haskovo_sofia, 5
+    refute @ticket_center.available? @sofia_ruse, 19
+    assert @ticket_center.available? @sofia_ruse, 18
+    refute @ticket_center.available? @ruse_burgas, 41
+    assert @ticket_center.available? @ruse_burgas, 40
+  end
+
+  def test_sell_working_right_with_tickets_on_promotion
+    @ticket_center.add @haskovo_sofia, 20, TicketSales::Promotion.build(amount: 2.5)
+    @ticket_center.add @ruse_burgas, 50, TicketSales::Promotion.build(hundred_kilometers: 1)
+    @ticket_center.add @sofia_ruse, 33, TicketSales::Promotion.build(percent: 10)
+
+    balance_before_sale = @user.balance
+    @ticket_center.sell @user, @haskovo_sofia, 2
+    assert_equal '25'.to_d, balance_before_sale - @user.balance
+
+    balance_before_sale = @user.balance
+    @ticket_center.sell @user, @ruse_burgas
+    assert_equal '30'.to_d, balance_before_sale - @user.balance
+
+    balance_before_sale = @user.balance
+    @ticket_center.sell @user, @sofia_ruse
+    assert_equal '13.5'.to_d, balance_before_sale - @user.balance
+
+    balance_before_sale = @user.balance
+    @ticket_center.sell @user, @haskovo_sofia.combine(@sofia_ruse)
+    assert_equal '26'.to_d, balance_before_sale - @user.balance
+
+    balance_before_sale = @user.balance
+    @ticket_center.sell @user, @sofia_ruse.combine(@ruse_burgas)
+    assert_equal '43.5'.to_d, balance_before_sale - @user.balance
+
+    balance_before_sale = @user.balance
+    @ticket_center.sell @user, @haskovo_sofia.combine(@sofia_ruse).combine(@ruse_burgas)
+    assert_equal '56'.to_d, balance_before_sale - @user.balance
+  end
+
+  def test_sell_ticket_not_working_with_unavailable_tickets
+    @ticket_center.add @haskovo_sofia, 15
+
+    assert_raises(TicketSales::UnavailableTicket) { @ticket_center.sell @user, @sofia_ruse }
+    assert_raises(TicketSales::UnavailableTicket) { @ticket_center.sell @user, @ruse_burgas }
+    assert_raises(TicketSales::UnavailableTicket) { @ticket_center.sell @user, @haskovo_sofia, 16 }
+
+    @ticket_center.add @sofia_ruse, 20
+    haskovo_ruse = @haskovo_sofia.combine @sofia_ruse
+    assert_raises(TicketSales::UnavailableTicket) { @ticket_center.sell @user, haskovo_ruse, 16 }
+
+    @ticket_center.sell @user, haskovo_ruse, 10
+    assert @ticket_center.available? haskovo_ruse, 5
+    assert_raises(TicketSales::UnavailableTicket) { @ticket_center.sell @user, @haskovo_sofia, 6 }
+    assert_raises(TicketSales::UnavailableTicket) { @ticket_center.sell @user, haskovo_ruse, 6 }
+  end
+
+  def test_finding_most_bought_tickets
+    @ticket_center.add @haskovo_sofia, 20
+    @ticket_center.add @sofia_ruse, 25
+    @ticket_center.add @ruse_burgas, 30
+
+    @ticket_center.sell @user, @sofia_ruse, 5
+    assert_equal @sofia_ruse, @ticket_center.most_bought_tickets.first
+
+    @ticket_center.sell @user, @haskovo_sofia, 6
+    assert_equal [@haskovo_sofia, @sofia_ruse, @ruse_burgas], @ticket_center.most_bought_tickets
+
+    haskovo_burgas = @haskovo_sofia.combine(@sofia_ruse).combine @ruse_burgas
+    @ticket_center.sell @user, haskovo_burgas, 7
+    assert_equal [@haskovo_sofia, @sofia_ruse, @ruse_burgas], @ticket_center.most_bought_tickets
+  end
+end

data/test/ticket_sales/ticket_list_test.rb

@@ -0,0 +1,150 @@
+require 'test_helper'
+
+class TestTicketList  < MiniTest::Unit::TestCase
+  def setup
+    @haskovo_sofia = TicketSales::Ticket.build 'Haskovo', 'Sofia', 230, Time.new(2013, 3, 3, 15),
+      Time.new(2013, 3, 3, 18), '15'.to_d, 'fast, class A, cheap'
+    @sofia_ruse = TicketSales::Ticket.build 'Sofia', 'Ruse', 400, Time.new(2013, 3, 3, 19),
+      Time.new(2013, 3, 4), '15'.to_d, 'long'
+    @ruse_burgas = TicketSales::Ticket.build 'Ruse', 'Burgas', 500, Time.new(2013, 3, 4, 5),
+      Time.new(2013, 3, 4, 13), '35'.to_d, ''
+    @burgas_moscow = TicketSales::Ticket.build 'Burgas', 'Moscow', 1900, Time.new(2013, 3, 4, 20),
+      Time.new(2013, 3, 4, 22), '120'.to_d, 'airplane'
+    @moscow_kazan = TicketSales::Ticket.build 'Moscow', 'Kazan', 700, Time.new(2013, 3, 5, 8),
+      Time.new(2013, 3, 5, 9, 30), '50'.to_d, 'fast'
+    @london_sofia = TicketSales::Ticket.build 'London', 'Sofia', 2000, Time.new(2013, 3, 3, 12),
+      Time.new(2013, 3, 3, 15), '100'.to_d, ''
+    @plovdiv_sofia = TicketSales::Ticket.build 'Plovdiv', 'Sofia', 150, Time.new(2013, 3, 3, 17),
+      Time.new(2013, 3, 3, 19, 20), '10'.to_d, ''
+    @haskovo_ruse_direct = TicketSales::Ticket.build 'Haskovo', 'Ruse', 500, Time.new(2013, 3, 3, 15),
+      Time.new(2013, 3, 3, 22), '35'.to_d, ''
+    @list = TicketSales::TicketList.new
+  end
+
+  # Tests
+
+  def test_enumerable
+    assert [:map, :select, :inject, :all?, :any?, :include?].all? { |m| @list.respond_to? m }
+  end
+
+  def test_add_for_single_ticket
+    @list.add @haskovo_sofia
+    @list.add @ruse_burgas
+
+    assert_includes @list, @haskovo_sofia
+    assert_includes @list, @ruse_burgas
+    refute_includes @list, @sofia_ruse
+  end
+
+  def test_ticket_filter
+    @list.add @haskovo_sofia
+    @list.add @ruse_burgas
+
+    filtered_tickets = @list.filter(TicketSales::Criteria.from('Haskovo'))
+    assert filtered_tickets.all? { |ticket| ticket.from == 'Haskovo' }
+  end
+
+  def test_tickets_arriving_at_location_before_time
+    @list.add @haskovo_sofia
+    @list.add @plovdiv_sofia
+
+    assert_includes @list.tickets_arriving_at('Sofia', Time.new(2013, 3, 3, 18, 30)), @haskovo_sofia
+    refute_includes @list.tickets_arriving_at('Sofia', Time.new(2013, 3, 3, 18, 30)), @plovdiv_sofia
+  end
+
+  def test_tickets_departing_from_location_after_time
+    @list.add @haskovo_sofia
+    @list.add @haskovo_ruse_direct
+
+    assert_includes @list.tickets_departing_from('Haskovo', Time.new(2013, 3, 3, 14, 59)), @haskovo_sofia
+    assert_includes @list.tickets_departing_from('Haskovo', Time.new(2013, 3, 3, 14, 59)), @haskovo_ruse_direct
+    refute_includes @list.tickets_departing_from('Haskovo', Time.new(2013, 3, 3, 15, 01)), @haskovo_sofia
+    refute_includes @list.tickets_departing_from('Haskovo', Time.new(2013, 3, 3, 15, 01)), @haskovo_ruse_direct
+  end
+
+  def test_add_combines_routes
+    @list.add @haskovo_sofia
+    @list.add @sofia_ruse
+    haskovo_ruse = @haskovo_sofia.combine @sofia_ruse
+
+    assert_includes @list, haskovo_ruse
+    assert_equal 3, @list.tickets.size
+
+    @list.add @ruse_burgas
+    haskovo_burgas = @haskovo_sofia.combine(@sofia_ruse).combine @ruse_burgas
+    sofia_burgas = @sofia_ruse.combine @ruse_burgas
+
+    assert_includes @list, haskovo_burgas
+    assert_includes @list, sofia_burgas
+    assert_equal 6, @list.tickets.size
+
+    @list.add @moscow_kazan
+    @list.add @burgas_moscow
+
+    burgas_kazan = @burgas_moscow.combine @moscow_kazan
+    assert_includes @list, burgas_kazan
+
+    ruse_moscow = @ruse_burgas.combine @burgas_moscow
+    ruse_kazan = ruse_moscow.combine @moscow_kazan
+    assert_includes @list, ruse_moscow
+    assert_includes @list, ruse_kazan
+
+    sofia_moscow = @sofia_ruse.combine ruse_moscow
+    sofia_kazan = sofia_moscow.combine @moscow_kazan
+    assert_includes @list, sofia_moscow
+    assert_includes @list, sofia_kazan
+
+    haskovo_moscow = haskovo_burgas.combine @burgas_moscow
+    haskovo_kazan = haskovo_moscow.combine @moscow_kazan
+    assert_includes @list, haskovo_moscow
+    assert_includes @list, haskovo_kazan
+
+    @list.add @london_sofia
+    tickets_from_london = @list.filter TicketSales::Criteria.from('London')
+    can_be_reached = ['Sofia', 'Ruse', 'Burgas', 'Moscow', 'Kazan'].all? do |location|
+      tickets_from_london.map(&:to).include? location
+    end
+    assert can_be_reached
+  end
+
+  def test_add_not_including_incorrect_tickets
+    @list.add @sofia_ruse
+    @list.add @plovdiv_sofia
+    incorrect_ticket = @plovdiv_sofia.combine @sofia_ruse
+
+    refute_includes @list, incorrect_ticket
+    assert_equal 2, @list.tickets.size
+
+    haskovo_sofia = TicketSales::Ticket.build 'Haskovo', 'Sofia', 230, Time.new(2013, 3, 3, 15),
+      Time.new(2013, 3, 3, 19), '20'.to_d, 'fast, class A, cheap'
+    @list.add haskovo_sofia
+    incorrect_ticket = haskovo_sofia.combine @sofia_ruse
+
+    refute_includes @list, incorrect_ticket
+    assert_equal 3, @list.tickets.size
+  end
+
+  def test_ticket_search_from_to
+    @list.add @haskovo_ruse_direct
+    @list.add @haskovo_sofia
+    @list.add @sofia_ruse
+
+    tickets_from_haskovo_to_ruse = @list.get_tickets('Haskovo', 'Ruse')
+    assert_equal 2, tickets_from_haskovo_to_ruse.size
+    assert_includes tickets_from_haskovo_to_ruse, @haskovo_ruse_direct
+    assert_includes tickets_from_haskovo_to_ruse, @haskovo_sofia.combine(@sofia_ruse)
+  end
+
+  def test_ticket_search
+    @list.add @haskovo_ruse_direct
+    @list.add @haskovo_sofia
+    @list.add @sofia_ruse
+    ticket = @haskovo_sofia.combine @sofia_ruse
+
+    cheapest_ticket = @list.cheapest_ticket('Haskovo', 'Ruse')
+    assert_equal ticket, cheapest_ticket
+
+    fastest_ticket = @list.fastest_ticket('Haskovo', 'Ruse')
+    assert_equal @haskovo_ruse_direct, fastest_ticket
+  end
+end