add a test for pad

cds-snc · Dec 2, 2019 · 4f1a474 · 4f1a474
1 parent 3621724
commit 4f1a474
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/utils/context.helpers.spec.js b/utils/context.helpers.spec.js
@@ -23,6 +23,22 @@ const res = {
   },
 }
 
+test('can pad an array', async () => {
+  await new Promise(resolve => contextMiddleware(req, res, resolve))
+
+  const pad = res.locals.pad
+
+  expect(pad([], 3)).toEqual([{}, {}, {}])
+  expect(pad([], 0)).toEqual([{}])
+  expect(pad(undefined, 3)).toEqual([{}, {}, {}])
+
+  expect(pad([1, 2, 3], 0)).toEqual([1, 2, 3])
+  expect(pad([1, 2, 3], 3)).toEqual([1, 2, 3])
+  expect(pad([1, 2, 3], 4)).toEqual([1, 2, 3, {}])
+
+  expect(pad(undefined, undefined)).toEqual([{}])
+})
+
 test('can traverse a data structure with no context', async () => {
   await new Promise(resolve => contextMiddleware(req, res, resolve))