Address different handling of runfiles in bzlmod.

comius · Sep 20, 2022 · 940e468 · 940e468
1 parent 27232d2
commit 940e468
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/tests/rules/proto_descriptor_set/proto_descriptor_set_test.cc b/tests/rules/proto_descriptor_set/proto_descriptor_set_test.cc
@@ -33,7 +33,12 @@ namespace {
 
 std::string GetRlocation(const std::string& file) {
   static std::unique_ptr<Runfiles> runfiles(Runfiles::CreateForTest());
-  return runfiles->Rlocation(file);
+  std::string path = runfiles->Rlocation(rulesproto::kWorkspaceRlocation + file);
+  std::ifstream input(path, std::ifstream::binary);
+  if (!input) {
+    path = runfiles->Rlocation(rulesproto::kWorkspaceRlocationBzlmod + file);
+  }
+  return path;
 }
 
 template <typename T, typename K>
@@ -66,7 +71,7 @@ void AssertFileDescriptorSetContains(
     const std::vector<std::string>& expected_proto_files) {
   std::vector<std::string> actual_proto_files =
       ReadFileDescriptorSet(
-          GetRlocation(rulesproto::kWorkspaceRlocation + path));
+          GetRlocation(path));
   EXPECT_EQ(expected_proto_files, actual_proto_files);
 }
 

diff --git a/tests/utils/workspace_constants.h b/tests/utils/workspace_constants.h
@@ -22,5 +22,6 @@
 namespace rulesproto {
 
 constexpr char kWorkspaceRlocation[] = "rules_proto/";
+constexpr char kWorkspaceRlocationBzlmod[] = "__main__/";
 
 }  // namespace rulesproto