2020-09-05 15:33:51 +02:00
import ModelQuery , { SelectFieldValue , WhereOperator } from "../src/db/ModelQuery" ;
2020-09-04 22:15:55 +02:00
import ModelFactory from "../src/db/ModelFactory" ;
import Model from "../src/db/Model" ;
describe ( 'Test ModelQuery' , ( ) = > {
2020-09-06 15:38:54 +02:00
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' ] ) ;
} ) ;
2020-09-05 16:09:30 +02:00
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' ) ;
2020-09-06 12:24:58 +02:00
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' ) ;
2020-09-05 16:09:30 +02:00
} ) ;
2020-09-05 15:33:51 +02:00
test ( 'update' , ( ) = > {
const query = ModelQuery . update ( { table : 'model' } as unknown as ModelFactory < Model > , {
'f1' : 'v1' ,
'f2' : 'v2' ,
'f3' : 'v3' ,
} ) . where ( 'f4' , 'v4' ) ;
2020-09-05 16:09:30 +02:00
expect ( query . toString ( true ) ) . toBe ( 'UPDATE `model` SET `model`.`f1`=?,`model`.`f2`=?,`model`.`f3`=? WHERE `f4`=?' ) ;
2020-09-07 14:38:30 +02:00
expect ( query . variables ) . toStrictEqual ( [ 'v1' , 'v2' , 'v3' , 'v4' ] ) ;
2020-09-05 15:33:51 +02:00
} ) ;
test ( 'function select' , ( ) = > {
const query = ModelQuery . select ( { table : 'model' } as unknown as ModelFactory < Model > , 'f1' , new SelectFieldValue ( '_count' , 'COUNT(*)' , true ) ) ;
2020-09-05 16:09:30 +02:00
expect ( query . toString ( true ) ) . toBe ( 'SELECT `model`.`f1`,(COUNT(*)) AS `_count` FROM `model`' ) ;
2020-09-05 15:33:51 +02:00
expect ( query . variables ) . toStrictEqual ( [ ] ) ;
} ) ;
test ( 'pivot' , ( ) = > {
const query = ModelQuery . select ( { table : 'model' } as unknown as ModelFactory < Model > , 'f1' ) ;
query . pivot ( 'pivot.f2' , 'f3' ) ;
2020-09-05 16:09:30 +02:00
expect ( query . toString ( true ) ) . toBe ( 'SELECT `model`.`f1`,`pivot`.`f2`,`model`.`f3` FROM `model`' ) ;
2020-09-05 15:33:51 +02:00
expect ( query . variables ) . toStrictEqual ( [ ] ) ;
} ) ;
2020-09-04 22:15:55 +02:00
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' ) ;
2020-09-05 16:09:30 +02:00
expect ( query . toString ( true ) ) . toBe ( 'SELECT `model`.* FROM `model` WHERE `f1`=? AND (`f2`=? AND `f3`=? OR (`f4`=?)) AND `f5`=?' ) ;
2020-09-07 14:38:30 +02:00
expect ( query . variables ) . toStrictEqual ( [ 'v1' , 'v2' , 'v3' , 'v4' , 'v5' ] ) ;
2020-09-04 22:15:55 +02:00
} ) ;
2020-09-05 14:55:19 +02:00
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' ) ;
2020-09-08 18:12:39 +02:00
query . recursive ( { localKey : 'local' , foreignKey : 'foreign' , idKey : 'foreign' } ) ;
2020-09-05 14:55:19 +02:00
query . sortBy ( 'f2' , 'ASC' ) . limit ( 8 ) ;
2020-09-08 18:12:39 +02:00
expect ( query . toString ( true ) ) . toBe ( "WITH RECURSIVE cte AS (SELECT `model`.*,1 AS __depth, ARRAY[`foreign`] AS __path FROM `model` WHERE `f1`=? UNION SELECT o.*, c.__depth + 1,c.__path || o.`foreign` 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 `f2` ASC LIMIT 8" ) ;
2020-09-05 15:33:51 +02:00
expect ( query . variables ) . toStrictEqual ( [ 'v1' ] ) ;
2020-09-05 14:55:19 +02:00
} ) ;
2020-09-06 15:06:40 +02:00
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 > , '*' ) ;
2020-09-06 16:02:53 +02:00
query2 . where ( 'f2' , 'v2' ) ;
2020-09-06 15:06:40 +02:00
query . union ( query2 , 'model.f1' , 'DESC' , false , 8 ) ;
2020-09-06 16:02:53 +02:00
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' ] )
2020-09-06 15:06:40 +02:00
} ) ;
2020-09-04 22:15:55 +02:00
} ) ;