register the target type

pantsbuild · Nov 26, 2022 · 679b5a9 · 679b5a9
1 parent 406bbc7
commit 679b5a9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/src/python/pants/backend/shell/goals/test.py b/src/python/pants/backend/shell/goals/test.py
@@ -27,6 +27,7 @@
 from pants.engine.process import FallibleProcessResult, Process, ProcessCacheScope
 from pants.engine.rules import collect_rules, rule
 from pants.engine.target import Target, WrappedTarget, WrappedTargetRequest
+from pants.engine.unions import UnionRule
 from pants.util.frozendict import FrozenDict
 from pants.util.logging import LogLevel
 
@@ -99,4 +100,5 @@ def rules():
         *collect_rules(),
         *shell_command.rules(),
         *ShellTestRequest.rules(),
+        UnionRule(TestFieldSet, TestShellCommandFieldSet),
     )
diff --git a/src/python/pants/backend/shell/register.py b/src/python/pants/backend/shell/register.py
@@ -12,6 +12,7 @@
 from pants.backend.shell.target_types import (
     ShellCommandRunTarget,
     ShellCommandTarget,
+    ShellCommandTestTarget,
     ShellSourcesGeneratorTarget,
     ShellSourceTarget,
     Shunit2TestsGeneratorTarget,
@@ -24,6 +25,7 @@ def target_types():
     return [
         ShellCommandTarget,
         ShellCommandRunTarget,
+        ShellCommandTestTarget,
         ShellSourcesGeneratorTarget,
         Shunit2TestsGeneratorTarget,
         ShellSourceTarget,

diff --git a/src/python/pants/core/goals/test.py b/src/python/pants/core/goals/test.py
@@ -70,7 +70,7 @@
 
 
 @dataclass(frozen=True)
-class   TestResult(EngineAwareReturnType):
+class TestResult(EngineAwareReturnType):
     exit_code: int | None
     stdout: str
     stdout_digest: FileDigest