AMT-Cheif
/
drift
mirror of https://github.com/AMT-Cheif/drift.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix code generation for `enum` with generic

This commit is contained in:
Nikita Dauhashei 2023-08-26 20:18:03 +02:00
parent 5aaeddd9d3
commit 9575612840
No known key found for this signature in database
GPG Key ID: 3B93F636D935382D
3 changed files with 28 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
drift_dev/lib/src/analysis/resolver/shared/dart_types.dart
View File

@ -357,9 +357,9 @@ AppliedTypeConverter readEnumConverter(
}
builder
..addText('<')
..addDartType(dartEnumType)
..addTopLevelElement(dartEnumType.element!)
..addText('>(')
..addDartType(dartEnumType)
..addTopLevelElement(dartEnumType.element!)
..addText('.values)');
});

5
drift_dev/test/analysis/resolver/dart/enum_columns_test.dart
View File

@ -17,10 +17,15 @@ void main() {
apple, orange, banana
}
enum FruitsWithGeneric<T> {
apple, orange, banana
}
class NotAnEnum {}
class ValidUsage extends Table {
IntColumn get intFruit => intEnum<Fruits>()();
IntColumn get intFruitsWithGeneric => intEnum<FruitsWithGeneric>()();
TextColumn get textFruit => textEnum<Fruits>()();
}

21
drift_dev/test/analysis/resolver/drift/table_test.dart
View File

@ -89,6 +89,7 @@ CREATE TABLE b (
CREATE TABLE foo (
fruitIndex ENUM(Fruits) NOT NULL,
fruitWithGenericIndex ENUM(FruitsWithGeneric) NOT NULL,
fruitName ENUMNAME(Fruits) NOT NULL,
anotherIndex ENUM(DoesNotExist) NOT NULL,
anotherName ENUMNAME(DoesNotExist) NOT NULL
@ -98,6 +99,10 @@ CREATE TABLE b (
enum Fruits {
apple, orange, banana
}
enum FruitsWithGeneric<T> {
apple, orange, banana
}
''',
});
@ -118,6 +123,22 @@ CREATE TABLE b (
.having((e) => e.dartType.getDisplayString(withNullability: true),
'dartType', 'Fruits'),
);
final withGenericIndexColumn = table.columns
.singleWhere((c) => c.nameInSql == 'fruitWithGenericIndex');
expect(withGenericIndexColumn.sqlType, DriftSqlType.int);
expect(
withGenericIndexColumn.typeConverter,
isA<AppliedTypeConverter>()
.having(
(e) => e.expression.toString(),
'expression',
contains('EnumIndexConverter<FruitsWithGeneric>'),
)
.having((e) => e.dartType.getDisplayString(withNullability: true),
'dartType', 'FruitsWithGeneric'),
);
final nameColumn =
table.columns.singleWhere((c) => c.nameInSql == 'fruitName');