Don't trigger provider scoped lints in tests

rrousselGit · Oct 29, 2023 · c3fc922 · c3fc922
1 parent e23f769
commit c3fc922
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/packages/riverpod_lint/lib/src/lints/scoped_providers_should_specify_dependencies.dart b/packages/riverpod_lint/lib/src/lints/scoped_providers_should_specify_dependencies.dart
@@ -17,6 +17,18 @@ extension SimpleIdentifierX on SimpleIdentifier {
     return libraryUri.scheme == 'package' &&
         libraryUri.pathSegments.first == 'flutter';
   }
+
+  bool get isPumpWidget {
+    if (name != 'pumpWidget') return false;
+
+    final library = staticElement?.library;
+    if (library == null) return false;
+    final libraryUri = Uri.tryParse(library.identifier);
+    if (libraryUri == null) return false;
+
+    return libraryUri.scheme == 'package' &&
+        libraryUri.pathSegments.first == 'flutter_test';
+  }
 }
 
 class ScopedProvidersShouldSpecifyDependencies extends RiverpodLintRule {
@@ -90,6 +102,7 @@ class ScopedProvidersShouldSpecifyDependencies extends RiverpodLintRule {
 
     // If the ProviderScope isn't directly as a child of runApp, it is scoped
     return enclosingExpression is! MethodInvocation ||
-        !enclosingExpression.methodName.isFlutterRunApp;
+        !enclosingExpression.methodName.isFlutterRunApp ||
+        !enclosingExpression.methodName.isPumpWidget;
   }
 }