Elements. Migrate FlutterWrapStreamBuilder.

Change-Id: I8109a541efdadaa14372101ecf9f5118ff3e0b0a
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/389580
Reviewed-by: Phil Quitslund <[email protected]>
Commit-Queue: Konstantin Shcheglov <[email protected]>

Author: scheglov

pkg/analysis_server/analyzer_use_new_elements.txt

@@ -326,6 +326,7 @@ lib/src/services/correction/dart/flutter_swap_with_child.dart
 lib/src/services/correction/dart/flutter_swap_with_parent.dart
 lib/src/services/correction/dart/flutter_wrap_builder.dart
 lib/src/services/correction/dart/flutter_wrap_generic.dart
+lib/src/services/correction/dart/flutter_wrap_stream_builder.dart
 lib/src/services/correction/dart/ignore_diagnostic.dart
 lib/src/services/correction/dart/import_add_show.dart
 lib/src/services/correction/dart/inline_invocation.dart

pkg/analysis_server/lib/src/services/correction/dart/flutter_wrap_stream_builder.dart

@@ -33,14 +33,14 @@ class FlutterWrapStreamBuilder extends ResolvedCorrectionProducer {
     var widgetSrc = utils.getNodeText(widgetExpr);
 
     var streamBuilderElement =
-        await sessionHelper.getFlutterClass('StreamBuilder');
+        await sessionHelper.getFlutterClass2('StreamBuilder');
     if (streamBuilderElement == null) {
       return;
     }
 
     await builder.addDartFileEdit(file, (builder) {
       builder.addReplacement(range.node(widgetExpr), (builder) {
-        builder.writeReference(streamBuilderElement);
+        builder.writeReference2(streamBuilderElement);
 
         builder.write('<');
         builder.addSimpleLinkedEdit('type', 'Object');