commit 6acdc6465eb2000dee877ee1b06bb1036b689af7
author Googler <no-reply@google.com> Thu Oct 19 08:03:40 2023 -0700
committer Copybara-Service <copybara-worker@google.com> Thu Oct 19 08:04:26 2023 -0700
tree 5b8ce26c0f9f4dd6fe4de3fe03a641684eb65703
parent f1f793d61054277cf26b4076a753945839dd515d

Store USR cache outside of EvidenceEmitter so that it can be shared in more places.

For example, the current function decl's USR is needed to look up whether inferable slots' nullabilities were inferred in previous rounds.

PiperOrigin-RevId: 574873175
Change-Id: Id96d5cb082e7d61464bcc2bdd6825858eac16015

nullability/inference/collect_evidence_test.cc

diff --git a/nullability/inference/collect_evidence_test.cc b/nullability/inference/collect_evidence_test.cc
index 5202fa7..5b08e1d 100644
--- a/nullability/inference/collect_evidence_test.cc
+++ b/nullability/inference/collect_evidence_test.cc
@@ -69,10 +69,13 @@
     llvm::StringRef Source) {
   std::vector<Evidence> Results;
   clang::TestAST AST(getInputsWithAnnotationDefinitions(Source));
+  USRCache usr_cache;
   auto Err = collectEvidenceFromImplementation(
       cast<FunctionDecl>(
           *dataflow::test::findValueDecl(AST.context(), "target")),
-      evidenceEmitter([&](const Evidence& E) { Results.push_back(E); }));
+      evidenceEmitter([&](const Evidence& E) { Results.push_back(E); },
+                      usr_cache),
+      usr_cache);
   if (Err) ADD_FAILURE() << toString(std::move(Err));
   return Results;
 }
@@ -80,9 +83,11 @@
 std::vector<Evidence> collectEvidenceFromTargetDecl(llvm::StringRef Source) {
   std::vector<Evidence> Results;
   clang::TestAST AST(getInputsWithAnnotationDefinitions(Source));
+  USRCache usr_cache;
   collectEvidenceFromTargetDeclaration(
       *dataflow::test::findValueDecl(AST.context(), "target"),
-      evidenceEmitter([&](const Evidence& E) { Results.push_back(E); }));
+      evidenceEmitter([&](const Evidence& E) { Results.push_back(E); },
+                      usr_cache));
   return Results;
 }
 
@@ -472,10 +477,13 @@
       "target", TargetInstantiationNodes);
   ASSERT_NE(InstantiationDecl, nullptr);
 
+  USRCache usr_cache;
   std::vector<Evidence> Results;
   auto Err = collectEvidenceFromImplementation(
       *InstantiationDecl,
-      evidenceEmitter([&](const Evidence& E) { Results.push_back(E); }));
+      evidenceEmitter([&](const Evidence& E) { Results.push_back(E); },
+                      usr_cache),
+      usr_cache);
   if (Err) ADD_FAILURE() << toString(std::move(Err));
   EXPECT_THAT(Results, IsEmpty());
 }
@@ -610,7 +618,8 @@
       dataflow::test::findValueDecl(AST.context(), "target");
   ASSERT_NE(TargetDecl, nullptr);
 
-  EXPECT_DEATH(evidenceEmitter([](const Evidence& e) {})(
+  USRCache usr_cache;
+  EXPECT_DEATH(evidenceEmitter([](const Evidence& e) {}, usr_cache)(
                    *TargetDecl, Slot{}, Evidence::ANNOTATED_UNKNOWN,
                    TargetDecl->getLocation()),
                "not an inference target");