test: Add a test with a custom companion name and a default data name

This commit is contained in:
ValentinVignal 2024-03-07 09:52:57 +08:00
parent d0ff1f29d6
commit 6b1f54d647
No known key found for this signature in database
GPG Key ID: 040FFDADFB7EF05A
1 changed files with 13 additions and 1 deletions

View File

@ -290,8 +290,14 @@ class SecondTable extends Table {
IntColumn get bar => integer()();
}
@DataClassName.custom(companion: 'ThirdCompanionClass')
class ThirdTable extends Table {
TextColumn get foo => text()();
IntColumn get bar => integer()();
}
@DriftDatabase(
tables: [FirstTable, SecondTable],
tables: [FirstTable, SecondTable, ThirdTable],
)
class Database extends _$Database {}
'''
@ -312,6 +318,12 @@ class Database extends _$Database {}
contains(
'class SecondTableCompanion extends UpdateCompanion<SecondDataClass> {',
),
contains(
'class ThirdTableData extends DataClass implements Insertable<ThirdTableData> {',
),
contains(
'class ThirdTableCompanion extends UpdateCompanion<ThirdTableData> {',
),
])),
}, writer.dartOutputs, writer.writer);
},