From 3cff5478cbc1b02e4f3ecdf1d996599e48a477fd Mon Sep 17 00:00:00 2001
From: Kevin Moore <kevmoo@google.com>
Date: Tue, 14 Jan 2020 13:30:38 -0800
Subject: [PATCH] Follow-up fix for dart-lang/build#2596

---
 pubspec.yaml                       |  2 +-
 test/init_library_reader_test.dart | 12 +++++++-----
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/pubspec.yaml b/pubspec.yaml
index e925913..90fccad 100644
--- a/pubspec.yaml
+++ b/pubspec.yaml
@@ -12,7 +12,7 @@ environment:
 dependencies:
   analyzer: '>=0.33.3 <0.40.0'
   build: '>=0.12.6 <2.0.0'
-  build_test: ^0.10.0
+  build_test: ^0.10.12
   dart_style: ^1.2.0
   meta: ^1.0.2
   path: ^1.3.2
diff --git a/test/init_library_reader_test.dart b/test/init_library_reader_test.dart
index 2cb2f9f..c7dbade 100644
--- a/test/init_library_reader_test.dart
+++ b/test/init_library_reader_test.dart
@@ -1,5 +1,6 @@
 import 'dart:io';
 
+import 'package:build/build.dart';
 import 'package:source_gen_test/src/init_library_reader.dart';
 import 'package:test/test.dart';
 
@@ -46,11 +47,12 @@ void main() {
     test('part instead', () async {
       await expectLater(
         () => initializeLibraryReaderForDirectory('test/src', 'test_part.dart'),
-        throwsA(isArgumentError
-            .having((ae) => ae.message, 'message',
-                'Does not seem to reference a Dart library.')
-            .having((ae) => ae.name, 'name', 'targetLibraryFileName')),
+        throwsA(isA<NonLibraryAssetException>().having(
+          (ae) => ae.assetId.toString(),
+          'assetId.toString()',
+          '__test__|lib/test_part.dart',
+        )),
       );
-    }, skip: 'See dart-lang/build#2596');
+    });
   });
 }