def test_FindOrAddStation(self):
n = Network()
stationid = "station"
stationid3 = "station3"
companyid = "company"
divisionid = "division"
s = Station(stationid)
s2 = Station(stationid)
n.AddStation(s)
self.assertEqual(n.FindOrAddStation(stationid).ID, stationid)
self.assertEqual(n.FindOrAddStation(stationid3,companyid, divisionid).ID, stationid3)
self.assertEqual(n.Stations[stationid3].ID, stationid3)
self.assertEqual(n.Companies[companyid].Stations[stationid3].ID, stationid3)
self.assertEqual(n.Companies[companyid].Divisions[divisionid].Stations[stationid3].ID, stationid3)
self.assertEqual(n.Stations[stationid3].Company.ID, companyid)
self.assertEqual(n.Stations[stationid3].Division.ID, divisionid)
return
