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

errors.test.js 2.63 KB
'use strict';

const chai = require('chai');
const expect = chai.expect;
const Support = require('../../support');
const dialect = Support.getTestDialect();
const Sequelize = require('../../../../index');
const DataTypes = require('../../../../lib/data-types');

if (dialect === 'mysql') {
  describe('[MYSQL Specific] Errors', () => {

    const validateError = async (promise, errClass, errValues) => {
      const wanted = { ...errValues };

      await expect(promise).to.have.been.rejectedWith(errClass);

      try {
        return await promise;
      } catch (err) {
        return Object.keys(wanted).forEach(k => expect(err[k]).to.eql(wanted[k]));
      }
    };

    describe('ForeignKeyConstraintError', () => {
      beforeEach(function() {
        this.Task = this.sequelize.define('task', { title: DataTypes.STRING });
        this.User = this.sequelize.define('user', { username: DataTypes.STRING });
        this.UserTasks = this.sequelize.define('tasksusers', { userId: DataTypes.INTEGER, taskId: DataTypes.INTEGER });

        this.User.belongsToMany(this.Task, { onDelete: 'RESTRICT', through: 'tasksusers' });
        this.Task.belongsToMany(this.User, { onDelete: 'RESTRICT', through: 'tasksusers' });

        this.Task.belongsTo(this.User, { foreignKey: 'primaryUserId', as: 'primaryUsers' });
      });

      it('in context of DELETE restriction', async function() {
        await this.sequelize.sync({ force: true });

        const [user1, task1] = await Promise.all([
          this.User.create({ id: 67, username: 'foo' }),
          this.Task.create({ id: 52, title: 'task' })
        ]);

        await user1.setTasks([task1]);

        await Promise.all([
          validateError(user1.destroy(), Sequelize.ForeignKeyConstraintError, {
            fields: ['userId'],
            table: 'users',
            value: undefined,
            index: 'tasksusers_ibfk_1',
            reltype: 'parent'
          }),
          validateError(task1.destroy(), Sequelize.ForeignKeyConstraintError, {
            fields: ['taskId'],
            table: 'tasks',
            value: undefined,
            index: 'tasksusers_ibfk_2',
            reltype: 'parent'
          })
        ]);
      });

      it('in context of missing relation', async function() {
        await this.sequelize.sync({ force: true });

        await validateError(
          this.Task.create({ title: 'task', primaryUserId: 5 }),
          Sequelize.ForeignKeyConstraintError,
          {
            fields: ['primaryUserId'],
            table: 'users',
            value: 5,
            index: 'tasks_ibfk_1',
            reltype: 'child'
          }
        );
      });
    });
  });
}