Add method override annotation (#402)

aws-amplify · Jun 13, 2022 · 44a260f · 44a260f
1 parent 8d72e4d
commit 44a260f
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 3 deletions.
diff --git a/...nc-modelgen-plugin/src/__tests__/visitors/__snapshots__/appsync-dart-visitor.test.ts.snap b/...nc-modelgen-plugin/src/__tests__/visitors/__snapshots__/appsync-dart-visitor.test.ts.snap
@@ -8149,15 +8149,18 @@ class ModelWithImplicitIDModelIdentifier implements ModelIdentifier<ModelWithImp
  const ModelWithImplicitIDModelIdentifier({
  required this.id});
 
+ @override
  Map<String, dynamic> serializeAsMap() => (<String, dynamic>{
  'id': id
  });
 
+ @override
  List<Map<String, dynamic>> serializeAsList() => serializeAsMap()
  .entries
  .map((entry) => (<String, dynamic>{ entry.key: entry.value }))
  .toList();
 
+ @override
  String serializeAsString() => serializeAsMap().values.join('#');
 
  @override
@@ -8352,15 +8355,18 @@ class ModelWithExplicitIDModelIdentifier implements ModelIdentifier<ModelWithExp
  const ModelWithExplicitIDModelIdentifier({
  required this.id});
 
+ @override
  Map<String, dynamic> serializeAsMap() => (<String, dynamic>{
  'id': id
  });
 
+ @override
  List<Map<String, dynamic>> serializeAsList() => serializeAsMap()
  .entries
  .map((entry) => (<String, dynamic>{ entry.key: entry.value }))
  .toList();
 
+ @override
  String serializeAsString() => serializeAsMap().values.join('#');
 
  @override
@@ -8550,15 +8556,18 @@ class ModelWithExplicitIDAndSDIModelIdentifier implements ModelIdentifier<ModelW
  const ModelWithExplicitIDAndSDIModelIdentifier({
  required this.id});
 
+ @override
  Map<String, dynamic> serializeAsMap() => (<String, dynamic>{
  'id': id
  });
 
+ @override
  List<Map<String, dynamic>> serializeAsList() => serializeAsMap()
  .entries
  .map((entry) => (<String, dynamic>{ entry.key: entry.value }))
  .toList();
 
+ @override
  String serializeAsString() => serializeAsMap().values.join('#');
 
  @override
@@ -8801,17 +8810,20 @@ class ModelWithIDPlusSortKeysModelIdentifier implements ModelIdentifier<ModelWit
  required this.title,
  required this.rating});
 
+ @override
  Map<String, dynamic> serializeAsMap() => (<String, dynamic>{
  'id': id,
  'title': title,
  'rating': rating
  });
 
+ @override
  List<Map<String, dynamic>> serializeAsList() => serializeAsMap()
  .entries
  .map((entry) => (<String, dynamic>{ entry.key: entry.value }))
  .toList();
 
+ @override
  String serializeAsString() => serializeAsMap().values.join('#');
 
  @override
@@ -9040,15 +9052,18 @@ class ModelWithExplicitlyDefinedPKModelIdentifier implements ModelIdentifier<Mod
  const ModelWithExplicitlyDefinedPKModelIdentifier({
  required this.modelID});
 
+ @override
  Map<String, dynamic> serializeAsMap() => (<String, dynamic>{
  'modelID': modelID
  });
 
+ @override
  List<Map<String, dynamic>> serializeAsList() => serializeAsMap()
  .entries
  .map((entry) => (<String, dynamic>{ entry.key: entry.value }))
  .toList();
 
+ @override
  String serializeAsString() => serializeAsMap().values.join('#');
 
  @override
@@ -9308,17 +9323,20 @@ class ModelWithExplicitlyDefinedPKPlusSortKeysAsCompositeKeyModelIdentifier impl
  required this.title,
  required this.rating});
 
+ @override
  Map<String, dynamic> serializeAsMap() => (<String, dynamic>{
  'modelID': modelID,
  'title': title,
  'rating': rating
  });
 
+ @override
  List<Map<String, dynamic>> serializeAsList() => serializeAsMap()
  .entries
  .map((entry) => (<String, dynamic>{ entry.key: entry.value }))
  .toList();
 
+ @override
  String serializeAsString() => serializeAsMap().values.join('#');
 
  @override

diff --git a/packages/appsync-modelgen-plugin/src/visitors/appsync-dart-visitor.ts b/packages/appsync-modelgen-plugin/src/visitors/appsync-dart-visitor.ts
@@ -403,6 +403,7 @@ export class AppSyncModelDartVisitor<
  '\n',
  ),
  { isBlock: false },
+ ['override']
  );
 
  classDeclarationBlock.addClassMethod(
@@ -416,11 +417,19 @@ export class AppSyncModelDartVisitor<
  indent('.toList();'),
  ].join('\n'),
  { isBlock: false },
+ ['override'],
  );
 
- classDeclarationBlock.addClassMethod('serializeAsString', 'String', undefined, " => serializeAsMap().values.join('#');", {
- isBlock: false,
- });
+ classDeclarationBlock.addClassMethod(
+ 'serializeAsString',
+ 'String',
+ undefined,
+ " => serializeAsMap().values.join('#');",
+ {
+ isBlock: false,
+ },
+ ['override'],
+ );
 
  classDeclarationBlock.addClassMethod(
  'toString',