mirror of https://github.com/AMT-Cheif/drift.git
57 lines
1.6 KiB
Dart
57 lines
1.6 KiB
Dart
import 'package:source_span/source_span.dart';
|
|
import 'package:sqlparser/sqlparser.dart';
|
|
import 'package:sqlparser/src/reader/tokenizer/token.dart';
|
|
import 'package:sqlparser/src/utils/ast_equality.dart';
|
|
import 'package:test/test.dart';
|
|
|
|
export 'package:sqlparser/src/reader/tokenizer/token.dart';
|
|
|
|
Token token(TokenType type) {
|
|
return Token(type, null);
|
|
}
|
|
|
|
InlineDartToken inlineDart(String dartCode) {
|
|
return InlineDartToken(fakeSpan('`$dartCode`'));
|
|
}
|
|
|
|
IdentifierToken identifier(String content) {
|
|
return IdentifierToken(false, fakeSpan(content));
|
|
}
|
|
|
|
void testMoorFile(String moorFile, MoorFile expected) {
|
|
final parsed =
|
|
SqlEngine(useMoorExtensions: true).parseMoorFile(moorFile).rootNode;
|
|
|
|
enforceHasSpan(parsed);
|
|
enforceEqual(parsed, expected);
|
|
}
|
|
|
|
void testStatement(String sql, AstNode expected, {bool moorMode = false}) {
|
|
final parsed = SqlEngine(useMoorExtensions: moorMode).parse(sql).rootNode;
|
|
enforceHasSpan(parsed);
|
|
enforceEqual(parsed, expected);
|
|
}
|
|
|
|
FileSpan fakeSpan(String content) {
|
|
return SourceFile.fromString(content).span(0);
|
|
}
|
|
|
|
void testAll(Map<String, AstNode> testCases) {
|
|
testCases.forEach((sql, expected) {
|
|
test('with $sql', () {
|
|
testStatement(sql, expected);
|
|
});
|
|
});
|
|
}
|
|
|
|
/// The parser should make sure [AstNode.hasSpan] is true on relevant nodes.
|
|
void enforceHasSpan(AstNode node) {
|
|
final problematic = [node].followedBy(node.allDescendants).firstWhere(
|
|
(node) => !node.hasSpan && !node.synthetic,
|
|
orElse: () => null);
|
|
|
|
if (problematic != null) {
|
|
throw ArgumentError('Node $problematic did not have a span');
|
|
}
|
|
}
|