drift/drift_dev/test/model/specified_table_test.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',
    );
  });
}