import * as assert from 'assert'
import {parseMapWithSubquerySelection} from '../../../lib/parsing/selection/maps_selection_parsing'
import {createMapSelection} from '../../../lib/parsing/selection/map_selection_parsing'
import {createParameterlessFilter} from '../../../lib/parsing/filtering/parameterless_filter_parsing'
import {createSubselectStatement} from '../../../lib/statements/subselect_statement'
import {createCountSelection} from '../../../lib/parsing/selection/count_selection'
import {createEqual, createGreaterThan} from '../../../lib/parsing/boolean_expressions/comparisons'
import {createGetColumn} from '../../../lib/parsing/value_expressions/get_column_parsing'
import {employees} from '../../../lib/test/test_tables'

describe('parseMapS can parse a map with a subquery', function () {

    it('with one filter', () => {
        const actual = parseMapWithSubquerySelection('employees', employees.columns, (st, e) => ({
            id: e.id,
            higherSalary: st.filter(se => se.salary > e.salary).count()
        }))

        const expectedSubselectStatement = createSubselectStatement(
            'employees',
            employees.columns,
            [
                createParameterlessFilter(
                    {se: 's1', e: 't1'},
                    createGreaterThan(createGetColumn('se', 'salary'), createGetColumn('e', 'salary'))
                )
            ],
            createCountSelection())

        const expected = createMapSelection(
            {
                e: 't1'
            },
            [
                ['id', createGetColumn('e', 'id')],
                ['higherSalary', expectedSubselectStatement]
            ])

        assert.deepEqual(actual, expected)
    })

    it('with two filters', () => {
        const actual = parseMapWithSubquerySelection('employees', employees.columns, (st, e) => ({
            id: e.id,
            higherSalary: st.filter(se => se.salary > e.salary).filter(se => se.departmentId === e.departmentId).count()
        }))

        const tableParameterNameToTableAlias = {se: 's1', e: 't1'}
        const expectedSubselectStatement = createSubselectStatement(
            'employees',
            employees.columns,
            [
                createParameterlessFilter(
                    tableParameterNameToTableAlias,
                    createGreaterThan(createGetColumn('se', 'salary'), createGetColumn('e', 'salary'))
                ),
                createParameterlessFilter(
                    tableParameterNameToTableAlias,
                    createEqual(createGetColumn('se', 'departmentId'), createGetColumn('e', 'departmentId'))
                )
            ],
            createCountSelection())

        const expected = createMapSelection(
            {
                e: 't1'
            },
            [
                ['id', createGetColumn('e', 'id')],
                ['higherSalary', expectedSubselectStatement]
            ])

        assert.deepEqual(actual, expected)
    })
})