Combine InMemoryAssetReader and InMemoryAssetWriter into InMemoryAssetReaderWriter.

Update tests so that reader and writer are backed by the same `InMemoryAssetReaderWriter`.

Author: davidmorgan

_test_common/lib/common.dart

@@ -14,8 +14,7 @@ export 'package:build_test/build_test.dart'
 export 'assets.dart';
 export 'builders.dart';
 export 'descriptors.dart';
-export 'in_memory_reader.dart';
-export 'in_memory_writer.dart';
+export 'in_memory_reader_writer.dart';
 export 'matchers.dart';
 export 'package_graphs.dart';
 export 'sdk.dart';

_test_common/lib/in_memory_reader.dart

@@ -1,6 +1,7 @@
 // Copyright (c) 2016, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
+
 import 'dart:async';
 import 'dart:convert';
 
@@ -10,10 +11,20 @@ import 'package:build_test/build_test.dart';
 import 'package:path/path.dart' as p;
 import 'package:watcher/watcher.dart';
 
-class InMemoryRunnerAssetWriter extends InMemoryAssetWriter
-    implements RunnerAssetWriter {
+class InMemoryRunnerAssetReaderWriter extends InMemoryAssetReaderWriter
+    implements RunnerAssetReader, RunnerAssetWriter {
+  final _onCanReadController = StreamController<AssetId>.broadcast();
+  Stream<AssetId> get onCanRead => _onCanReadController.stream;
   void Function(AssetId)? onDelete;
 
+  InMemoryRunnerAssetReaderWriter({super.sourceAssets, super.rootPackage});
+
+  @override
+  Future<bool> canRead(AssetId id) {
+    _onCanReadController.add(id);
+    return super.canRead(id);
+  }
+
   @override
   Future writeAsBytes(AssetId id, List<int> bytes) async {
     var type = assets.containsKey(id) ? ChangeType.MODIFY : ChangeType.ADD;

_test_common/lib/in_memory_writer.dart renamed to _test_common/lib/in_memory_reader_writer.dart

@@ -4,29 +4,26 @@
 
 import 'dart:async';
 
-// ignore: implementation_imports
-import 'package:build_runner_core/src/asset/reader.dart';
-// ignore: implementation_imports
-import 'package:build_runner_core/src/asset/writer.dart';
-// ignore: implementation_imports
-import 'package:build_runner_core/src/environment/build_environment.dart';
+import 'package:build_runner_core/build_runner_core.dart';
 import 'package:logging/logging.dart';
 
 import 'common.dart';
+import 'in_memory_reader_writer.dart';
 
 /// A [BuildEnvironment] for testing.
 ///
-/// Defaults to using an [InMemoryRunnerAssetReader] and
-/// [InMemoryRunnerAssetWriter].
+/// Defaults to using an [InMemoryRunnerAssetReaderWriter].
 ///
 /// To handle prompts you must first set `nextPromptResponse`. Alternatively
 /// you can set `throwOnPrompt` to `true` to emulate a
 /// [NonInteractiveBuildException].
 class TestBuildEnvironment extends BuildEnvironment {
+  final InMemoryRunnerAssetReaderWriter _readerWriter;
+
   @override
-  final RunnerAssetReader reader;
+  RunnerAssetReader get reader => _readerWriter;
   @override
-  final RunnerAssetWriter writer;
+  RunnerAssetWriter get writer => _readerWriter;
 
   /// If true, this will throw a [NonInteractiveBuildException] for all calls to
   /// [prompt].
@@ -44,11 +41,9 @@ class TestBuildEnvironment extends BuildEnvironment {
   int? _nextPromptResponse;
 
   TestBuildEnvironment(
-      {RunnerAssetReader? reader,
-      RunnerAssetWriter? writer,
+      {InMemoryRunnerAssetReaderWriter? readerWriter,
       this.throwOnPrompt = false})
-      : reader = reader ?? InMemoryRunnerAssetReader(),
-        writer = writer ?? InMemoryRunnerAssetWriter();
+      : _readerWriter = readerWriter ?? InMemoryRunnerAssetReaderWriter();
 
   @override
   void onLog(LogRecord record) => logRecords.add(record);

_test_common/lib/test_environment.dart

@@ -11,8 +11,7 @@ import 'package:build_test/build_test.dart';
 import 'package:logging/logging.dart';
 import 'package:test/test.dart';
 
-import 'in_memory_reader.dart';
-import 'in_memory_writer.dart';
+import 'in_memory_reader_writer.dart';
 import 'package_graphs.dart';
 
 Future<void> wait(int milliseconds) =>
@@ -89,16 +88,13 @@ Future<BuildResult> testBuilders(
   void Function(AssetId id)? onDelete,
 }) async {
   packageGraph ??= buildPackageGraph({rootPackage('a'): []});
-  final writer =
-      resumeFrom == null ? InMemoryRunnerAssetWriter() : resumeFrom.writer;
-  writer.onDelete = onDelete;
-  final reader = resumeFrom == null
-      ? InMemoryRunnerAssetReader.shareAssetCache(writer.assets,
-          rootPackage: packageGraph.root.name)
-      : resumeFrom.reader;
+  final readerWriter = resumeFrom == null
+      ? InMemoryRunnerAssetReaderWriter(rootPackage: packageGraph.root.name)
+      : resumeFrom.readerWriter;
+  readerWriter.onDelete = onDelete;
   var pkgConfigId =
       AssetId(packageGraph.root.name, '.dart_tool/package_config.json');
-  if (!await reader.canRead(pkgConfigId)) {
+  if (!await readerWriter.canRead(pkgConfigId)) {
     var packageConfig = {
       'configVersion': 2,
       'packages': [
@@ -111,21 +107,21 @@ Future<BuildResult> testBuilders(
           },
       ],
     };
-    await writer.writeAsString(pkgConfigId, jsonEncode(packageConfig));
+    await readerWriter.writeAsString(pkgConfigId, jsonEncode(packageConfig));
   }
 
   inputs.forEach((serializedId, contents) {
     var id = makeAssetId(serializedId);
     if (contents is String) {
-      reader.cacheStringAsset(id, contents);
+      readerWriter.cacheStringAsset(id, contents);
     } else if (contents is List<int>) {
-      reader.cacheBytesAsset(id, contents);
+      readerWriter.cacheBytesAsset(id, contents);
     }
   });
 
   builderConfigOverrides ??= const {};
   var environment = OverrideableEnvironment(IOEnvironment(packageGraph),
-      reader: reader, writer: writer, onLog: onLog);
+      reader: readerWriter, writer: readerWriter, onLog: onLog);
   var logSubscription =
       LogSubscription(environment, verbose: verbose, logLevel: logLevel);
   var options = await BuildOptions.create(logSubscription,
@@ -152,14 +148,13 @@ Future<BuildResult> testBuilders(
   if (checkBuildStatus) {
     checkBuild(result,
         outputs: outputs,
-        writer: writer,
+        writer: readerWriter,
         status: status,
         rootPackage: packageGraph.root.name,
         expectedGeneratedDir: expectedGeneratedDir);
   }
 
-  result._attachReader(reader);
-  result._attachWriter(writer);
+  result._attachReaderWriter(readerWriter);
 
   return result;
 }
@@ -199,21 +194,13 @@ void checkBuild(BuildResult result,
 }
 
 extension BuildResultExtension on BuildResult {
-  /// The reader used in this `testBuilders` build.
-  InMemoryRunnerAssetReader get reader => _readerExpando[this]!;
+  /// The reader/writer used in this `testBuilders` build.
+  InMemoryRunnerAssetReaderWriter get readerWriter =>
+      _readerWriterExpando[this]!;
 
-  /// The writer used in this `testBuilders` build.
-  InMemoryRunnerAssetWriter get writer => _writerExpando[this]!;
-
-  void _attachReader(InMemoryRunnerAssetReader reader) {
-    _readerExpando[this] = reader;
-  }
-
-  void _attachWriter(InMemoryRunnerAssetWriter writer) {
-    _writerExpando[this] = writer;
+  void _attachReaderWriter(InMemoryRunnerAssetReaderWriter readerWriter) {
+    _readerWriterExpando[this] = readerWriter;
   }
 }
 
-final Expando<InMemoryRunnerAssetReader> _readerExpando = Expando();
-
-final Expando<InMemoryRunnerAssetWriter> _writerExpando = Expando();
+final Expando<InMemoryRunnerAssetReaderWriter> _readerWriterExpando = Expando();

_test_common/lib/test_phases.dart

@@ -2,6 +2,7 @@
 
 - Add `package:build/src/internal.dart` for use by `build_resolvers`,
   `build_runner_core` and `build_test`.
+- Use `build_test` 3.0.0.
 
 ## 2.4.2
 

build/CHANGELOG.md

@@ -20,7 +20,7 @@ dependencies:
 
 dev_dependencies:
   build_resolvers: ^2.4.0
-  build_test: ^2.0.0
+  build_test: ^3.0.0-wip
   dart_flutter_team_lints: ^3.1.0
   test: ^1.16.0
 

build/pubspec.yaml

@@ -78,8 +78,7 @@ void main() {
     late InMemoryAssetReader reader;
 
     setUp(() {
-      writer = InMemoryAssetWriter();
-      reader = InMemoryAssetReader.shareAssetCache(writer.assets);
+      writer = reader = InMemoryAssetReaderWriter();
     });
 
     test('tracks outputs created by a builder', () async {

build/test/builder/build_step_impl_test.dart

@@ -29,8 +29,7 @@ void main() {
     });
     builder = TestBuilder(
         extraWork: (buildStep, __) => buildStep.fetchResource(resource));
-    writer = InMemoryAssetWriter();
-    reader = InMemoryAssetReader.shareAssetCache(writer.assets);
+    writer = reader = InMemoryAssetReaderWriter();
     addAssets(inputs, writer);
   });
 

build/test/generate/run_builder_test.dart

@@ -1,3 +1,7 @@
+## 5.0.12-wip
+
+- Use `build_test` 3.0.0.
+
 ## 5.0.11
 
 - Support 3.8.0 pre-release sdks.

build_modules/CHANGELOG.md

@@ -1,5 +1,5 @@
 name: build_modules
-version: 5.0.11
+version: 5.0.12-wip
 description: >-
   Builders to analyze and split Dart code into individually compilable modules
   based on imports.
@@ -32,7 +32,7 @@ dev_dependencies:
     path: test/fixtures/b
   # Used inside tests
   build_runner: ^2.0.0
-  build_test: ^2.0.0
+  build_test: ^3.0.0-wip
   json_serializable: ^6.0.0
   test: ^1.16.0
 

build_modules/pubspec.yaml

@@ -17,15 +17,13 @@ Future<void> testBuilderAndCollectAssets(
     void Function(LogRecord log)? onLog,
     void Function(AssetId, Iterable<AssetId>)?
         reportUnusedAssetsForInput}) async {
-  var writer = InMemoryAssetWriter();
   onLog ??= (log) => printOnFailure('${log.level}: ${log.message}');
-  await testBuilder(builder, assets,
+  final result = await testBuilder(builder, assets,
       generateFor: generateFor,
       outputs: outputs,
       onLog: onLog,
-      reportUnusedAssetsForInput: reportUnusedAssetsForInput,
-      writer: writer);
-  writer.assets.forEach((id, value) {
+      reportUnusedAssetsForInput: reportUnusedAssetsForInput);
+  result.readerWriter.assets.forEach((id, value) {
     assets['${id.package}|${id.path}'] = value;
   });
 }

build_modules/test/util.dart

@@ -3,6 +3,7 @@
 - Start using `package:build/src/internal.dart`.
 - Switch `BuildAssetUriResolver` dependency crawl to an iterative
   algorithm, preventing stack overflows.
+- Use `build_test` 3.0.0.
 
 ## 2.4.4
 

build_resolvers/CHANGELOG.md

@@ -24,7 +24,7 @@ dependencies:
   yaml: ^3.0.0
 
 dev_dependencies:
-  build_test: ^2.0.0
+  build_test: ^3.0.0-wip
   dart_flutter_team_lints: ^3.1.0
   test: ^1.16.0