Prevent not following the instructions

Author: SleeplessByte

exercises/concept/train-driver/.meta/exemplar.js

@@ -17,7 +17,7 @@ export function getListOfWagons(...ids) {
 /**
  * Reorder the array of wagons by moving the first 2 wagons to the end of the array.
  *
- * @param {number[]} ids
+ * @param {Iterable<number>} ids
  * @returns {number[]} reordered list of wagons
  */
 export function fixListOfWagons([first, second, ...rest]) {
@@ -27,8 +27,8 @@ export function fixListOfWagons([first, second, ...rest]) {
 /**
  * Fixes the array of wagons by inserting an array of wagons after the first element in eachWagonsID.
  *
- * @param {number[]} ids
- * @param {number[]} missingWagons
+ * @param {Iterable<number>} ids
+ * @param {Iterable<number>} missingWagons
  * @returns {number[]} corrected list of wagons
  */
 export function correctListOfWagons([first, ...rest], missingWagons) {

exercises/concept/train-driver/train-driver.js

@@ -17,7 +17,7 @@ export function getListOfWagons(a, b, c, d, e, f, g, h, i, j, k, l, m, n) {
 /**
  * Reorder the array of wagons by moving the first 2 wagons to the end of the array.
  *
- * @param {number[]} ids
+ * @param {Iterable<number>} ids
  * @returns {number[]} reorderd list of wagons
  */
 export function fixListOfWagons(ids) {
@@ -27,8 +27,8 @@ export function fixListOfWagons(ids) {
 /**
  * Fixes the array of wagons by inserting an array of wagons after the first element in eachWagonsID.
  *
- * @param {number[]} ids
- * @param {number[]} missingWagons
+ * @param {Iterable<number>} ids
+ * @param {Iterable<number>} missingWagons
  * @returns {number[]} corrected list of wagons
  */
 export function correctListOfWagons(ids, missingWagons) {

exercises/concept/train-driver/train-driver.spec.js

@@ -8,6 +8,20 @@ import {
   separateTimeOfArrival,
 } from './train-driver';
 
+class LimitedArray {
+  constructor(values) {
+    this.values = values;
+  }
+
+  [Symbol.iterator]() {
+    return this.values[Symbol.iterator]();
+  }
+}
+
+function list(...values) {
+  return new LimitedArray(values);
+}
+
 describe('getListOfWagons', () => {
   test('returns the correct array', () => {
     expect(getListOfWagons(1, 5, 2, 7, 4)).toEqual([1, 5, 2, 7, 4]);
@@ -30,29 +44,29 @@ describe('getListOfWagons', () => {
 
 describe('fixListOfWagons', () => {
   test('reorders the first 2 wagons to the end of the array', () => {
-    const eachWagonsID = [3, 7, 1, 14, 10, 4, 12, 6, 23, 17, 13, 20, 8, 19];
+    const eachWagonsID = list(3, 7, 1, 14, 10, 4, 12, 6, 23, 17, 13, 20, 8, 19);
     const expected = [1, 14, 10, 4, 12, 6, 23, 17, 13, 20, 8, 19, 3, 7];
 
     expect(fixListOfWagons(eachWagonsID)).toEqual(expected);
   });
 
   test('works when only 3 wagons given', () => {
-    const eachWagonsID = [4, 2, 1];
+    const eachWagonsID = list(4, 2, 1);
 
     expect(fixListOfWagons(eachWagonsID)).toEqual([1, 4, 2]);
   });
 
   test('works for a few wagons', () => {
-    const eachWagonsID = [3, 4, 1, 5, 7, 9, 10];
+    const eachWagonsID = list(3, 4, 1, 5, 7, 9, 10);
 
     expect(fixListOfWagons(eachWagonsID)).toEqual([1, 5, 7, 9, 10, 3, 4]);
   });
 });
 
 describe('correctListOfWagons', () => {
   test('returns a wagon weight list with the inserted array of values', () => {
-    const eachWagonsID = [1, 6, 11, 15, 13, 14, 17, 22, 2, 16, 19, 21];
-    const missingWagons = [8, 10, 5, 9, 3, 7, 20];
+    const eachWagonsID = list(1, 6, 11, 15, 13, 14, 17, 22, 2, 16, 19, 21);
+    const missingWagons = list(8, 10, 5, 9, 3, 7, 20);
     const expected = [
       1, 8, 10, 5, 9, 3, 7, 20, 6, 11, 15, 13, 14, 17, 22, 2, 16, 19, 21,
     ];
@@ -61,16 +75,16 @@ describe('correctListOfWagons', () => {
   });
 
   test('works for short arrays', () => {
-    const eachWagonsID = [1, 7, 15, 24];
-    const missingWagons = [8, 6, 4];
+    const eachWagonsID = list(1, 7, 15, 24);
+    const missingWagons = list(8, 6, 4);
     const expected = [1, 8, 6, 4, 7, 15, 24];
 
     expect(correctListOfWagons(eachWagonsID, missingWagons)).toEqual(expected);
   });
 
   test('works when missingWagons is longer', () => {
-    const eachWagonsID = [1, 7, 15, 24];
-    const missingWagons = [8, 6, 4, 5, 9, 21, 2, 13];
+    const eachWagonsID = list(1, 7, 15, 24);
+    const missingWagons = list(8, 6, 4, 5, 9, 21, 2, 13);
     const expected = [1, 8, 6, 4, 5, 9, 21, 2, 13, 7, 15, 24];
 
     expect(correctListOfWagons(eachWagonsID, missingWagons)).toEqual(expected);