mirror of https://github.com/AMT-Cheif/drift.git
51 lines
1.6 KiB
Dart
51 lines
1.6 KiB
Dart
import 'package:test/test.dart';
|
|
import 'package:sqlparser/sqlparser.dart';
|
|
|
|
void main() {
|
|
test('resolves simple functions', () {
|
|
final context = SqlEngine().analyze('SELECT ABS(-3)');
|
|
|
|
final select = context.root as SelectStatement;
|
|
final column = select.columns.single as ExpressionResultColumn;
|
|
|
|
expect((column.expression as FunctionExpression).resolved, abs);
|
|
});
|
|
|
|
test('correctly resolves return columns', () {
|
|
final id = TableColumn('id', const ResolvedType(type: BasicType.int));
|
|
final content =
|
|
TableColumn('content', const ResolvedType(type: BasicType.text));
|
|
|
|
final demoTable = Table(
|
|
name: 'demo',
|
|
resolvedColumns: [id, content],
|
|
);
|
|
final engine = SqlEngine()..registerTable(demoTable);
|
|
|
|
final context =
|
|
engine.analyze('SELECT id, d.content, *, 3 + 4 FROM demo AS d');
|
|
|
|
final select = context.root as SelectStatement;
|
|
final resolvedColumns = select.resolvedColumns;
|
|
|
|
expect(resolvedColumns.map((c) => c.name),
|
|
['id', 'content', 'id', 'content', '3 + 4']);
|
|
|
|
expect(resolvedColumns.map((c) => context.typeOf(c).type.type), [
|
|
BasicType.int,
|
|
BasicType.text,
|
|
BasicType.int,
|
|
BasicType.text,
|
|
BasicType.int,
|
|
]);
|
|
|
|
final firstColumn = select.columns[0] as ExpressionResultColumn;
|
|
final secondColumn = select.columns[1] as ExpressionResultColumn;
|
|
final from = select.from[0] as TableReference;
|
|
|
|
expect((firstColumn.expression as Reference).resolved, id);
|
|
expect((secondColumn.expression as Reference).resolved, content);
|
|
expect(from.resolved, demoTable);
|
|
});
|
|
}
|