Refs #2751: Add UPDATE_TOTAL_COUNT for vj

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

2 files changed, 19 insertions, 0 deletions

diff --git a/spec/javascripts/vehicle_journeys/actions_spec.js b/spec/javascripts/vehicle_journeys/actions_spec.js
index 8817bc34c..ec52036d6 100644
--- a/spec/javascripts/vehicle_journeys/actions_spec.js
+++ b/spec/javascripts/vehicle_journeys/actions_spec.js
@@ -382,3 +382,13 @@ describe('when user clicked either on filter or reset button in filters', () =>
     expect(actions.createQueryString()).toEqual(expectedAction)
   })
 })
+describe('when submitting new vj', () => {
+  it('should create an action to update pagination totalCount', () => {
+    const diff = 1
+    const expectedAction = {
+      type: 'UPDATE_TOTAL_COUNT',
+      diff
+    }
+    expect(actions.updateTotalCount(diff)).toEqual(expectedAction)
+  })
+})
diff --git a/spec/javascripts/vehicle_journeys/reducers/pagination_spec.js b/spec/javascripts/vehicle_journeys/reducers/pagination_spec.js
index 55e48dfd6..ff74e1a0d 100644
--- a/spec/javascripts/vehicle_journeys/reducers/pagination_spec.js
+++ b/spec/javascripts/vehicle_journeys/reducers/pagination_spec.js
@@ -26,6 +26,15 @@ describe('pagination reducer, given parameters allowing page change', () => {
     ).toEqual(Object.assign({}, state, {page: 1}))
   })
 
+  it('should handle UPDATE_TOTAL_COUNT', () => {
+    expect(
+      reducer(state, {
+        type: 'UPDATE_TOTAL_COUNT',
+        diff: 1
+      })
+    ).toEqual(Object.assign({}, state, {totalCount: 24}))
+  })
+
   it('should handle GO_TO_NEXT_PAGE and change state', () => {
     expect(
       reducer(state, {