91 lines
4.8 KiB
TypeScript
91 lines
4.8 KiB
TypeScript
import ModelQuery, {SelectFieldValue, WhereOperator} from "../src/db/ModelQuery";
|
|
import ModelFactory from "../src/db/ModelFactory";
|
|
import Model from "../src/db/Model";
|
|
|
|
describe('Test ModelQuery', () => {
|
|
test('select', () => {
|
|
const query = ModelQuery.select({table: 'model'} as unknown as ModelFactory<Model>, 'f1', '"Test" as f2')
|
|
.where('f4', 'v4');
|
|
expect(query.toString(true)).toBe('SELECT `model`.`f1`,"Test" as f2 FROM `model` WHERE `f4`=?');
|
|
expect(query.variables).toStrictEqual(['v4']);
|
|
});
|
|
|
|
test('order by', () => {
|
|
const query = ModelQuery.select({table: 'model'} as unknown as ModelFactory<Model>)
|
|
.sortBy('model.f2', 'ASC');
|
|
expect(query.toString(true)).toBe('SELECT `model`.* FROM `model` ORDER BY `model`.`f2` ASC');
|
|
|
|
const queryRaw = ModelQuery.select({table: 'model'} as unknown as ModelFactory<Model>)
|
|
.sortBy('coalesce(model.f1, model.f2)', 'ASC', true);
|
|
expect(queryRaw.toString(true)).toBe('SELECT `model`.* FROM `model` ORDER BY coalesce(model.f1, model.f2) ASC');
|
|
});
|
|
|
|
test('update', () => {
|
|
const query = ModelQuery.update({table: 'model'} as unknown as ModelFactory<Model>, {
|
|
'f1': 'v1',
|
|
'f2': 'v2',
|
|
'f3': 'v3',
|
|
}).where('f4', 'v4');
|
|
expect(query.toString(true)).toBe('UPDATE `model` SET `model`.`f1`=?,`model`.`f2`=?,`model`.`f3`=? WHERE `f4`=?');
|
|
expect(query.variables).toStrictEqual(['v1', 'v2', 'v3', 'v4']);
|
|
});
|
|
|
|
test('function select', () => {
|
|
const query = ModelQuery.select(
|
|
{table: 'model'} as unknown as ModelFactory<Model>,
|
|
'f1',
|
|
new SelectFieldValue('_count', 'COUNT(*)', true),
|
|
);
|
|
expect(query.toString(true)).toBe('SELECT `model`.`f1`,(COUNT(*)) AS `_count` FROM `model`');
|
|
expect(query.variables).toStrictEqual([]);
|
|
});
|
|
|
|
test('pivot', () => {
|
|
const query = ModelQuery.select({table: 'model'} as unknown as ModelFactory<Model>, 'f1');
|
|
query.pivot('pivot.f2', 'f3');
|
|
|
|
expect(query.toString(true)).toBe('SELECT `model`.`f1`,`pivot`.`f2`,`model`.`f3` FROM `model`');
|
|
expect(query.variables).toStrictEqual([]);
|
|
});
|
|
|
|
test('groupWhere generates proper query', () => {
|
|
const query = ModelQuery.select({table: 'model'} as unknown as ModelFactory<Model>, '*');
|
|
query.where('f1', 'v1');
|
|
query.groupWhere(q => q.where('f2', 'v2').where('f3', 'v3')
|
|
.groupWhere(q => q.where('f4', 'v4'), WhereOperator.OR))
|
|
.where('f5', 'v5');
|
|
|
|
expect(query.toString(true)).toBe('SELECT `model`.* FROM `model` WHERE `f1`=? AND (`f2`=? AND `f3`=? OR (`f4`=?)) AND `f5`=?');
|
|
expect(query.variables).toStrictEqual(['v1', 'v2', 'v3', 'v4', 'v5']);
|
|
});
|
|
|
|
test('recursive queries', () => {
|
|
const query = ModelQuery.select({table: 'model'} as unknown as ModelFactory<Model>, '*');
|
|
query.where('f1', 'v1');
|
|
query.leftJoin('test').on('model.j1', 'test.j2');
|
|
query.recursive({localKey: 'local', foreignKey: 'foreign'}, false);
|
|
query.limit(8);
|
|
|
|
expect(query.toString(true)).toBe("WITH RECURSIVE cte AS (SELECT `model`.*,1 AS __depth, CONCAT(`local`) AS __path FROM `model` WHERE `f1`=? UNION SELECT o.*,c.__depth + 1,CONCAT(c.__path,'/',o.`local`) AS __path FROM `model` AS o, cte AS c WHERE o.`foreign`=c.`local`) SELECT * FROM cte LEFT JOIN `test` ON `model`.`j1`=`test`.`j2` ORDER BY __path ASC LIMIT 8");
|
|
expect(query.variables).toStrictEqual(['v1']);
|
|
|
|
const reversedQuery = ModelQuery.select({table: 'model'} as unknown as ModelFactory<Model>, '*');
|
|
reversedQuery.where('f1', 'v1');
|
|
reversedQuery.leftJoin('test').on('model.j1', 'test.j2');
|
|
reversedQuery.recursive({localKey: 'local', foreignKey: 'foreign'}, true);
|
|
|
|
expect(reversedQuery.toString(true)).toBe("WITH RECURSIVE cte AS (SELECT `model`.*,1 AS __depth, CONCAT(`foreign`) AS __path FROM `model` WHERE `f1`=? UNION SELECT o.*,c.__depth + 1,CONCAT(c.__path,'/',o.`foreign`) AS __path FROM `model` AS o, cte AS c WHERE o.`foreign`=c.`local`) SELECT * FROM cte LEFT JOIN `test` ON `model`.`j1`=`test`.`j2` ORDER BY __path DESC");
|
|
expect(reversedQuery.variables).toStrictEqual(['v1']);
|
|
});
|
|
|
|
test('union queries', () => {
|
|
const query = ModelQuery.select({table: 'model'} as unknown as ModelFactory<Model>, '*');
|
|
const query2 = ModelQuery.select({table: 'model2'} as unknown as ModelFactory<Model>, '*');
|
|
query2.where('f2', 'v2');
|
|
query.union(query2, 'model.f1', 'DESC', false, 8);
|
|
|
|
expect(query.toString(true)).toBe("(SELECT `model`.* FROM `model`) UNION (SELECT `model2`.* FROM `model2` WHERE `f2`=?) ORDER BY `model`.`f1` DESC LIMIT 8");
|
|
expect(query.variables).toStrictEqual(['v2']);
|
|
});
|
|
});
|