不要怂,就是干,撸起袖子干!

associations.test.js 6.52 KB
var chai      = require('chai')
  , expect    = chai.expect
  , Support   = require(__dirname + '/../support')
  , dialect   = Support.getTestDialect()
  , config    = require(__dirname + '/../config/config')
  , DataTypes = require(__dirname + "/../../lib/data-types")

chai.config.includeStack = true

if (dialect.match(/^postgres/)) {
  describe('[POSTGRES Specific] associations', function() {
    describe('many-to-many', function() {
      describe('where tables have the same prefix', function() {
        it("should create a table wp_table1wp_table2s", function(done) {
          var Table2 = this.sequelize.define('wp_table2', {foo: DataTypes.STRING})
            , Table1 = this.sequelize.define('wp_table1', {foo: DataTypes.STRING})

          Table1.hasMany(Table2)
          Table2.hasMany(Table1)

          expect(this.sequelize.daoFactoryManager.getDAO('wp_table1swp_table2s')).to.exist
          done()
        })
      })

      describe('when join table name is specified', function() {
        beforeEach(function(done){
          var Table2 = this.sequelize.define('ms_table1', {foo: DataTypes.STRING})
            , Table1 = this.sequelize.define('ms_table2', {foo: DataTypes.STRING})

          Table1.hasMany(Table2, {joinTableName: 'table1_to_table2'})
          Table2.hasMany(Table1, {joinTableName: 'table1_to_table2'})

          setTimeout(function () {
            done()
          }, 50)
        })

        it("should not use a combined name", function(done) {
          expect(this.sequelize.daoFactoryManager.getDAO('ms_table1sms_table2s')).not.to.exist
          setTimeout(function () {
            done()
          }, 50)
        })

        it("should use the specified name", function(done) {
          expect(this.sequelize.daoFactoryManager.getDAO('table1_to_table2')).to.exist
          setTimeout(function () {
            done()
          }, 50)
        })
      })
    })

    describe('HasMany', function() {
      describe('addDAO / getDAO', function() {
        beforeEach(function(done) {
          var self = this

          //prevent periods from occurring in the table name since they are used to delimit (table.column)
          this.User  = this.sequelize.define('User' + config.rand(), { name: DataTypes.STRING })
          this.Task  = this.sequelize.define('Task' + config.rand(), { name: DataTypes.STRING })
          this.users = null
          this.tasks = null

          this.User.hasMany(this.Task, {as:'Tasks', through: 'usertasks'})
          this.Task.hasMany(this.User, {as:'Users', through: 'usertasks'})

          var self = this
            , users = []
            , tasks = []

          for (var i = 0; i < 5; ++i) {
            users[users.length] = {name: 'User' + Math.random()}
          }

          for (var x = 0; x < 5; ++x) {
            tasks[tasks.length] = {name: 'Task' + Math.random()}
          }

          this.sequelize.sync({ force: true }).success(function() {
            self.User.bulkCreate(users).success(function() {
              self.Task.bulkCreate(tasks).success(function() {
                self.User.all().success(function(_users) {
                  self.Task.all().success(function(_tasks) {
                    self.user = _users[0]
                    self.task = _tasks[0]
                    done()
                  })
                })
              })
            })
          })
        })

        it('should correctly add an association to the dao', function(done) {
          var self = this

          self.user.getTasks().on('success', function(_tasks) {
            expect(_tasks).to.have.length(0)
            self.user.addTask(self.task).on('success', function() {
              self.user.getTasks().on('success', function(_tasks) {
                expect(_tasks).to.have.length(1)
                done()
              })
            })
          })
        })
      })

      describe('removeDAO', function() {
        it("should correctly remove associated objects", function(done) {
          var self = this
            , users = []
            , tasks = []

          //prevent periods from occurring in the table name since they are used to delimit (table.column)
          this.User  = this.sequelize.define('User' + config.rand(), { name: DataTypes.STRING })
          this.Task  = this.sequelize.define('Task' + config.rand(), { name: DataTypes.STRING })
          this.users = null
          this.tasks = null

          this.User.hasMany(this.Task, {as:'Tasks', through: 'usertasks'})
          this.Task.hasMany(this.User, {as:'Users', through: 'usertasks'})

          for (var i = 0; i < 5; ++i) {
            users[users.length] = {id: i+1, name: 'User' + Math.random()}
          }

          for (var x = 0; x < 5; ++x) {
            tasks[tasks.length] = {id: x+1, name: 'Task' + Math.random()}
          }

          this.sequelize.sync({ force: true }).success(function() {
            self.User.bulkCreate(users).done(function(err) {
              expect(err).not.to.be.ok
              self.Task.bulkCreate(tasks).done(function(err) {
                expect(err).not.to.be.ok
                self.User.all().success(function(_users) {
                  self.Task.all().success(function(_tasks) {
                    self.user = _users[0]
                    self.task = _tasks[0]
                    self.users = _users
                    self.tasks = _tasks

                    self.user.getTasks().on('success', function(__tasks) {
                      expect(__tasks).to.have.length(0)
                      self.user.setTasks(self.tasks).on('success', function() {
                        self.user.getTasks().on('success', function(_tasks) {
                          expect(_tasks).to.have.length(self.tasks.length)
                          self.user.removeTask(self.tasks[0]).on('success', function() {
                            self.user.getTasks().on('success', function(_tasks) {
                              expect(_tasks).to.have.length(self.tasks.length - 1)
                              self.user.removeTasks([self.tasks[1], self.tasks[2]]).on('success', function() {
                                self.user.getTasks().on('success', function(_tasks) {
                                  expect(_tasks).to.have.length(self.tasks.length - 3)
                                  done()
                                })
                              })
                            })
                          })
                        })
                      })
                    })
                  })
                })
              })
            })
          })
        })
      })
    })
  })
}