mongo-query-to-postgres-jsonb/test/select.js

Converts MongoDB queries to postgresql queries for jsonb fields.

var assert = require('chai').assert
var mongoQueryPostgres = require('../index')
var convertSelect = mongoQueryPostgres.convertSelect

describe('select: ', function() {
  describe('wildcard', function () {
    it('wildcards', function () {
      assert.equal(convertSelect('data', undefined), 'data')
      assert.equal(convertSelect('data', {}), 'data')
    })
  })

  describe('named fields', function () {
    it('single field', function () {
      assert.equal(convertSelect('data', { field: 1 }), 'jsonb_build_object(\'field\', data->\'field\', \'_id\', data->\'_id\') as data')
    })
    it('single field exclude _id', function () {
      assert.equal(convertSelect('data', { field: 1, _id: 0 }), 'jsonb_build_object(\'field\', data->\'field\') as data')
    })
    it('nested field', function () {
      assert.equal(convertSelect('data', { 'field.inner': 1 }), 'jsonb_build_object(\'field\', jsonb_build_object(\'inner\', data->\'field\'->\'inner\'), \'_id\', data->\'_id\') as data')
    })
    it('nested field alternate', function () {
      assert.equal(convertSelect('data', { field: { inner: 1 } }), 'jsonb_build_object(\'field\', jsonb_build_object(\'inner\', data->\'field\'->\'inner\'), \'_id\', data->\'_id\') as data')
    })
    it('multiple fields', function () {
      assert.equal(convertSelect('data', { a: 1, b: 1 }), 'jsonb_build_object(\'a\', data->\'a\', \'b\', data->\'b\', \'_id\', data->\'_id\') as data')
    })
    it('multiple fields', function () {
      assert.equal(convertSelect('data', { a: 1, 'b.g': 1, 'b.abc': 1 }), 'jsonb_build_object(\'a\', data->\'a\', \'b\', jsonb_build_object(\'g\', data->\'b\'->\'g\', \'abc\', data->\'b\'->\'abc\'), \'_id\', data->\'_id\') as data')
    })
  })

  describe('excluded fields', function () {
    it('single excluded', function () {
      assert.equal(convertSelect('data', { b: 0 }), 'data #- \'{b}\' as data')
    })
    it('exclude deep', function () {
      assert.equal(convertSelect('data', { 'field.inner': 0 }), 'data #- \'{field,inner}\' as data')
    })
    it('exclude deep', function () {
      assert.equal(convertSelect('data', { 'field.inner': 0 }), 'data #- \'{field,inner}\' as data')
    })
    it('combined exclusion and inclusion', function () {
      assert.throws(() => convertSelect('data', { a: 1, b: 0 }), 'Projection cannot have a mix of inclusion and exclusion.')
    })
  })

  describe('array fields', function () {
    it('nothing special when requesting array field itself', function () {
      assert.equal(convertSelect('data', { 'arr': 1 }, ['arr']),
        'jsonb_build_object(\'arr\', data->\'arr\', \'_id\', data->\'_id\') as data')
    })

    it('single field', function () {
      assert.equal(convertSelect('data', { 'arr.color': 1 }, ['arr']),
        'jsonb_build_object(\'arr\', (SELECT jsonb_agg(r) FROM (SELECT jsonb_build_object(' +
          '\'color\', v->\'color\') as r FROM jsonb_array_elements(data->\'arr\') as v)' +
          ' AS obj), \'_id\', data->\'_id\') as data')
    })

    it('field within two nested arrays', function () {
      assert.equal(convertSelect('data', { 'arr.subarr.color': 1 }, ['arr', 'arr.subarr']),
        'jsonb_build_object(\'arr\', (SELECT jsonb_agg(r) FROM (SELECT jsonb_build_object(\'subarr\', ' +
          '(SELECT jsonb_agg(r) FROM (SELECT jsonb_build_object(\'color\', v->\'color\') as r FROM ' +
          'jsonb_array_elements(v->\'subarr\') as v) AS obj)) as r FROM jsonb_array_elements(data->\'arr\') ' +
          'as v) AS obj), \'_id\', data->\'_id\') as data')
    })

    it('should properly escape special characters in field names for array projections', function () {
      assert.equal(convertSelect('data', { 'stores.owner\'s_name': 1 }, ['stores']),
        'jsonb_build_object(\'stores\', (SELECT jsonb_agg(r) FROM (SELECT jsonb_build_object(' +
          '\'owner\'\'s_name\', v->\'owner\'\'s_name\') as r FROM jsonb_array_elements(data->\'stores\') as v)' +
          ' AS obj), \'_id\', data->\'_id\') as data')
    })

    it('should handle array field names with special characters', function () {
      assert.equal(convertSelect('data', { 'user\'s_stores': 1 }, ['user\'s_stores']),
        'jsonb_build_object(\'user\'\'s_stores\', data->\'user\'\'s_stores\', \'_id\', data->\'_id\') as data')
    })
  })
})