drift/drift_dev/test/model/specified_table_test.dart

20 lines
559 B
Dart

import 'package:drift_dev/moor_generator.dart';
import 'package:drift_dev/src/analyzer/options.dart';
import 'package:test/test.dart';
void main() {
test('companion names', () {
final table = MoorTable(
sqlName: 'user', overriddenName: 'GoogleUser', dartTypeName: 'User');
expect(table.getNameForCompanionClass(const MoorOptions.defaults()),
'GoogleUserCompanion');
expect(
table.getNameForCompanionClass(
const MoorOptions.defaults(useDataClassNameForCompanions: true)),
'UserCompanion',
);
});
}