rail_feeds 0.0.1

data/spec/fixtures/network_rail/schedule/parser/train_create.json

@@ -0,0 +1 @@
+{"JsonScheduleV1":{"CIF_bank_holiday_running":null,"CIF_stp_indicator":"P","CIF_train_uid":"W20309","applicable_timetable":"Y","atoc_code":"IL","new_schedule_segment":{"traction_class":"","uic_code":""},"schedule_days_runs":"1111100","schedule_end_date":"2018-12-07","schedule_segment":{"signalling_id":"2U10","CIF_train_category":"OO","CIF_headcode":"","CIF_course_indicator":1,"CIF_train_service_code":"24627006","CIF_business_sector":"??","CIF_power_type":"EMU","CIF_timing_load":"483","CIF_speed":"045","CIF_operating_characteristics":null,"CIF_train_class":"S","CIF_sleepers":null,"CIF_reservations":null,"CIF_connection_indicator":null,"CIF_catering_code":null,"CIF_service_branding":"","schedule_location":[{"location_type":"LO","record_identity":"LO","tiploc_code":"SHANKLN","tiploc_instance":null,"departure":"0718","public_departure":"0718","platform":null,"line":null,"engineering_allowance":null,"pathing_allowance":null,"performance_allowance":null},{"location_type":"LI","record_identity":"LI","tiploc_code":"LAKEIOW","tiploc_instance":null,"arrival":"0720","departure":"0721","pass":null,"public_arrival":"0720","public_departure":"0721","platform":null,"line":null,"path":null,"engineering_allowance":null,"pathing_allowance":null,"performance_allowance":null},{"location_type":"LI","record_identity":"LI","tiploc_code":"SNDOWN","tiploc_instance":null,"arrival":"0723","departure":"0724","pass":null,"public_arrival":"0723","public_departure":"0724","platform":null,"line":null,"path":null,"engineering_allowance":null,"pathing_allowance":null,"performance_allowance":null},{"location_type":"LI","record_identity":"LI","tiploc_code":"BRDING","tiploc_instance":null,"arrival":"0727H","departure":"0728","pass":null,"public_arrival":"0728","public_departure":"0728","platform":null,"line":null,"path":null,"engineering_allowance":null,"pathing_allowance":null,"performance_allowance":null},{"location_type":"LI","record_identity":"LI","tiploc_code":"SMALBRK","tiploc_instance":null,"arrival":null,"departure":null,"pass":"0732","public_arrival":null,"public_departure":null,"platform":null,"line":null,"path":null,"engineering_allowance":null,"pathing_allowance":null,"performance_allowance":null},{"location_type":"LI","record_identity":"LI","tiploc_code":"RYDS","tiploc_instance":null,"arrival":"0735","departure":"0735H","pass":null,"public_arrival":"0735","public_departure":"0735","platform":null,"line":null,"path":null,"engineering_allowance":null,"pathing_allowance":null,"performance_allowance":null},{"location_type":"LI","record_identity":"LI","tiploc_code":"RYDE","tiploc_instance":null,"arrival":"0738H","departure":"0740","pass":null,"public_arrival":"0739","public_departure":"0740","platform":null,"line":null,"path":null,"engineering_allowance":null,"pathing_allowance":null,"performance_allowance":null},{"location_type":"LT","record_identity":"LT","tiploc_code":"RYDP","tiploc_instance":null,"arrival":"0742","public_arrival":"0742","platform":null,"path":null}]},"schedule_start_date":"2018-05-21","train_status":"P","transaction_type":"Create"}}

data/spec/fixtures/network_rail/schedule/parser/train_delete.json

@@ -0,0 +1 @@
+{"JsonScheduleV1":{"CIF_train_uid":"C77605","schedule_start_date":"2018-08-25","CIF_stp_indicator":"O","transaction_type":"Delete"}}

data/spec/fixtures/network_rail/schedule/train_schedule/json-data.yaml

@@ -0,0 +1,67 @@
+---
+JsonScheduleV1:
+  CIF_bank_holiday_running: B
+  CIF_stp_indicator: P
+  CIF_train_uid: W20309
+  applicable_timetable: Y
+  atoc_code: IL
+  schedule_start_date: "2018-05-21"
+  train_status: S
+  transaction_type: Create
+  schedule_days_runs: "1111100"
+  schedule_end_date: "2018-12-07"
+  new_schedule_segment:
+    traction_class: ""
+    uic_code: "12345"
+  schedule_segment:
+    signalling_id: 2U10
+    CIF_train_category: "OO"
+    CIF_headcode: "1234"
+    CIF_course_indicator:
+    CIF_train_service_code: "24627006"
+    CIF_business_sector: ??
+    CIF_power_type: EMU
+    CIF_timing_load: "483"
+    CIF_speed: "045"
+    CIF_operating_characteristics: OPER-CHAR
+    CIF_train_class: S
+    CIF_sleepers: B
+    CIF_reservations: R
+    CIF_connection_indicator:
+    CIF_catering_code: C
+    CIF_service_branding: Br
+    schedule_location:
+    - location_type: LO
+      record_identity: LO
+      tiploc_code: SHANKLN
+      tiploc_instance:
+      departure: "0718"
+      public_departure: "0718"
+      platform: P
+      line: L
+      engineering_allowance: "1"
+      pathing_allowance: "1H"
+      performance_allowance: "2"
+    - location_type: LI
+      record_identity: LI
+      tiploc_code: LAKEIOW
+      tiploc_instance: nil
+      arrival: "0720"
+      departure: "0721"
+      pass: "0722"
+      public_arrival: "0720"
+      public_departure: "0721"
+      platform: P
+      line: L
+      path: p
+      engineering_allowance:
+      pathing_allowance:
+      performance_allowance:
+    - location_type: LT
+      record_identity: LT
+      tiploc_code: RYDP
+      tiploc_instance:
+      arrival: "0742"
+      public_arrival: "0742"
+      platform: P
+      path: p

data/spec/rail_feeds/credentials_spec.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+describe RailFeeds::Credentials do
+  context 'Using system wide credentials' do
+    before :each do
+      described_class.configure(
+        username: 'user@example.com',
+        password: 'P@55word'
+      )
+    end
+
+    it 'Sets class attributes' do
+      expect(described_class.username).to eq 'user@example.com'
+      expect(described_class.password).to eq 'P@55word'
+    end
+
+    it 'A new instance defaults to class attributes' do
+      expect(subject.username).to eq 'user@example.com'
+      expect(subject.password).to eq 'P@55word'
+    end
+  end
+
+
+  context 'Using specific credentials' do
+    subject do
+      described_class.new(
+        username: 'user2@example.com',
+        password: 'P@55word2'
+      )
+    end
+
+    before :each do
+      described_class.configure username: nil, password: nil
+    end
+
+    it 'Sets instance attributes' do
+      expect(subject.username).to eq 'user2@example.com'
+      expect(subject.password).to eq 'P@55word2'
+    end
+
+    it 'Leaves class attributes alone' do
+      expect(described_class.username).to eq ''
+      expect(described_class.password).to eq ''
+    end
+  end
+end

data/spec/rail_feeds/logging_spec.rb

@@ -0,0 +1,81 @@
+# frozen_string_literal: true
+
+class DummyClassForLoggerTests
+  include RailFeeds::Logging
+end
+
+describe RailFeeds::Logging do
+  let(:global_logger) { double Logger }
+
+  it 'Default system logger' do
+    described_class.logger = nil
+    formatter = double Proc
+    expect(described_class).to receive(:formatter).and_return(formatter)
+    expect(Logger).to receive(:new).with(
+      STDOUT,
+      formatter: formatter,
+      level: Logger::DEBUG
+    ).and_return(global_logger)
+    expect(described_class.logger).to be global_logger
+  end
+
+  describe 'Default system formatter' do
+    subject { described_class.formatter }
+    it 'With all items passed in' do
+      expect(subject.call('s', 'd', 'p', 'm')).to eq "d p s: m\n"
+    end
+
+    it 'Without progname' do
+      expect(subject.call('S', 'D', nil, 'M')).to eq "D S: M\n"
+    end
+  end
+
+  context 'included in a class' do
+    let(:klass) { DummyClassForLoggerTests }
+    let(:instance) { klass.new }
+    before(:each) do
+      allow(described_class).to receive(:logger).and_return(global_logger)
+    end
+    let(:class_logger) { double Logger }
+    let(:instance_logger) { double Logger }
+
+    describe 'Defaults to using global logger' do
+      it '::logger' do
+        expect(klass.logger).to be global_logger
+      end
+
+      it '#logger' do
+        expect(instance.logger).to be global_logger
+      end
+    end
+
+    describe 'Setting custom logger for class' do
+      before(:each) { klass.logger = class_logger }
+
+      it 'Global logger is unaffected' do
+        expect(described_class.logger).to be global_logger
+      end
+      it 'Instance defaults to using class logger' do
+        expect(instance.logger).to be class_logger
+      end
+      it 'Sets class logger' do
+        expect(klass.logger).to be class_logger
+      end
+    end
+
+    describe 'Setting custom logger for instance' do
+      before(:each) { klass.logger = class_logger }
+      before(:each) { instance.logger = instance_logger }
+
+      it 'Global logger is unaffected' do
+        expect(described_class.logger).to be global_logger
+      end
+      it 'Class logger is unaffected' do
+        expect(klass.logger).to be class_logger
+      end
+      it 'Sets instance logger' do
+        expect(instance.logger).to be instance_logger
+      end
+    end
+  end
+end

data/spec/rail_feeds/network_rail/corpus_spec.rb

@@ -0,0 +1,92 @@
+# frozen_string_literal: true
+
+describe RailFeeds::NetworkRail::CORPUS do
+  let :json do
+    '{"TIPLOCDATA":[{"NLCDESC":"MERSEYRAIL ELECTRICS","NLC":"8","TIPLOC":"MERSELE","3A' \
+    'LPHA":"ABC","STANOX":"1","NLCDESC16":"MPTE HQ","UIC":"2"},{"NLCDESC":" ","NLC":" ' \
+    '","TIPLOC":" ","3ALPHA":" ","STANOX":" ","NLCDESC16":" ","UIC":" "}]}'
+  end
+  let(:http_client) { double RailFeeds::NetworkRail::HTTPClient }
+  let(:reader) { double Zlib::GzipReader }
+
+  describe '::download' do
+    let(:http_client) { double RailFeeds::NetworkRail::HTTPClient }
+    let(:temp_file) { double Tempfile }
+
+    it 'Using default credentials' do
+      expect(RailFeeds::NetworkRail::HTTPClient).to receive(:new)
+        .with(credentials: RailFeeds::NetworkRail::Credentials).and_return(http_client)
+      expect(http_client).to receive(:download)
+        .with('ntrod/SupportingFileAuthenticate?type=CORPUS', 'file')
+      described_class.download 'file'
+    end
+
+    it 'Using passed credentials' do
+      credentials = double RailFeeds::NetworkRail::Credentials
+      expect(RailFeeds::NetworkRail::HTTPClient).to receive(:new)
+        .with(credentials: credentials).and_return(http_client)
+      expect(http_client).to receive(:download)
+      described_class.download 'file', credentials: credentials
+    end
+  end
+
+  describe '::fetch' do
+    let(:http_client) { double RailFeeds::NetworkRail::HTTPClient }
+    let(:temp_file) { double Tempfile }
+
+    it 'Using default credentials' do
+      expect(RailFeeds::NetworkRail::HTTPClient).to receive(:new)
+        .with(credentials: RailFeeds::NetworkRail::Credentials).and_return(http_client)
+      expect(http_client).to receive(:fetch)
+        .with('ntrod/SupportingFileAuthenticate?type=CORPUS').and_return(temp_file)
+      expect(described_class.fetch).to eq temp_file
+    end
+
+    it 'Using passed credentials' do
+      credentials = double RailFeeds::NetworkRail::Credentials
+      expect(RailFeeds::NetworkRail::HTTPClient).to receive(:new)
+        .with(credentials: credentials).and_return(http_client)
+      expect(http_client).to receive(:fetch).and_return(temp_file)
+      expect(described_class.fetch(credentials: credentials)).to eq temp_file
+    end
+  end
+
+  describe '::load_file' do
+    it 'json file' do
+      expect(Zlib::GzipReader).to receive(:open).with('filename') { fail Zlib::GzipFile::Error }
+      expect(File).to receive(:read).with('filename').and_return(json)
+      data = described_class.load_file('filename')
+      expect(data[0].tiploc).to eq 'MERSELE'
+      expect(data[0].stanox).to eq 1
+      expect(data[0].crs).to eq 'ABC'
+      expect(data[0].uic).to eq 2
+      expect(data[0].nlc).to eq '8'
+      expect(data[0].nlc_description).to eq 'MERSEYRAIL ELECTRICS'
+      expect(data[0].nlc_short_description).to eq 'MPTE HQ'
+      expect(data[1].tiploc).to be_nil
+      expect(data[1].stanox).to be_nil
+      expect(data[1].crs).to be_nil
+      expect(data[1].uic).to be_nil
+      expect(data[1].nlc).to be_nil
+      expect(data[1].nlc_description).to be_nil
+      expect(data[1].nlc_short_description).to be_nil
+    end
+
+    it 'json.gz file' do
+      expect(Zlib::GzipReader).to receive(:open).with('filename')
+                                                .and_yield(StringIO.new(json))
+      expect(described_class.load_file('filename').count).to eq 2
+    end
+  end
+
+  it '::fetch_data' do
+    expect(RailFeeds::NetworkRail::HTTPClient)
+      .to receive(:new).with(credentials: RailFeeds::NetworkRail::Credentials)
+                       .and_return(http_client)
+    expect(http_client).to receive(:fetch_unzipped)
+      .with('ntrod/SupportingFileAuthenticate?type=CORPUS')
+      .and_yield(reader)
+    expect(reader).to receive(:read).and_return(json)
+    expect(described_class.fetch_data.count).to eq 2
+  end
+end

data/spec/rail_feeds/network_rail/credentials_spec.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+describe RailFeeds::NetworkRail::Credentials do
+  subject { described_class.new username: 'user-i', password: 'pass-i' }
+
+  it 'Has seperate values to RailFeeds::Credentials' do
+    described_class.configure username: 'a', password: 'b'
+    expect(described_class.username).to_not eq RailFeeds::Credentials.username
+    expect(described_class.password).to_not eq RailFeeds::Credentials.password
+  end
+
+  describe 'Outputs an array' do
+    it '::to_a' do
+      described_class.configure username: 'user', password: 'pass'
+      expect(described_class.to_a).to eq ['user', 'pass']
+    end
+
+    it '#to_a' do
+      expect(subject.to_a).to eq ['user-i', 'pass-i']
+    end
+  end
+end

data/spec/rail_feeds/network_rail/http_client_spec.rb

@@ -0,0 +1,88 @@
+# frozen_string_literal: true
+
+describe RailFeeds::NetworkRail::HTTPClient do
+  let(:temp_file) { double Tempfile }
+  let(:uri) { double URI }
+
+  describe '#download' do
+    let(:file) { double File }
+
+    it 'Saves the file' do
+      expect(URI).to receive(:parse).with('https://datafeeds.networkrail.co.uk/path').and_return(uri)
+      expect(uri).to receive(:open).and_return(temp_file)
+      expect(File).to receive(:open).with('file', 'w').and_yield(file)
+      expect(IO).to receive(:copy_stream).with(temp_file, file)
+      subject.download('path', 'file')
+    end
+
+    it 'Adds credentials when getting path' do
+      credentials = RailFeeds::NetworkRail::Credentials.new(
+        username: 'user',
+        password: 'pass'
+      )
+      expect(URI).to receive(:parse).and_return(uri)
+      expect(uri).to receive(:open)
+        .with(http_basic_authentication: ['user', 'pass'])
+        .and_return(temp_file)
+      expect(File).to receive(:open).and_yield(file)
+      expect(IO).to receive(:copy_stream).with(temp_file, file)
+      subject = described_class.new credentials: credentials
+      subject.download('path', 'file')
+    end
+
+    it 'Handles special characters in credentials' do
+      credentials = RailFeeds::NetworkRail::Credentials.new(
+        username: 'a@example.com',
+        password: '!:@'
+      )
+      expect(URI).to receive(:parse).and_return(uri)
+      expect(uri).to receive(:open).and_return(temp_file)
+      expect(File).to receive(:open).and_yield(file)
+      expect(IO).to receive(:copy_stream).with(temp_file, file)
+      subject = described_class.new credentials: credentials
+      expect { subject.download('path', 'file') }.to_not raise_error
+    end
+  end
+
+  describe '#fetch' do
+    it 'Yields what uri.open does' do
+      expect(URI).to receive(:parse).with('https://datafeeds.networkrail.co.uk/path').and_return(uri)
+      expect(uri).to receive(:open).and_return(temp_file)
+      expect { |a| subject.fetch('path', &a) }.to yield_with_args(temp_file)
+    end
+
+    it 'Adds credentials when getting path' do
+      credentials = RailFeeds::NetworkRail::Credentials.new(
+        username: 'user',
+        password: 'pass'
+      )
+      expect(URI).to receive(:parse).and_return(uri)
+      expect(uri).to receive(:open)
+        .with(http_basic_authentication: ['user', 'pass'])
+        .and_return(temp_file)
+      subject = described_class.new credentials: credentials
+      subject.fetch('path') {}
+    end
+
+    it 'Handles special characters in credentials' do
+      credentials = RailFeeds::NetworkRail::Credentials.new(
+        username: 'a@example.com',
+        password: '!:@'
+      )
+      expect(URI).to receive(:parse).and_return(uri)
+      expect(uri).to receive(:open).and_return(temp_file)
+      subject = described_class.new credentials: credentials
+      expect { subject.fetch('path') {} }.to_not raise_error
+    end
+  end
+
+  describe '#fetch_unzipped' do
+    it 'Returns what Zlib::GzipReader.open does' do
+      reader = double Zlib::GzipReader
+      expect(subject).to receive(:fetch).with('path').and_yield(temp_file)
+      expect(temp_file).to receive(:path).and_return('gz_file_path')
+      expect(Zlib::GzipReader).to receive(:open).with('gz_file_path').and_return(reader)
+      expect { |a| subject.fetch_unzipped('path', &a) }.to yield_with_args(reader)
+    end
+  end
+end

data/spec/rail_feeds/network_rail/schedule/association_spec.rb

@@ -0,0 +1,205 @@
+# frozen_string_literal: true
+
+describe RailFeeds::NetworkRail::Schedule::Association do
+  let(:line) do
+    'AANmmmmmmaaaaaa0102030405060101010AcDTTTTTTT12T                                P'
+  end
+  subject { described_class.from_cif line }
+
+  describe '::from_cif' do
+    it 'Sets attributes' do
+      expect(subject.main_train_uid).to eq 'mmmmmm'
+      expect(subject.associated_train_uid).to eq 'aaaaaa'
+      expect(subject.start_date).to eq Date.new(2001, 2, 3)
+      expect(subject.end_date).to eq Date.new(2004, 5, 6)
+      expect(subject.days).to eq [false, true, false, true, false, true, false]
+      expect(subject.category).to eq 'Ac'
+      expect(subject.date_indicator).to eq 'D'
+      expect(subject.tiploc).to eq 'TTTTTTT'
+      expect(subject.main_location_suffix).to eq 1
+      expect(subject.associated_location_suffix).to eq 2
+      expect(subject.stp_indicator).to eq :permanent
+    end
+
+    it 'Fails for invalid line' do
+      expect { described_class.from_cif('bad line') }
+        .to raise_error ArgumentError, "Invalid line:\nbad line"
+    end
+
+    it 'Delete line' do
+      line = 'AADW43767W43768180529                ASHFKY   T                                C'
+      subject = described_class.from_cif line
+      expect(subject.main_train_uid).to eq 'W43767'
+      expect(subject.associated_train_uid).to eq 'W43768'
+      expect(subject.start_date).to eq Date.new(2018, 5, 29)
+      expect(subject.end_date).to be_nil
+      expect(subject.days).to eq [false, false, false, false, false, false, false]
+      expect(subject.category).to be_nil
+      expect(subject.date_indicator).to be_nil
+      expect(subject.tiploc).to eq 'ASHFKY'
+      expect(subject.main_location_suffix).to be_nil
+      expect(subject.associated_location_suffix).to be_nil
+      expect(subject.stp_indicator).to eq :stp_cancellation
+    end
+  end
+
+  it '::from_json' do
+    subject = described_class.from_json(
+      '{"JsonAssociationV1":{"transaction_type":"Create","main_train_uid":"C69888",' \
+      '"assoc_train_uid":"C69870","assoc_start_date":"2018-10-08T00:00:00Z",' \
+      '"assoc_end_date":"2018-12-07T00:00:00Z","assoc_days":"1111100","category":"NP",' \
+      '"date_indicator":"S","location":"NTNG","base_location_suffix":null,' \
+      '"assoc_location_suffix":null,"diagram_type":"T","CIF_stp_indicator":"P"}}'
+    )
+
+    expect(subject.main_train_uid).to eq 'C69888'
+    expect(subject.associated_train_uid).to eq 'C69870'
+    expect(subject.start_date).to eq Date.new(2018, 10, 8)
+    expect(subject.end_date).to eq Date.new(2018, 12, 7)
+    expect(subject.days).to eq [true, true, true, true, true, false, false]
+    expect(subject.category).to eq 'NP'
+    expect(subject.date_indicator).to eq 'S'
+    expect(subject.tiploc).to eq 'NTNG'
+    expect(subject.main_location_suffix).to be_nil
+    expect(subject.associated_location_suffix).to be_nil
+    expect(subject.stp_indicator).to eq :permanent
+  end
+
+  describe 'Helper methods' do
+    context 'Join association' do
+      before(:each) { subject.category = 'JJ' }
+      it { should be_join }
+      it { should_not be_divide }
+      it { should_not be_next }
+    end
+
+    context 'Divide association' do
+      before(:each) { subject.category = 'VV' }
+      it { should_not be_join }
+      it { should be_divide }
+      it { should_not be_next }
+    end
+
+    context 'Next association' do
+      before(:each) { subject.category = 'NP' }
+      it { should_not be_join }
+      it { should_not be_divide }
+      it { should be_next }
+    end
+
+    context 'Happens on the same day' do
+      before(:each) { subject.date_indicator = 'S' }
+      it { should be_same_day }
+      it { should_not be_over_next_midnight }
+      it { should_not be_over_previous_midnight }
+    end
+
+    context 'Happens over the following midnight' do
+      before(:each) { subject.date_indicator = 'N' }
+      it { should_not be_same_day }
+      it { should be_over_next_midnight }
+      it { should_not be_over_previous_midnight }
+    end
+
+    context 'Happens over the previous midnight' do
+      before(:each) { subject.date_indicator = 'P' }
+      it { should_not be_same_day }
+      it { should_not be_over_next_midnight }
+      it { should be_over_previous_midnight }
+    end
+
+    it '#main_train_event_id' do
+      subject.tiploc = 'TTTTTTT'
+      subject.main_location_suffix = 1
+      expect(subject.main_train_event_id).to eq 'TTTTTTT-1'
+    end
+
+    it '#associated_train_event_id' do
+      subject.tiploc = 'TTTTTTT'
+      subject.associated_location_suffix = 2
+      expect(subject.associated_train_event_id).to eq 'TTTTTTT-2'
+    end
+  end
+
+  it '#to_cif' do
+    expect(subject.to_cif).to eq "#{line}\n"
+  end
+
+  it '#to_json' do
+    expect(subject.to_json).to eq '{"JsonAssociationV1":{"transaction_type":"Create",' \
+                                  '"main_train_uid":"mmmmmm","assoc_train_uid":"aaaaaa",' \
+                                  '"assoc_start_date":"2001-02-03T00:00:00Z",' \
+                                  '"assoc_end_date":"2004-05-06T00:00:00Z",' \
+                                  '"assoc_days":"0101010","category":"Ac",' \
+                                  '"date_indicator":"D","location":"TTTTTTT",' \
+                                  '"base_location_suffix":1,"assoc_location_suffix":2,' \
+                                  '"diagram_type":"T","CIF_stp_indicator":"P"}}'
+  end
+
+  describe '#hash' do
+    it 'Uses tiploc, main_location_suffix and associated_location_suffix' do
+      subject.tiploc = 'TIPLOC'
+      subject.main_location_suffix = 1
+      subject.associated_location_suffix = 2
+      expect(subject.hash).to eq 'TIPLOC-1-2'
+    end
+  end
+
+  describe '#<=>' do
+    let(:association1) { described_class.new start_date: Date.new(2000, 1, 1) }
+    let(:association2) { described_class.new start_date: Date.new(2000, 1, 2) }
+
+    it 'Match' do
+      association2.start_date = Date.new 2000, 1, 1
+      expect(association1 <=> association2).to eq 0
+      expect(association2 <=> association1).to eq 0
+    end
+
+    it 'Doesn\'t match' do
+      expect(association1 <=> association2).to eq(-1)
+      expect(association2 <=> association1).to eq 1
+    end
+
+    it 'Compares to nil without error' do
+      expect { association1 <=> nil }.to_not raise_error
+    end
+  end
+
+  describe '#==' do
+    let(:other_association) { double described_class }
+    before :each do
+      subject.tiploc = 'TTTTTTT'
+      subject.main_location_suffix = 1
+      subject.associated_location_suffix = 2
+      allow(other_association).to receive(:main_train_event_id).and_return(nil)
+      allow(other_association).to receive(:associated_train_event_id).and_return(nil)
+    end
+
+    it 'Matches on neither main or associated train\'s event' do
+      expect(subject == other_association).to eq false
+    end
+
+    it 'Matches on main train\'s event only' do
+      allow(other_association).to receive(:main_train_event_id).and_return('TTTTTTT-1')
+      expect(subject == other_association).to eq false
+    end
+
+    it 'Matches on associated train\'s event only' do
+      allow(other_association).to receive(:associated_train_event_id).and_return('TTTTTTT-2')
+      expect(subject == other_association).to eq false
+    end
+
+    it 'Matches on both main and associated train\'s event' do
+      allow(other_association).to receive(:main_train_event_id).and_return('TTTTTTT-1')
+      allow(other_association).to receive(:associated_train_event_id).and_return('TTTTTTT-2')
+      expect(subject == other_association).to eq true
+    end
+
+    it 'Compares to nil without error' do
+      expect(subject).to_not eq nil
+    end
+  end
+
+  it_behaves_like 'it has an STP indicator'
+  it_behaves_like 'it has a days array'
+end