diff --git a/weboob/backends/transilien/test.py b/weboob/backends/transilien/test.py index 3691a219..0eb39941 100644 --- a/weboob/backends/transilien/test.py +++ b/weboob/backends/transilien/test.py @@ -17,7 +17,9 @@ # You should have received a copy of the GNU Affero General Public License # along with weboob. If not, see . +import datetime +from weboob.capabilities.travel import RoadmapFilters from weboob.tools.test import BackendTest class TransilienTest(BackendTest): @@ -30,5 +32,10 @@ class TransilienTest(BackendTest): list(self.backend.iter_station_departures(stations[0].id)) def test_roadmap(self): - roadmap = list(self.backend.iter_roadmap('Puteaux', u'École Militaire')) + filters = RoadmapFilters() + roadmap = list(self.backend.iter_roadmap('Puteaux', u'École Militaire', filters)) + self.assertTrue(len(roadmap) > 0) + + filters.arrival_time = datetime.datetime.now() + datetime.timedelta(days=1) + roadmap = list(self.backend.iter_roadmap('Puteaux', u'Aulnay-sous-Bois', filters)) self.assertTrue(len(roadmap) > 0)