import { assert } from 'chai';
import { Project } from 'ts-morph';
import { upgradeProjectStringParameters } from '../../src/upgrade/upgradeRunner';

describe('upgradeProjectStringParameters', function() {
    this.timeout(1000000);
    it('should upgrade where', async () => {


        const project = new Project({
            tsConfigFilePath: './upgradeTestProjects/v2-v3-stringParameters/tsconfig.json',
        });


        const code = `
            import { ITypedQueryBuilder } from './typedKnexTypes';

            const a = {} as ITypedQueryBuilder<{}, {}, {}>;
            a.where(i => i.name, 'this name1');
            a.where(i => i.other.id, 'id1');
        `;


        const sourceFile = project.createSourceFile('./upgradeTestProjects/v2-v3-stringParameters/src/test.ts', code);

        assert.equal(project.getPreEmitDiagnostics().length, 0);

        upgradeProjectStringParameters(project);

        assert.equal(sourceFile.getText(), `import { ITypedQueryBuilder } from './typedKnexTypes';

            const a = {} as ITypedQueryBuilder<{}, {}, {}>;
            a.where('name', 'this name1');
            a.where('other.id', 'id1');
        `)


    });

    it('should upgrade select single column', async () => {


        const project = new Project({
            tsConfigFilePath: './upgradeTestProjects/v2-v3-stringParameters/tsconfig.json',
        });


        const code = `
            import { ITypedQueryBuilder } from './typedKnexTypes';

            const a = {} as ITypedQueryBuilder<{}, {}, {}>;
            a.select(i => i.name);
        `;


        const sourceFile = project.createSourceFile('./upgradeTestProjects/v2-v3-stringParameters/src/test.ts', code);

        assert.equal(project.getPreEmitDiagnostics().length, 0);

        upgradeProjectStringParameters(project);

        assert.equal(sourceFile.getText(), `import { ITypedQueryBuilder } from './typedKnexTypes';

            const a = {} as ITypedQueryBuilder<{}, {}, {}>;
            a.select('name');
        `)


    });


    it('should upgrade select column array', async () => {


        const project = new Project({
            tsConfigFilePath: './upgradeTestProjects/v2-v3-stringParameters/tsconfig.json',
        });


        const code = `
            import { ITypedQueryBuilder } from './typedKnexTypes';

            const a = {} as ITypedQueryBuilder<{}, {}, {}>;
            a.select(i => [i.name, i.other.id]);
        `;


        const sourceFile = project.createSourceFile('./upgradeTestProjects/v2-v3-stringParameters/src/test.ts', code);

        assert.equal(project.getPreEmitDiagnostics().length, 0);

        upgradeProjectStringParameters(project);

        assert.equal(sourceFile.getText(), `import { ITypedQueryBuilder } from './typedKnexTypes';

            const a = {} as ITypedQueryBuilder<{}, {}, {}>;
            a.select('name','other.id');
        `)


    });

    it('should upgrade both column parameters', async () => {


        const project = new Project({
            tsConfigFilePath: './upgradeTestProjects/v2-v3-stringParameters/tsconfig.json',
        });


        const code = `
            import { ITypedQueryBuilder } from './typedKnexTypes';

            const a = {} as ITypedQueryBuilder<{}, {}, {}>;
            a.joinOn(i => i.name, 'op', i => i.other.id);
        `;


        const sourceFile = project.createSourceFile('./upgradeTestProjects/v2-v3-stringParameters/src/test.ts', code);

        assert.equal(project.getPreEmitDiagnostics().length, 0);

        upgradeProjectStringParameters(project);

        assert.equal(sourceFile.getText(), `import { ITypedQueryBuilder } from './typedKnexTypes';

            const a = {} as ITypedQueryBuilder<{}, {}, {}>;
            a.joinOn('name', 'op', 'other.id');
        `)


    });


    it('should upgrade whereColumn', async () => {


        const project = new Project({
            tsConfigFilePath: './upgradeTestProjects/v2-v3-stringParameters/tsconfig.json',
        });


        const code = `
            import { ITypedQueryBuilder } from './typedKnexTypes';

            const a = {} as ITypedQueryBuilder<{}, {}, {}>;
            const parent = {id:1};
            a.whereColumn(i => i.name, '=', parent.id);
        `;


        const sourceFile = project.createSourceFile('./upgradeTestProjects/v2-v3-stringParameters/src/test.ts', code);

        assert.equal(project.getPreEmitDiagnostics().length, 0);

        upgradeProjectStringParameters(project);

        assert.equal(sourceFile.getText(), `import { ITypedQueryBuilder } from './typedKnexTypes';

            const a = {} as ITypedQueryBuilder<{}, {}, {}>;
            const parent = {id:1};
            a.whereColumn('name', '=', 'id');
        `)


    });

    it('should upgrade whereExists', async () => {
        const project = new Project({
            tsConfigFilePath: './upgradeTestProjects/v2-v3-stringParameters/tsconfig.json',
        });

        const code = `
            import { ITypedQueryBuilder } from './typedKnexTypes';

            class TableClass {}
            const a = {} as ITypedQueryBuilder<{}, {}, {}>;
            a.whereExists(TableClass, (subQuery, _parent) => subQuery.select(i => i.id));
        `;

        const sourceFile = project.createSourceFile('./upgradeTestProjects/v2-v3-stringParameters/src/test.ts', code);

        assert.equal(project.getPreEmitDiagnostics().length, 0);

        upgradeProjectStringParameters(project);

        assert.equal(sourceFile.getText(), `import { ITypedQueryBuilder } from './typedKnexTypes';

            class TableClass {}
            const a = {} as ITypedQueryBuilder<{}, {}, {}>;
            a.whereExists(TableClass, (subQuery) => subQuery.select('id'));
        `)
    });
});