'use strict';

const expect = require('chai').expect;
const { isMssql, isOracle, isPgBased, isSQLite } = require('../../util/db-helpers');

module.exports = function (knex) {
  describe('unions', function () {
    it('handles unions with a callback', function () {
      return knex('accounts')
        .select('*')
        .where('id', '=', 1)
        .union(function () {
          this.select('*').from('accounts').where('id', 2);
        });
    });

    it('handles unions with an array of callbacks', function () {
      return knex('accounts')
        .select('*')
        .where('id', '=', 1)
        .union([
          function () {
            this.select('*').from('accounts').where('id', 2);
          },
          function () {
            this.select('*').from('accounts').where('id', 3);
          },
        ]);
    });

    it('handles unions with a list of callbacks', function () {
      return knex('accounts')
        .select('*')
        .where('id', '=', 1)
        .union(
          function () {
            this.select('*').from('accounts').where('id', 2);
          },
          function () {
            this.select('*').from('accounts').where('id', 3);
          }
        );
    });

    it('handles unions with an array of builders', function () {
      return knex('accounts')
        .select('*')
        .where('id', '=', 1)
        .union([
          knex.select('*').from('accounts').where('id', 2),
          knex.select('*').from('accounts').where('id', 3),
        ]);
    });

    it('handles unions with a list of builders', function () {
      return knex('accounts')
        .select('*')
        .where('id', '=', 1)
        .union(
          knex.select('*').from('accounts').where('id', 2),
          knex.select('*').from('accounts').where('id', 3)
        );
    });

    it('handles unions with a raw query', function () {
      return knex('accounts')
        .select('*')
        .where('id', '=', 1)
        .union(
          knex.raw('select * from ?? where ?? = ?', ['accounts', 'id', 2])
        );
    });

    it('handles unions with an array raw queries', function () {
      return knex('accounts')
        .select('*')
        .where('id', '=', 1)
        .union([
          knex.raw('select * from ?? where ?? = ?', ['accounts', 'id', 2]),
          knex.raw('select * from ?? where ?? = ?', ['accounts', 'id', 3]),
        ]);
    });

    it('handles unions with a list of raw queries', function () {
      return knex('accounts')
        .select('*')
        .where('id', '=', 1)
        .union(
          knex.raw('select * from ?? where ?? = ?', ['accounts', 'id', 2]),
          knex.raw('select * from ?? where ?? = ?', ['accounts', 'id', 3])
        );
    });
  });

  if (isPgBased(knex) || isMssql(knex) || isOracle(knex) || isSQLite(knex)) {
    describe('intersects', function () {
      before(function () {
        return knex.schema.createTable('intersect_test', function (t) {
          t.integer('id');
          t.integer('test_col_1');
          t.integer('test_col_2');
          t.integer('test_col_3');
        });
      });

      beforeEach(function () {
        return knex('intersect_test').insert([
          {
            id: 1,
            test_col_1: 1,
            test_col_2: 2,
            test_col_3: 1,
          },
          {
            id: 2,
            test_col_1: 2,
            test_col_2: 3,
            test_col_3: 1,
          },
          {
            id: 3,
            test_col_1: 2,
            test_col_2: 3,
            test_col_3: 2,
          },
          {
            id: 4,
            test_col_1: 1,
            test_col_2: 2,
            test_col_3: 2,
          },
          {
            id: 5,
            test_col_1: 1,
            test_col_2: 2,
            test_col_3: 1,
          },
        ]);
      });

      after(function () {
        return knex.schema.dropTable('intersect_test');
      });

      it('handles intersects with a callback', function () {
        return knex('intersect_test')
          .select('*')
          .where('test_col_1', '=', 1)
          .intersect(function () {
            this.select('*').from('intersect_test').where('test_col_2', 2);
          })
          .then(function (result) {
            expect(result.length).to.equal(3);
            expect(result.map((r) => r.id)).to.have.members([1, 4, 5]);
          });
      });

      it('handles intersects with an array of callbacks', function () {
        return knex('intersect_test')
          .select('*')
          .where('test_col_1', '=', 1)
          .intersect([
            function () {
              this.select('*').from('intersect_test').where('test_col_2', 2);
            },
            function () {
              this.select('*').from('intersect_test').where('test_col_3', 1);
            },
          ])
          .then(function (result) {
            expect(result.length).to.equal(2);
            expect(result.map((r) => r.id)).to.have.members([1, 5]);
          });
      });

      it('handles intersects with a list of callbacks', function () {
        return knex('intersect_test')
          .select('*')
          .where('test_col_1', '=', 1)
          .intersect(
            function () {
              this.select('*').from('intersect_test').where('test_col_2', 2);
            },
            function () {
              this.select('*').from('intersect_test').where('test_col_3', 1);
            }
          )
          .then(function (result) {
            expect(result.length).to.equal(2);
            expect(result.map((r) => r.id)).to.have.members([1, 5]);
          });
      });

      it('handles intersects with an array of builders', function () {
        return knex('intersect_test')
          .select('*')
          .where('test_col_1', '=', 1)
          .intersect([
            knex.select('*').from('intersect_test').where('test_col_2', 2),
            knex.select('*').from('intersect_test').where('test_col_3', 1),
          ])
          .then(function (result) {
            expect(result.length).to.equal(2);
            expect(result.map((r) => r.id)).to.have.members([1, 5]);
          });
      });

      it('handles intersects with a list of builders', function () {
        return knex('intersect_test')
          .select('*')
          .where('test_col_1', '=', 1)
          .intersect(
            knex.select('*').from('intersect_test').where('test_col_2', 2),
            knex.select('*').from('intersect_test').where('test_col_3', 1)
          )
          .then(function (result) {
            expect(result.length).to.equal(2);
            expect(result.map((r) => r.id)).to.have.members([1, 5]);
          });
      });

      it('handles intersects with a raw query', function () {
        return knex('intersect_test')
          .select('*')
          .where('test_col_1', '=', 2)
          .intersect(
            knex.raw('select * from ?? where ?? = ?', [
              'intersect_test',
              'test_col_2',
              3,
            ])
          )
          .then(function (result) {
            expect(result.length).to.equal(2);
            expect(result.map((r) => r.id)).to.have.members([2, 3]);
          });
      });

      it('handles intersects with an array raw queries', function () {
        return knex('intersect_test')
          .select('*')
          .where('test_col_1', '=', 1)
          .intersect([
            knex.raw('select * from ?? where ?? = ?', [
              'intersect_test',
              'test_col_2',
              2,
            ]),
            knex.raw('select * from ?? where ?? = ?', [
              'intersect_test',
              'test_col_3',
              1,
            ]),
          ])
          .then(function (result) {
            expect(result.length).to.equal(2);
            expect(result.map((r) => r.id)).to.have.members([1, 5]);
          });
      });

      it('handles intersects with a list of raw queries', function () {
        return knex('intersect_test')
          .select('*')
          .where('test_col_1', '=', 1)
          .intersect(
            knex.raw('select * from ?? where ?? = ?', [
              'intersect_test',
              'test_col_2',
              2,
            ]),
            knex.raw('select * from ?? where ?? = ?', [
              'intersect_test',
              'test_col_3',
              1,
            ])
          )
          .then(function (result) {
            expect(result.length).to.equal(2);
            expect(result.map((r) => r.id)).to.have.members([1, 5]);
          });
      });
    });
  }
};