From ccae3b4ee64af36bbe8b350b05adbd7650219888 Mon Sep 17 00:00:00 2001
From: Simon Binder <oss@simonbinder.eu>
Date: Tue, 23 Apr 2024 15:04:22 +0200
Subject: [PATCH] Expand allowlist for assets in test

---
 .../test/backends/build/build_integration_test.dart      | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/drift_dev/test/backends/build/build_integration_test.dart b/drift_dev/test/backends/build/build_integration_test.dart
index e1828132..9027541f 100644
--- a/drift_dev/test/backends/build/build_integration_test.dart
+++ b/drift_dev/test/backends/build/build_integration_test.dart
@@ -578,13 +578,15 @@ class MyDatabase {
     };
     final outputs = await emulateDriftBuild(inputs: inputs);
     final readAssets = outputs.readAssetsByBuilder;
+    // Allow reading SDK or other package assets to set up the analyzer.
+    final isFromExternalPackage =
+        isA<AssetId>().having((e) => e.package, 'package', isNot('a'));
 
     Matcher onlyReadsJsonsAnd(dynamic other) {
       return everyElement(
         anyOf(
           isA<AssetId>().having((e) => e.extension, 'extension', '.json'),
-          // Allow reading SDK or other package assets to set up the analyzer.
-          isA<AssetId>().having((e) => e.package, 'package', isNot('a')),
+          isFromExternalPackage,
           other,
         ),
       );
@@ -608,7 +610,8 @@ class MyDatabase {
     // However, the discover builder should not read other drift files.
     for (final input in inputs.keys) {
       if (input.endsWith('.drift')) {
-        expectReadsForBuilder(input, DriftDiscover, [makeAssetId(input)]);
+        expectReadsForBuilder(input, DriftDiscover,
+            everyElement(anyOf(makeAssetId(input), isFromExternalPackage)));
       } else {
         expectReadsForBuilder(
           input,