Zipmerge: allow test to be run from internal repo

Author: redsun82

misc/bazel/internal/zipmerge/zipmerge_test.cpp

@@ -5,13 +5,15 @@
 #include <sstream>
 #include <string>
 #include <cstring>
+#include <filesystem>
 
 #include <gtest/gtest.h>
 #include <gmock/gmock.h>
 #include "tools/cpp/runfiles/runfiles.h"
 
 using bazel::tools::cpp::runfiles::Runfiles;
 using namespace std::string_literals;
+namespace fs = std::filesystem;
 
 namespace codeql_testing {
 
@@ -72,7 +74,15 @@ std::string get_file(const char* name) {
     EXPECT_TRUE(ret) << error;
     return ret;
   }();
-  return runfiles->Rlocation("_main/misc/bazel/internal/zipmerge/test-files/"s + name);
+  // this works from both `codeql` and the internal repository
+  for (auto prefix : {"_main", "codeql~"}) {
+    auto ret = runfiles->Rlocation(prefix + "/misc/bazel/internal/zipmerge/test-files/"s + name);
+    if (fs::exists(ret)) {
+      return ret;
+    }
+  }
+  EXPECT_TRUE(false) << "test file " << name << " not found";
+  return "";
 }
 
 void expect_same_file(const char* actual, const char* expected) {