def rule_runner() -> RuleRunner:
return RuleRunner(
rules=[
*source_files_rules(),
QueryRule(SourceFiles, (SourceFilesRequest,)),
],
)
def rules():
return [
*collect_rules(),
*artifact_mapper.rules(),
*java_parser_rules(),
*symbol_mapper.rules(),
*source_files_rules(),
UnionRule(InferDependenciesRequest, InferJavaSourceDependencies),
]
def rule_runner() -> RuleRunner:
return RuleRunner(
rules=[
*docker_rules(),
*source_files_rules(),
QueryRule(BuiltPackage, [DockerFieldSet]),
],
target_types=[DockerImage],
)
def rule_runner() -> RuleRunner:
return RuleRunner(
rules=[
*docker_rules(),
*source_files_rules(),
package.find_all_packageable_targets,
QueryRule(BuiltPackage, [DockerFieldSet]),
],
target_types=[DockerImageTarget],
)
def rule_runner(request) -> RuleRunner:
rule_runner_args: Mapping[str, Any] = dict(
rules=[
*dockerfile_rules(),
*source_files_rules(),
QueryRule(SourceFiles, [SourceFilesRequest]),
],
target_types=[DockerImageTarget],
)
if hasattr(request, "param") and callable(request.param):
request.param(rule_runner_args)
return RuleRunner(**rule_runner_args)
def rule_runner() -> RuleRunner:
rule_runner = RuleRunner(
rules=[
*archive_rules(),
*shell_command_rules(),
*source_files_rules(),
*target_type_rules(),
QueryRule(GeneratedSources,
[GenerateFilesFromShellCommandRequest]),
QueryRule(TransitiveTargets, [TransitiveTargetsRequest]),
QueryRule(SourceFiles, [SourceFilesRequest]),
],
target_types=[ShellCommand, ShellSourcesGeneratorTarget, Files],
)
rule_runner.set_options([], env_inherit={"PATH"})
return rule_runner
def test_relocated_files() -> None:
rule_runner = RuleRunner(
rules=[
*target_type_rules(),
*source_files_rules(),
QueryRule(GeneratedSources, [RelocateFilesViaCodegenRequest]),
QueryRule(TransitiveTargets, [TransitiveTargetsRequest]),
QueryRule(SourceFiles, [SourceFilesRequest]),
],
target_types=[Files, RelocatedFiles],
)
def assert_prefix_mapping(
*,
original: str,
src: str,
dest: str,
expected: str,
) -> None:
rule_runner.create_file(original)
rule_runner.add_to_build_file(
"",
dedent(f"""\
files(name="original", sources=[{repr(original)}])
relocated_files(
name="relocated",
files_targets=[":original"],
src={repr(src)},
dest={repr(dest)},
)
"""),
overwrite=True,
)
tgt = rule_runner.get_target(Address("", target_name="relocated"))
result = rule_runner.request(
GeneratedSources,
[RelocateFilesViaCodegenRequest(EMPTY_SNAPSHOT, tgt)])
assert result.snapshot.files == (expected, )
# We also ensure that when looking at the transitive dependencies of the `relocated_files`
# target and then getting all the code of that closure, we only end up with the relocated
# files. If we naively marked the original files targets as a typical `Dependencies` field,
# we would hit this issue.
transitive_targets = rule_runner.request(
TransitiveTargets, [TransitiveTargetsRequest([tgt.address])])
all_sources = rule_runner.request(
SourceFiles,
[
SourceFilesRequest(
(tgt.get(Sources) for tgt in transitive_targets.closure),
enable_codegen=True,
for_sources_types=(FilesSources, ),
)
],
)
assert all_sources.snapshot.files == (expected, )
# No-op.
assert_prefix_mapping(original="old_prefix/f.ext",
src="",
dest="",
expected="old_prefix/f.ext")
assert_prefix_mapping(
original="old_prefix/f.ext",
src="old_prefix",
dest="old_prefix",
expected="old_prefix/f.ext",
)
# Remove prefix.
assert_prefix_mapping(original="old_prefix/f.ext",
src="old_prefix",
dest="",
expected="f.ext")
assert_prefix_mapping(original="old_prefix/subdir/f.ext",
src="old_prefix",
dest="",
expected="subdir/f.ext")
# Add prefix.
assert_prefix_mapping(original="f.ext",
src="",
dest="new_prefix",
expected="new_prefix/f.ext")
assert_prefix_mapping(
original="old_prefix/f.ext",
src="",
dest="new_prefix",
expected="new_prefix/old_prefix/f.ext",
)
# Replace prefix.
assert_prefix_mapping(
original="old_prefix/f.ext",
src="old_prefix",
dest="new_prefix",
expected="new_prefix/f.ext",
)
assert_prefix_mapping(
original="old_prefix/f.ext",
src="old_prefix",
dest="new_prefix/subdir",
expected="new_prefix/subdir/f.ext",
)
# Replace prefix, but preserve a common start.
assert_prefix_mapping(
original="common_prefix/foo/f.ext",
src="common_prefix/foo",
dest="common_prefix/bar",
expected="common_prefix/bar/f.ext",
)
assert_prefix_mapping(
original="common_prefix/subdir/f.ext",
src="common_prefix/subdir",
dest="common_prefix",
expected="common_prefix/f.ext",
)
def rules():
return (*collect_rules(), *source_root_rules(), *source_files_rules())
def rule_runner() -> RuleRunner:
return RuleRunner(rules=[
*source_files_rules(),
QueryRule(SourceFiles, (SourceFilesRequest, OptionsBootstrapper)),
], )
def rules(cls):
return (*super().rules(), *source_files_rules())
