Add calculateLength() code

Signed-off-by: Fred Boniface <fred@fjla.uk>

src/utils/translators/ldb/staffStation.ts

@@ -105,7 +105,15 @@ function transformLocation(input: any): ServiceLocation[] {
 }
 
 function calculateLength(input: TrainServices): number | undefined {
-  console.log(input);
+  let length: number;
+  if (input?.length) {
+    length = input.length
+    return length
+  }
+  if (input?.formation?.coaches?.coach) {
+    length = input.formation.coaches.coach.length
+    return length
+  }
   return undefined
 }
 

test/utils/translators/ldb/staffStation.utils.test.ts

@@ -19,3 +19,5 @@ describe('transform', () => {
     });
   }
 });
+
+// Write test for calculateLength(input: TrainServices): number | undefined