From ac3198ca95ef446884c109082b8b2188af193aa7 Mon Sep 17 00:00:00 2001
From: Fabian Meumertzheim <fabian@meumertzhe.im>
Date: Fri, 4 Nov 2022 22:53:46 +0100
Subject: [PATCH] Add integration test

---
 src/test/py/bazel/bzlmod/bazel_module_test.py | 63 +++++++++++++++++++
 1 file changed, 63 insertions(+)

diff --git a/src/test/py/bazel/bzlmod/bazel_module_test.py b/src/test/py/bazel/bzlmod/bazel_module_test.py
index 3a64544303514b..095f16f2469451 100644
--- a/src/test/py/bazel/bzlmod/bazel_module_test.py
+++ b/src/test/py/bazel/bzlmod/bazel_module_test.py
@@ -695,5 +695,68 @@ def testRunfilesRepoMappingManifest(self):
         self.Path('bazel-bin/external/bar~2.0/bar.runfiles_manifest')) as f:
       self.assertIn("_repo_mapping ", f.read())
 
+  def testJavaRunfilesLibraryRepoMapping(self):
+    self.main_registry.setModuleBasePath('projects')
+    projects_dir = self.main_registry.projects
+
+    self.main_registry.createLocalPathModule('data', '1.0', 'data')
+    projects_dir.joinpath('data').mkdir(exist_ok=True)
+    scratchFile(projects_dir.joinpath('data', 'WORKSPACE'))
+    scratchFile(projects_dir.joinpath('data', 'foo.txt'), ['hello'])
+    scratchFile(projects_dir.joinpath('data', 'BUILD'),
+                ['exports_files(["foo.txt"])'])
+
+    self.main_registry.createLocalPathModule('test', '1.0', 'test',
+                                             {'data': '1.0'})
+    projects_dir.joinpath('test').mkdir(exist_ok=True)
+    scratchFile(projects_dir.joinpath('test', 'WORKSPACE'))
+    scratchFile(projects_dir.joinpath('test', 'BUILD'), [
+      'java_test(',
+      '    name = "test",'
+      '    srcs = ["Test.java"],',
+      '    main_class = "com.example.Test",',
+      '    use_testrunner = False,',
+      '    data = ["@data//:foo.txt"],',
+      '    args = ["$(rlocationpath @data//:foo.txt)"],',
+      '    deps = ["@bazel_tools//tools/java/runfiles"],',
+      ')'
+    ])
+    scratchFile(projects_dir.joinpath('test', 'Test.java'), [
+      'package com.example;',
+      '',
+      'import com.google.devtools.build.runfiles.AutoBazelRepository;',
+      'import com.google.devtools.build.runfiles.Runfiles;',
+      '',
+      'import java.io.File;',
+      'import java.io.IOException;',
+      '',
+      '@AutoBazelRepository',
+      'public class Test {',
+      '  public static void main(String[] args) throws IOException {',
+      '    Runfiles.Preloaded rp = Runfiles.preload();',
+      '    if (!new File(rp.unmapped().rlocation(args[0])).exists()) {',
+      '      System.exit(1);',
+      '    }',
+      '    if (!new File(rp.withSourceRepository(AutoBazelRepository_Test.NAME).rlocation("data/foo.txt")).exists()) {',
+      '      System.exit(1);',
+      '    }',
+      '  }',
+      '}',
+    ])
+
+    self.ScratchFile('MODULE.bazel', ['bazel_dep(name="test",version="1.0")'])
+    self.ScratchFile('WORKSPACE')
+
+    # Run sandboxed on Linux and macOS.
+    exit_code, stderr, stdout = self.RunBazel(
+        ['test', '@test//:test', '--test_output=errors',
+         '--test_env=RUNFILES_LIB_DEBUG=1'], allow_failure=True)
+    self.AssertExitCode(exit_code, 0, stderr, stdout)
+    # Run unsandboxed on all platforms.
+    exit_code, stderr, stdout = self.RunBazel(
+        ['run', '@test//:test'], allow_failure=True,
+        env_add={"RUNFILES_LIB_DEBUG": "1"})
+    self.AssertExitCode(exit_code, 0, stderr, stdout)
+
 if __name__ == '__main__':
   unittest.main()