Add test for postImportTableFunction

TransitApp · Mar 9, 2018 · 6ca3161 · 6ca3161
1 parent b1cfbd4
commit 6ca3161
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/gtfs_options.js b/tests/gtfs_options.js
@@ -33,4 +33,16 @@ describe('Tests on GTFS constructor options', () => {
 
     done();
   });
+
+  it('Test on postImportTableFunction', (done) => {
+    const path = `${__dirname}/samples/1/`;
+    const postImportTableFunction = (item) => { item.temp = 'some value'; };
+    const gtfs = new Gtfs(path, { postImportTableFunction });
+
+    const route = gtfs.getRouteWithId('route_0')
+
+    expect(route.temp).to.equal('some value');
+
+    done();
+  });
 });