Extend test coverage

linkedconnections · Jul 7, 2022 · d282861 · d282861
1 parent c440b9f
commit d282861
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/lib/Gtfsrt2LC.js b/lib/Gtfsrt2LC.js
@@ -150,7 +150,7 @@ class Gtfsrt2LC {
 
                            // Check if this is a new update based on recorded Connections history (if any)
                            // and skip it if is not new.
-                           if (this.historyDB && !(await this.differentiaUpdate({
+                           if (this.historyDB && !(await this.differentialUpdate({
                               route: r,
                               trip: t,
                               stopTimes: st,
@@ -626,7 +626,7 @@ class Gtfsrt2LC {
       return update;
    }
 
-   async differentiaUpdate(params) {
+   async differentialUpdate(params) {
       const {
          route,
          trip,

diff --git a/test/gtfsrt2lc.test.js b/test/gtfsrt2lc.test.js
@@ -582,4 +582,19 @@ test('Cover GtfsIndex functions', async () => {
     try {
         await gti.getIndexes();
     } catch (err) { }
+});
+
+test('Cover Gtfsrt2LC functions', async () => {
+    grt = new Gtfsrt2lc({ path: rt_path, uris: mock_uris, headers: {} });
+    // Test for resolveScheduleRelationship
+    const notAvailable = grt.resolveScheduleRelationship(1, 1);
+    const mustPhone = grt.resolveScheduleRelationship(0, 2);
+    const mustCoordinate = grt.resolveScheduleRelationship(0, 3);
+    // Test for getting headers
+    const headers = grt.headers();
+
+    expect(notAvailable).toBe('gtfs:NotAvailable');
+    expect(mustPhone).toBe('gtfs:MustPhone');
+    expect(mustCoordinate).toBe('gtfs:MustCoordinateWithDriver');
+    expect(headers).toBeDefined();
 });