Merge pull request #868 from jwilm/model-accessor

Add Sequelize#model to fetch DAOFactory of previously defined model

lib/sequelize.js

@@ -195,6 +195,20 @@ module.exports = (function() {
     return factory
     return factory
   }
   }
 
 
+  /**
+   Fetch a DAO factory
+
+   @param {String} daoName The name of a model defined with Sequelize.define
+   @returns {DAOFactory} The DAOFactory for daoName
+   */
+  Sequelize.prototype.model = function(daoName) {
+    if(!this.isDefined(daoName)) {
+      throw new Error(daoName + ' has not been defined')
+    }
+
+    return this.daoFactoryManager.getDAO(daoName)
+  }
+
   Sequelize.prototype.isDefined = function(daoName) {
   Sequelize.prototype.isDefined = function(daoName) {
     var daos = this.daoFactoryManager.daos
     var daos = this.daoFactoryManager.daos
     return (daos.filter(function(dao) { return dao.name === daoName }).length !== 0)
     return (daos.filter(function(dao) { return dao.name === daoName }).length !== 0)

test/sequelize.test.js

@@ -51,6 +51,23 @@ describe(Support.getTestDialectTeaser("Sequelize"), function () {
     })
     })
   })
   })
 
 
+  describe('model', function() {
+    it('throws an error if the dao being accessed is undefined', function() {
+      var self = this
+      expect(function() {
+        self.sequelize.model('Project')
+      }).to.throw(/project has not been defined/i)
+    })
+
+    it('returns the dao factory defined by daoName', function() {
+      var project = this.sequelize.define('Project', {
+        name: DataTypes.STRING
+      })
+
+      expect(this.sequelize.model('Project')).to.equal(project)
+    })
+  })
+
   describe('query', function() {
   describe('query', function() {
     afterEach(function(done) {
     afterEach(function(done) {
       this.sequelize.options.quoteIdentifiers = true
       this.sequelize.options.quoteIdentifiers = true