Sequelize/packages/core/test/integration/model/findAll/group.test.js

'use strict';

const chai = require('chai');

const expect = chai.expect;
const Support = require('../../support');

const { DataTypes, Sequelize } = require('@sequelize/core');

const current = Support.sequelize;

describe(Support.getTestDialectTeaser('Model'), () => {
  describe('findAll', () => {
    describe('group', () => {
      it('should correctly group with attributes, #3009', async () => {
        const Post = current.define('Post', {
          id: { type: DataTypes.INTEGER, autoIncrement: true, primaryKey: true },
          name: { type: DataTypes.STRING, allowNull: false },
        });

        const Comment = current.define('Comment', {
          id: { type: DataTypes.INTEGER, autoIncrement: true, primaryKey: true },
          text: { type: DataTypes.STRING, allowNull: false },
        });

        Post.hasMany(Comment);

        await current.sync({ force: true });

        // Create an enviroment
        await Post.bulkCreate([{ name: 'post-1' }, { name: 'post-2' }]);

        await Comment.bulkCreate([
          { text: 'Market', postId: 1 },
          { text: 'Text', postId: 2 },
          { text: 'Abc', postId: 2 },
          { text: 'Semaphor', postId: 1 },
          { text: 'Text', postId: 1 },
        ]);

        const posts = await Post.findAll({
          attributes: [[Sequelize.fn('COUNT', Sequelize.col('comments.id')), 'comment_count']],
          include: [{ model: Comment, attributes: [] }],
          group: ['Post.id'],
          order: [['id']],
        });

        expect(Number.parseInt(posts[0].get('comment_count'), 10)).to.equal(3);
        expect(Number.parseInt(posts[1].get('comment_count'), 10)).to.equal(2);
      });

      it('should not add primary key when grouping using a belongsTo association', async () => {
        const Post = current.define('Post', {
          id: { type: DataTypes.INTEGER, autoIncrement: true, primaryKey: true },
          name: { type: DataTypes.STRING, allowNull: false },
        });

        const Comment = current.define('Comment', {
          id: { type: DataTypes.INTEGER, autoIncrement: true, primaryKey: true },
          text: { type: DataTypes.STRING, allowNull: false },
        });

        Post.hasMany(Comment);
        Comment.belongsTo(Post);

        await current.sync({ force: true });

        await Post.bulkCreate([{ name: 'post-1' }, { name: 'post-2' }]);

        await Comment.bulkCreate([
          { text: 'Market', postId: 1 },
          { text: 'Text', postId: 2 },
          { text: 'Abc', postId: 2 },
          { text: 'Semaphor', postId: 1 },
          { text: 'Text', postId: 1 },
        ]);

        const posts = await Comment.findAll({
          attributes: [
            'postId',
            [Sequelize.fn('COUNT', Sequelize.col('Comment.id')), 'comment_count'],
          ],
          include: [{ model: Post, attributes: [] }],
          group: ['postId'],
          order: [['postId']],
        });

        expect(posts[0].get().hasOwnProperty('id')).to.equal(false);
        expect(posts[1].get().hasOwnProperty('id')).to.equal(false);
        expect(Number.parseInt(posts[0].get('comment_count'), 10)).to.equal(3);
        expect(Number.parseInt(posts[1].get('comment_count'), 10)).to.equal(2);
      });
    });
  });
});