Add test and fix forcedSchema

TransitApp · Mar 9, 2018 · b1cfbd4 · b1cfbd4
1 parent 9b3f445
commit b1cfbd4
Show file tree

Hide file tree

Showing 17 changed files with 41 additions and 3 deletions.
diff --git a/gtfs.js b/gtfs.js
@@ -213,7 +213,7 @@ class Gtfs {
     this._shouldThrow = throws;
     this._postImportTableFunction = postImportTableFunction;
     this._tables = new Map();
-    this._schema = defaultSchema || forcedSchema;
+    this._schema = forcedSchema || defaultSchema;
   }
 
   /* Input/Output */

diff --git a/package.json b/package.json
@@ -4,7 +4,7 @@
   "description": "A Node.js library for GTFS",
   "main": "index.js",
   "scripts": {
-    "test": "mocha tests.js",
+    "test": "mocha tests/*",
     "publish-major": "npm version major && npm publish --access public && git push --tags",
     "publish-minor": "npm version minor && npm publish --access public && git push --tags",
     "publish-patch": "npm version patch && npm publish --access public && git push --tags"

diff --git a/tests/gtfs_options.js b/tests/gtfs_options.js
@@ -0,0 +1,36 @@
+'use strict';
+
+/* Run the tests with mocha: mocha tests.js */
+
+// eslint-disable-next-line import/no-extraneous-dependencies
+const { expect } = require('chai');
+
+const { Gtfs } = require('../index');
+
+describe('Tests on GTFS constructor options', () => {
+  it('Test on forcedSchema', (done) => {
+    const forcedSchema = Gtfs.getDefaultSchema();
+    forcedSchema.deepnessByTableName.modes = 1;
+    forcedSchema.indexKeysByTableName.modes = { indexKey: 'mode_id' };
+    forcedSchema.keysByTableName.modes = [
+      'mode_id',
+      'mode_name',
+      'mode_url',
+      'mode_timezone',
+      'mode_lang',
+      'mode_phone',
+      'mode_fare_url',
+      'mode_email',
+    ];
+    forcedSchema.tableNames.push('modes');
+
+    const path = `${__dirname}/samples/1/`;
+    const gtfs = new Gtfs(path, { forcedSchema });
+
+    const mode = gtfs.getItemWithIndexInTable('mode_0', 'modes');
+
+    expect(mode.mode_name).to.equal('mode 0');
+
+    done();
+  });
+});
diff --git a/samples/1/agency.txt → tests/samples/1/agency.txt b/samples/1/agency.txt → tests/samples/1/agency.txt
diff --git a/samples/1/calendar.txt → tests/samples/1/calendar.txt b/samples/1/calendar.txt → tests/samples/1/calendar.txt
diff --git a/samples/1/calendar_dates.txt → tests/samples/1/calendar_dates.txt b/samples/1/calendar_dates.txt → tests/samples/1/calendar_dates.txt
diff --git a/samples/1/feed_info.txt → tests/samples/1/feed_info.txt b/samples/1/feed_info.txt → tests/samples/1/feed_info.txt
diff --git a/samples/1/frequencies.txt → tests/samples/1/frequencies.txt b/samples/1/frequencies.txt → tests/samples/1/frequencies.txt
diff --git a/tests/samples/1/modes.txt b/tests/samples/1/modes.txt
@@ -0,0 +1,2 @@
+mode_id,mode_name,mode_url,mode_timezone,mode_lang,mode_phone,mode_fare_url,mode_email
+mode_0,mode 0,http://google.com,America/New_York,en,(310) 555-0222,http://google.com,contact@google.com
diff --git a/samples/1/routes.txt → tests/samples/1/routes.txt b/samples/1/routes.txt → tests/samples/1/routes.txt
diff --git a/samples/1/shapes.txt → tests/samples/1/shapes.txt b/samples/1/shapes.txt → tests/samples/1/shapes.txt
diff --git a/samples/1/stop_times.txt → tests/samples/1/stop_times.txt b/samples/1/stop_times.txt → tests/samples/1/stop_times.txt
diff --git a/samples/1/stops.txt → tests/samples/1/stops.txt b/samples/1/stops.txt → tests/samples/1/stops.txt
diff --git a/samples/1/transfers.txt → tests/samples/1/transfers.txt b/samples/1/transfers.txt → tests/samples/1/transfers.txt
diff --git a/samples/1/trips.txt → tests/samples/1/trips.txt b/samples/1/trips.txt → tests/samples/1/trips.txt
diff --git a/samples/2/stops.txt → tests/samples/2/stops.txt b/samples/2/stops.txt → tests/samples/2/stops.txt
diff --git a/tests.js → tests/tests.js b/tests.js → tests/tests.js
@@ -6,7 +6,7 @@
 const { expect } = require('chai');
 const fs = require('fs-extra');
 
-const { Gtfs } = require('./index');
+const { Gtfs } = require('../index');
 
 describe('Tests on GTFS', () => {
   it('Test on meta functions', (done) => {
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		mode_id,mode_name,mode_url,mode_timezone,mode_lang,mode_phone,mode_fare_url,mode_email
		mode_0,mode 0,http://google.com,America/New_York,en,(310) 555-0222,http://google.com,contact@google.com