feat: add support for custom output formatting

lib/src/generate_for_element.dart

@@ -19,8 +19,11 @@ final _testAnnotationWarnings = <String>{};
 Future<String> generateForElement<T>(
   GeneratorForAnnotation<T> generator,
   LibraryReader libraryReader,
-  String name,
-) async {
+  String name, {
+  String Function(String code)? formatOutput,
+}) async {
+  formatOutput ??= _formatter.format;
+
   final elements =
       libraryReader.allElements.where((e) => e.name == name).toList();
 
@@ -94,7 +97,7 @@ Future<String> generateForElement<T>(
 
   final generated = await generatedStream.join('\n\n');
 
-  return _formatter.format(generated);
+  return formatOutput(generated);
 }
 
 // ignore: subtype_of_sealed_class

lib/src/test_annotated_classes.dart

@@ -29,13 +29,15 @@ void testAnnotatedElements<T>(
   Map<String, GeneratorForAnnotation<T>>? additionalGenerators,
   Iterable<String>? expectedAnnotatedTests,
   Iterable<String>? defaultConfiguration,
+  String Function(String code)? formatOutput,
 }) {
   for (var entry in getAnnotatedClasses<T>(
     libraryReader,
     defaultGenerator,
     additionalGenerators: additionalGenerators,
     expectedAnnotatedTests: expectedAnnotatedTests,
     defaultConfiguration: defaultConfiguration,
+    formatOutput: formatOutput,
   )) {
     entry._registerTest();
   }
@@ -50,6 +52,7 @@ List<AnnotatedTest<T>> getAnnotatedClasses<T>(
   Map<String, GeneratorForAnnotation<T>>? additionalGenerators,
   Iterable<String>? expectedAnnotatedTests,
   Iterable<String>? defaultConfiguration,
+  String Function(String code)? formatOutput,
 }) {
   final generators = <String, GeneratorForAnnotation<T>>{
     _defaultConfigurationName: defaultGenerator,
@@ -174,6 +177,7 @@ List<AnnotatedTest<T>> getAnnotatedClasses<T>(
           configuration,
           entry.elementName,
           entry.expectation,
+          formatOutput: formatOutput,
         ),
       );
     }
@@ -205,6 +209,7 @@ class AnnotatedTest<T> {
   final LibraryReader _libraryReader;
   final TestExpectation expectation;
   final String _elementName;
+  final String Function(String code)? _formatOutput;
 
   String get _testName {
     var value = _elementName;
@@ -219,8 +224,9 @@ class AnnotatedTest<T> {
     this.generator,
     this.configuration,
     this._elementName,
-    this.expectation,
-  );
+    this.expectation, {
+    String Function(String code)? formatOutput,
+  }) : _formatOutput = formatOutput;
 
   void _registerTest() {
     if (expectation is ShouldGenerate) {
@@ -233,8 +239,12 @@ class AnnotatedTest<T> {
     throw StateError('Should never get here.');
   }
 
-  Future<String> _generate() =>
-      generateForElement<T>(generator, _libraryReader, _elementName);
+  Future<String> _generate() => generateForElement<T>(
+        generator,
+        _libraryReader,
+        _elementName,
+        formatOutput: _formatOutput,
+      );
 
   Future<void> _shouldGenerateTest() async {
     final output = await _generate();

test/generate_for_element_test.dart

@@ -98,6 +98,75 @@ const TestClass2NameLowerCase = testclass2;
 ''',
       );
     });
+
+    group(
+      'formatOutput',
+      () {
+        test(
+          'should format the generated source code with `DartFormatter` when `formatOutput` is `null`.',
+          () async {
+            final output = await generateForElement(
+              const TestGenerator(),
+              reader,
+              'TestClassThatHasAVeryLongNameThatShouldNotWrapWhenFormatOutputIsANop',
+              formatOutput: null,
+            );
+            printOnFailure(output);
+            expect(
+              output,
+              r'''
+const TestClassThatHasAVeryLongNameThatShouldNotWrapWhenFormatOutputIsANopNameLength =
+    68;
+
+const TestClassThatHasAVeryLongNameThatShouldNotWrapWhenFormatOutputIsANopNameLowerCase =
+    testclassthathasaverylongnamethatshouldnotwrapwhenformatoutputisanop;
+''',
+            );
+          },
+        );
+
+        test(
+          'should not format the generated source code when `formatOutput` is a NOP.',
+          () async {
+            final output = await generateForElement(
+              const TestGenerator(),
+              reader,
+              'TestClassThatHasAVeryLongNameThatShouldNotWrapWhenFormatOutputIsANop',
+              formatOutput: (code) => code,
+            );
+            printOnFailure(output);
+            expect(
+              output,
+              r'''
+const TestClassThatHasAVeryLongNameThatShouldNotWrapWhenFormatOutputIsANopNameLength = 68;
+
+const TestClassThatHasAVeryLongNameThatShouldNotWrapWhenFormatOutputIsANopNameLowerCase = testclassthathasaverylongnamethatshouldnotwrapwhenformatoutputisanop;''',
+            );
+          },
+        );
+
+        test(
+          'should format the generated source code with a custom formatter when `formatOutput` is a custom formatter.',
+          () async {
+            final output = await generateForElement(
+              const TestGenerator(),
+              reader,
+              'TestClassThatHasAVeryLongNameThatShouldNotWrapWhenFormatOutputIsANop',
+              formatOutput: (code) => '$code\n',
+            );
+            printOnFailure(output);
+            expect(
+              output,
+              r'''
+const TestClassThatHasAVeryLongNameThatShouldNotWrapWhenFormatOutputIsANopNameLength = 68;
+
+const TestClassThatHasAVeryLongNameThatShouldNotWrapWhenFormatOutputIsANopNameLowerCase = testclassthathasaverylongnamethatshouldnotwrapwhenformatoutputisanop;
+''',
+            );
+          },
+        );
+      },
+    );
   });
 
   test('throwsInvalidGenerationSourceError', () async {

test/init_library_reader_test.dart

@@ -32,6 +32,7 @@ void main() {
           'class TestClassWithBadMember',
           'int badTestFunc()',
           'int badTestField',
+          'class TestClassThatHasAVeryLongNameThatShouldNotWrapWhenFormatOutputIsANop',
           'class TestClass2',
           'import source /source_gen_test/lib/annotations.dart',
           'import source /__test__/lib/test_annotation.dart',

test/src/test_library.dart

@@ -78,3 +78,6 @@ int badTestFunc() => 42;
 )
 @TestAnnotation()
 const badTestField = 42;
+
+@TestAnnotation()
+class TestClassThatHasAVeryLongNameThatShouldNotWrapWhenFormatOutputIsANop {}