drift/sqlparser/test/analysis/reference_resolver_test.dart

36 lines
1.1 KiB
Dart

import 'package:test/test.dart';
import 'package:sqlparser/sqlparser.dart';
import 'data.dart';
void main() {
test('correctly resolves return columns', () {
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);
});
}