Extend the schema for sessions

Screwtapello · Dec 1, 2012 · 3d921d9 · 3d921d9
1 parent 881dbd5
commit 3d921d9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/lib/schema.js b/lib/schema.js
@@ -16,7 +16,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-var _ = require("underscore"),
+var express = require("express"),
+    _ = require("underscore"),
     Activity = require("./model/activity").Activity,
     ActivityObject = require("./model/activityobject").ActivityObject,
     User = require("./model/user").User,
@@ -28,6 +29,7 @@ var _ = require("underscore"),
     Nonce = require("./model/nonce").Nonce,
     Credentials = require("./model/credentials").Credentials,
     DialbackRequest = require("./model/dialbackrequest").DialbackRequest,
+    DatabankStore = require('connect-databank')(express),
     Other = require("./model/other").Other;
 
 var getSchema = function() {
@@ -49,6 +51,8 @@ var getSchema = function() {
     schema[Other.type] = Other.schema;
     schema[DialbackRequest.type] = DialbackRequest.schema;
 
+    _.extend(schema, DatabankStore.schema);
+
     for (i = 0; i < ActivityObject.objectTypes.length; i++) {
         type = ActivityObject.objectTypes[i];
         Cls = ActivityObject.toClass(type);

diff --git a/test/schema-test.js b/test/schema-test.js
@@ -67,7 +67,8 @@ var types = [
     "credentials",
     "other",
     "dialbackrequest",
-    "session"
+    "session",
+    "sessionlist"
 ];
 
 vows.describe("schema module interface").addBatch({