Refs #2507: add timetable to filters

Signed-off-by: Thomas Shawarma Haddad <thomas.haddad@af83.com>

2 files changed, 28 insertions, 0 deletions

diff --git a/spec/javascripts/vehicle_journeys/actions_spec.js b/spec/javascripts/vehicle_journeys/actions_spec.js
index 301f5d8ba..6a868baf9 100644
--- a/spec/javascripts/vehicle_journeys/actions_spec.js
+++ b/spec/javascripts/vehicle_journeys/actions_spec.js
@@ -306,3 +306,21 @@ describe('when setting new interval', () => {
     expect(actions.updateEndTimeFilter(val, unit)).toEqual(expectedAction)
   })
 })
+describe('when using select2 to pick a timetable in the filters', () => {
+  it('should create an action to select a timetable as a filter', () => {
+    let selectedTT = {
+      id: 1,
+      objectid: 2,
+      comment: 'test',
+    }
+    const expectedAction = {
+      type: 'SELECT_TT_FILTER',
+      selectedItem:{
+        id: selectedTT.id,
+        objectid: selectedTT.objectid,
+        comment: selectedTT.comment,
+      }
+    }
+    expect(actions.filterSelect2Timetable(selectedTT)).toEqual(expectedAction)
+  })
+})
diff --git a/spec/javascripts/vehicle_journeys/reducers/filters_spec.js b/spec/javascripts/vehicle_journeys/reducers/filters_spec.js
index 53ac5d75b..7d62e01f2 100644
--- a/spec/javascripts/vehicle_journeys/reducers/filters_spec.js
+++ b/spec/javascripts/vehicle_journeys/reducers/filters_spec.js
@@ -120,4 +120,14 @@ describe('filters reducer', () => {
     ).toEqual(state)
   })
 
+  it('should handle SELECT_TT_FILTER', () => {
+    let newTimetable = {timetable : {id: 1}}
+    let newQuery = Object.assign({}, state.query, newTimetable)
+    expect(
+      statusReducer(state, {
+        type: 'SELECT_TT_FILTER',
+        selectedItem: {id: 1}
+      })
+    ).toEqual(Object.assign({}, state, {query: newQuery}))
+  })
 })