henitai 0.5.3 → 0.6.0

data/sig/henitai.rbs

@@ -128,14 +128,42 @@ module Henitai
     def pause: (Float) -> void
   end
 
+  module Integration::RspecPreload
+    def fork_server_supported?: () -> bool
+    def preload_environment: (Array[String]) -> untyped
+  end
+
+  module Integration::MinitestPreload
+    def fork_server_supported?: () -> bool
+    def preload_environment: (Array[String]) -> untyped
+  end
+
   module Integration::RspecChildRunner
     private
 
-    def run_rspec_runner: (Array[String]) -> untyped
+    def run_rspec_runner: (Array[String]) ?{ () -> untyped } -> untyped
     def build_rspec_runner: () -> untyped
     def configure_rspec_runner: (untyped) -> void
     def load_rspec_spec_files: (Array[String]) -> void
-    def run_rspec_specs: (untyped) -> untyped
+    def run_rspec_specs: (untyped, untyped) -> untyped
+  end
+
+  module RspecOutcomes
+    EVENTS: Array[Symbol]
+    HOOK_ERROR: String
+
+    def self.record: (TestFileOutcomes, untyped) -> void
+    def self.register: (TestFileOutcomes, untyped) -> void
+    def self.group_file: (untyped) -> String
+  end
+
+  class RspecOutcomes::Listener
+    def initialize: (TestFileOutcomes) -> void
+    def example_finished: (untyped) -> void
+    def example_group_started: (untyped) -> void
+    def example_group_finished: (untyped) -> void
+    def message: (untyped) -> void
+    def dump_summary: (untyped) -> void
   end
 
   module Integration::RspecTestSelection
@@ -187,6 +215,9 @@ module Henitai
     attr_reader ignore_patterns: Array[untyped]
     attr_reader max_mutants_per_line: Integer
     attr_reader max_flaky_retries: Integer
+    attr_reader control_runs: bool
+    attr_reader flaky_detection: bool
+    attr_reader fork_server: bool
     attr_reader max_log_bytes: Integer
     attr_reader max_timeout: Float
     attr_reader sampling: untyped
@@ -227,9 +258,14 @@ module Henitai
     attr_reader stderr: String
     attr_reader exit_status: Integer?
     attr_reader log_path: String
+    attr_reader status_reason: String?
+    attr_reader killing_tests: Array[String]
 
     def self.build: (wait_result: untyped, stdout: String?, stderr: String?, log_path: String) -> ScenarioExecutionResult
-    def initialize: (status: Symbol, stdout: String?, stderr: String?, log_path: String, ?exit_status: Integer?) -> void
+    def self.build_for_mutant: (wait_result: untyped, report: untyped, stdout: String?, stderr: String?, log_path: String) -> ScenarioExecutionResult
+    def self.copy_reason: (untyped, untyped) -> String?
+    def initialize: (status: Symbol, stdout: String?, stderr: String?, log_path: String, ?exit_status: Integer?, ?status_reason: String?, ?killing_tests: Array[String]) -> void
+    def release_output!: () -> self
     def survived?: () -> bool
     def killed?: () -> bool
     def timeout?: () -> bool
@@ -251,9 +287,10 @@ module Henitai
     attr_reader method_type: Symbol
     attr_reader source_file: String?
     attr_reader source_range: Range[Integer]?
+    attr_reader lexical_scope: Array[String]?
 
     def self.parse: (String) -> Subject
-    def initialize: (?expression: String, ?namespace: String, ?method_name: String, ?method_type: Symbol, ?source_location: Hash[Symbol, untyped], ?ast_node: untyped) -> void
+    def initialize: (?expression: String, ?namespace: String, ?method_name: String, ?method_type: Symbol, ?source_location: Hash[Symbol, untyped], ?ast_node: untyped, ?lexical_scope: Array[String]?) -> void
     def expression: () -> String
     def wildcard?: () -> bool
 
@@ -281,11 +318,14 @@ module Henitai
     attr_accessor covered_by: Array[String]?
     attr_accessor tests_completed: Integer?
     attr_accessor ignore_reason: String?
+    attr_accessor status_reason: String?
+    attr_accessor identity_discriminator: String?
     attr_accessor from_cache: bool
     attr_reader precomputed_stable_id: String?
-    attr_reader precomputed_activation_source: String?
+    attr_reader precomputed_activation: untyped
+    attr_reader precomputed_replacement: String?
 
-    def initialize: (subject: Subject, operator: String, nodes: Hash[Symbol, untyped], description: String, location: Hash[Symbol, untyped], ?precomputed_stable_id: String?, ?precomputed_activation_source: String?) -> void
+    def initialize: (subject: Subject, operator: String, nodes: Hash[Symbol, untyped], description: String, location: Hash[Symbol, untyped], ?precomputed_stable_id: String?, ?precomputed_activation: untyped, ?precomputed_replacement: String?) -> void
     def stable_id: () -> String
     def mutated_source: () -> String
     def from_cache?: () -> bool
@@ -295,6 +335,10 @@ module Henitai
     def ignored?: () -> bool
     def equivalent?: () -> bool
     def to_s: () -> String
+
+    private
+
+    def reset_outcome: () -> void
   end
 
   class SyntaxValidator
@@ -317,13 +361,22 @@ module Henitai
     def current_snapshot: () -> untyped
     def warn_missing_coverage: () -> void
     def new_lines: (untyped) -> Hash[String, Array[Integer]]
-    def new_line_numbers: (untyped, untyped) -> Array[Integer]
-    def previous_line_counts: (String) -> Array[untyped]
-    def line_counts_for: (untyped) -> Array[untyped]
     def source_file?: (String) -> bool
     def serializable_report: () -> Hash[String, Hash[String, Array[Integer]]]
   end
 
+  class PerTestCoverageDelta
+    def executed_lines: (untyped, untyped) -> Array[Integer]
+
+    private
+
+    def counted_lines: (untyped, untyped) -> Array[Integer]
+    def called_method_lines: (untyped, untyped) -> Array[Integer]
+    def untracked_method_lines: (Array[untyped], Array[untyped]) -> Array[Integer]
+    def line_counts: (untyped) -> Array[untyped]
+    def method_counts: (untyped) -> Hash[untyped, untyped]
+  end
+
   class CoverageFormatter
     REPORT_DIR_ENV: String
     REPORT_FILE_NAME: String
@@ -344,22 +397,54 @@ module Henitai
   end
 
   class SamplingStrategy
+    DEFAULT_SEED: Integer
+
     def initialize: () -> void
-    def sample: (Array[Mutant], ratio: Float, ?strategy: Symbol) -> Array[Mutant]
+    def sample: (Array[Mutant], ratio: Float, ?strategy: Symbol, ?seed: Integer) -> Array[Mutant]
   end
 
   class TimeoutCalibrator
     FLOOR_SECONDS: Float
 
-    def initialize: (timing_source: untyped, multiplier: untyped) -> void
+    def initialize: (timing_source: untyped, multiplier: untyped, ?boot_seconds: ^() -> Float, ?include_boot: bool) -> void
     def timeout_for: (untyped) -> Float?
 
     private
 
+    def boot_component: () -> Float
     def durations: () -> Hash[String, untyped]
     def normalize: (untyped) -> String
   end
 
+  class TimeoutCeiling
+    def initialize: (ceiling: Float?, warner: untyped) -> void
+    def cap: (Float, Float) -> Float
+
+    private
+
+    def warn_baseline_exceeds_ceiling: (Float) -> void
+  end
+
+  class SuiteBootClock
+    STARTED_AT_ENV: String
+
+    def self.from_env: (Hash[String, String], ?clock: ^() -> Float) -> SuiteBootClock
+    def self.parse_timestamp: (untyped) -> Float?
+    def initialize: (started_at: Float?, ?clock: ^() -> Float) -> void
+    def mark_first_test_started: () -> void
+    def boot_seconds: () -> Float
+    def measured?: () -> bool
+  end
+
+  class ProcessWakeup
+    def initialize: (?signal_name: String) -> void
+    def install: () -> self
+    def watch: () { () -> Array[untyped] } -> void
+    def wait: (Float?) -> void
+    def drain: () -> void
+    def close: () -> void
+  end
+
   class TestPrioritizer
     def initialize: (?timing_source: untyped?) -> void
     def sort: (Array[String], untyped, Hash[untyped, untyped]) -> Array[String]
@@ -432,6 +517,11 @@ module Henitai
       def run_child_activation_and_tests: (mutant: Mutant, test_files: Array[String], log_paths: Hash[Symbol, String]) -> Integer
     end
 
+    class CappedLogSink
+      def initialize: (untyped, Integer) -> void
+      def write: (String) -> void
+    end
+
     class ScenarioLogSupport
       def capture_child_output: (Hash[Symbol, String]) { () -> untyped } -> untyped
       def with_coverage_dir: (String) { () -> untyped } -> untyped
@@ -465,8 +555,9 @@ module Henitai
     class ChildHandle
       attr_accessor pid: Integer?
       attr_accessor log_paths: Hash[Symbol, String]
+      attr_accessor timeout: Float?
 
-      def initialize: (pid: Integer?, log_paths: Hash[Symbol, String]) -> void
+      def initialize: (pid: Integer?, log_paths: Hash[Symbol, String], ?timeout: Float?) -> void
     end
 
     class RspecProcessRunner
@@ -478,6 +569,7 @@ module Henitai
     class Rspec < Base
       include MutantRunSupport
       include RspecChildRunner
+      include RspecPreload
       include RspecTestSelection
 
       DEFAULT_SUITE_TIMEOUT: Float
@@ -491,7 +583,8 @@ module Henitai
 
       private
 
-      def run_tests: (Array[String]) -> Integer
+      def outcome_recording_hook: (TestFileOutcomes, (^() -> untyped)?) -> (^() -> untyped)
+      def run_tests: (Array[String], TestFileOutcomes) ?{ () -> untyped } -> Integer
     end
 
     module CoverageRuntimeSuppressors
@@ -522,7 +615,7 @@ module Henitai
 
       private
 
-      def run_tests: (Array[String]) -> Integer
+      def run_tests: (Array[String], TestFileOutcomes) ?{ () -> untyped } -> Integer
       def preload_environment: () -> void
       def setup_load_path: () -> void
       def suppress_minitest_autorun!: () -> nil
@@ -541,12 +634,13 @@ module Henitai
 
   class GitDiffAnalyzer
     def changed_files: (from: String, to: String, ?dir: String) -> Array[String]
+    def changed_files_since: (String, ?dir: String) -> Array[String]
     def working_tree_changed_files: (?dir: String) -> Array[String]
     def head_sha: (?dir: String) -> String?
   end
 
   class MutantGenerator
-    def generate: (Array[Subject], Array[untyped], ?config: untyped) -> Array[Mutant]
+    def generate: (Array[Subject], Array[untyped], ?config: untyped, ?sample: bool) -> Array[Mutant]
   end
 
   module MutantIdentity
@@ -557,7 +651,7 @@ module Henitai
   class MutantHistoryStore
     attr_reader path: String
 
-    def initialize: (path: String, ?per_test_coverage: PerTestCoverage?) -> void
+    def initialize: (path: String, ?per_test_coverage: PerTestCoverage?, ?reuse_context: ReuseContext?) -> void
     def record: (untyped, version: String, ?recorded_at: Time) -> void
     def verdict_for: (String) -> Hash[Symbol, untyped]?
     def killed_verdict_for: (String) -> Hash[Symbol, untyped]?
@@ -607,6 +701,7 @@ module Henitai
     def coverage_lines_by_file: (?String) -> Hash[String, Array[Integer]]
     def test_lines_by_file: (?String) -> Hash[String, Hash[String, Array[Integer]]]
     def durations_by_test: (?String) -> Hash[String, Float]
+    def boot_seconds: (?String) -> Float
   end
 
   class ProcessWorkerRunner
@@ -649,14 +744,15 @@ module Henitai
   end
 
   class IncrementalFilter
-    def initialize: (history_store: untyped, ?per_test_coverage: PerTestCoverage?, ?dependency_fingerprint: String?) -> void
+    def initialize: (history_store: untyped, ?per_test_coverage: PerTestCoverage?, ?reuse_context: ReuseContext?) -> void
     def apply: (Array[Mutant]) -> Array[Mutant]
 
     private
 
     def history_store: () -> untyped
     def per_test_coverage: () -> PerTestCoverage?
-    def dependency_fingerprint: () -> String?
+    def reuse_context: () -> ReuseContext?
+    def context_sha: () -> String?
     def ambiguous_stable_ids: (Array[Mutant]) -> Set[String]
     def reusable_status: (Mutant) -> Symbol?
     def killed_reusable?: (Mutant, Hash[Symbol, untyped]) -> bool
@@ -664,15 +760,25 @@ module Henitai
     def subject_source_unchanged?: (Mutant, Hash[Symbol, untyped]) -> bool
   end
 
+  class ReuseContext
+    def initialize: (config: untyped, ?version: String, ?root: String) -> void
+    def fingerprint: () -> String?
+
+    private
+
+    def compute: () -> String?
+    def components: (String) -> Hash[String, untyped]
+  end
+
   module VerdictFingerprint
     DEPENDENCY_GLOBS: Array[String]
 
     def self?.subject_source_hash: (untyped) -> String?
-    def self?.tests_fingerprint: (untyped) -> String?
-    def self?.tests_fingerprint_current?: (String?) -> bool
+    def self?.tests_fingerprint: (untyped, context_sha: String?) -> String?
+    def self?.tests_fingerprint_current?: (String?, context_sha: String?) -> bool
     def self?.dependency_fingerprint: (?String) -> String?
-    def self?.survivor_tests_fingerprint: (untyped, dependency_sha: String?) -> String?
-    def self?.survivor_fingerprint_current?: (String?, live_paths: Array[String], dependency_sha: String?) -> bool
+    def self?.survivor_tests_fingerprint: (untyped, context_sha: String?) -> String?
+    def self?.survivor_fingerprint_current?: (String?, live_paths: Array[String], context_sha: String?) -> bool
     def self?.survivor_fingerprint_matches?: (untyped, Array[String], String?) -> bool
     def self?.combined_content_sha: (untyped) -> String?
   end
@@ -741,9 +847,13 @@ module Henitai
     def map: () -> Hash[String, Hash[String, Array[Integer]]]
   end
 
+  module SignatureLine
+    def self.on?: (untyped) -> bool
+  end
+
   class PerTestCoverageSelector
     def initialize: (?coverage_report_reader: CoverageReportReader) -> void
-    def filter: (Array[String], Mutant, reports_dir: String) -> Array[String]
+    def select: (Mutant, reports_dir: String, test_files: ^() -> Array[String], fallback: ^() -> Array[String]) -> Array[String]
 
     private
 
@@ -767,7 +877,6 @@ module Henitai
     def test_prioritizer: (untyped) -> TestPrioritizer
     def timing_source: (untyped) -> untyped
     def test_history: (untyped) -> Hash[untyped, untyped]
-    def run_with_flaky_retry: (Mutant, untyped, untyped, Array[String], ?untyped) -> ScenarioExecutionResult
     def warn_flaky_mutants: (Integer) -> void
     def with_reports_dir: (untyped) { () -> untyped } -> untyped
   end
@@ -780,8 +889,6 @@ module Henitai
 
     private
 
-    def load_target: (Subject) -> untyped
-    def load_source_file: (Subject) -> void
     def target_for: (Subject) -> untyped
     def method_source: (Mutant) -> String
     def body_source: (Mutant) -> String
@@ -790,11 +897,28 @@ module Henitai
     def same_node?: (untyped, untyped) -> bool
     def replace_child: (untyped, untyped, untyped) -> untyped
     def method_body: (untyped) -> untyped
-    def source_file_from_ast: (Subject) -> String?
     def node_location_signature: (untyped) -> Array[untyped]?
     def compile_safe_unparse: (untyped) -> String
   end
 
+  class Mutant::LexicalTarget
+    class SubjectNotLoadedError < StandardError
+    end
+
+    def self.resolve: (Subject, untyped) -> untyped
+    def self.load_target: (Subject) -> untyped
+    def resolve: (Subject, untyped) -> untyped
+    def load_target: (Subject) -> untyped
+
+    private
+
+    def ensure_loaded: (Subject, Array[String]) -> void
+    def load_toplevel_source: (Subject) -> untyped
+    def load_subject_source: (Subject) -> void
+    def load_source_file: (Subject) -> void
+    def source_file_from_ast: (Subject) -> String?
+  end
+
   class Mutant::ParameterSource
     SERIALIZER_METHODS: Hash[Symbol, Symbol]
 
@@ -830,6 +954,7 @@ module Henitai
       def history_store: () -> MutantHistoryStore?
       def authoritative?: (untyped) -> bool
       def canonical_path: () -> String
+      def merged_schema_for: (untyped) -> untyped
     end
 
     class Terminal < Base
@@ -991,10 +1116,12 @@ module Henitai
     attr_reader survivor_stats: Hash[Symbol, untyped]?
     attr_reader session_id: String
     attr_reader git_sha: String?
+    attr_reader complete_files: Array[String]
     attr_reader since: String?
     attr_reader coverage_criteria: Hash[Symbol, bool]
+    attr_reader sampling: Hash[Symbol, untyped]?
 
-    def initialize: (mutants: Array[Mutant], started_at: Time, finished_at: Time, ?thresholds: Hash[Symbol, Integer]?, ?partial_rerun: bool, ?survivor_stats: Hash[Symbol, untyped]?, ?session_id: String, ?git_sha: String?, ?source_provider: ^(String) -> String, ?authoritative: bool, ?since: String?, ?coverage_criteria: Hash[Symbol, bool]?) -> void
+    def initialize: (mutants: Array[Mutant], started_at: Time, finished_at: Time, ?thresholds: Hash[Symbol, Integer]?, ?partial_rerun: bool, ?survivor_stats: Hash[Symbol, untyped]?, ?session_id: String, ?git_sha: String?, ?source_provider: ^(String) -> String, ?authoritative: bool, ?since: String?, ?coverage_criteria: Hash[Symbol, bool]?, ?complete_files: Array[String]?, ?sampling: Hash[Symbol, untyped]?) -> void
     def partial_rerun?: () -> bool
     def authoritative?: () -> bool
     def killed: () -> Integer
@@ -1017,6 +1144,7 @@ module Henitai
     def base_schema: () -> Hash[Symbol, untyped]
     def unmatched_survivor_ids: () -> Array[String]
     def build_files_section: () -> Hash[Symbol, untyped]
+    def file_provenance: () -> Hash[Symbol, String]
     def mutant_to_schema: (Mutant) -> Hash[Symbol, untyped]
     def coverage_schema: (Mutant) -> Hash[Symbol, untyped]
     def replacement_for: (Mutant) -> String
@@ -1047,17 +1175,6 @@ module Henitai
     def load: () -> Report
   end
 
-  class SurvivorTestFilter
-    def initialize: (
-      coverage_map: Hash[String, Array[String]],
-      git_sha: String?,
-      ?dirty_source_files: bool,
-      ?worktree_changed_files: Array[String],
-      ?diff_analyzer: GitDiffAnalyzer
-    ) -> void
-    def apply: (Array[Mutant]) -> Hash[Symbol, Array[Mutant]]
-  end
-
   class SurvivorSelector
     DRIFT_THRESHOLD: Float
 
@@ -1086,6 +1203,9 @@ module Henitai
     def integration: () -> untyped
     def operators: () -> Array[Operator]
     def per_test_coverage: () -> PerTestCoverage
+    def source_selection_coverage: () -> PerTestCoverage
+    def reuse_context: () -> ReuseContext
+    def reuse_filter: () -> IncrementalFilter
     def history_store: () -> untyped
     def progress_reporter: (full_run: bool) -> untyped
     def source_provider: () -> ^(String) -> String
@@ -1144,6 +1264,8 @@ module Henitai
     def progress_reporter: () -> untyped?
     def history_store: () -> untyped
     def per_test_coverage: () -> PerTestCoverage
+    def reuse_filter: () -> IncrementalFilter
+    def source_selection_coverage: () -> PerTestCoverage
     def history_store_path: () -> String
     def source_files: () -> Array[String]
     def source_file_selection: () -> SourceFileSelection
@@ -1164,9 +1286,11 @@ module Henitai
     def with_reports_dir: () { () -> untyped } -> untyped
     def result_thresholds: () -> Hash[Symbol, Integer]?
     def result_coverage_criteria: () -> Hash[Symbol, bool]?
+    def result_sampling: () -> Hash[Symbol, untyped]?
     def optional_config: (Symbol) -> untyped
     def survivor_rerun?: () -> bool
     def full_run?: () -> bool
+    def complete_files: () -> Array[String]
     def survivor_strategy: () -> SurvivorRerunStrategy
     def safe_head_sha: () -> String?
   end
@@ -1177,27 +1301,26 @@ module Henitai
     def initialize: (survivors_from: String?, config: Configuration, git_diff_analyzer: untyped) -> void
     def active?: () -> bool
     def try_recipe_run: () -> Array[Mutant]?
-    def apply_selection: (Array[Mutant]) -> Array[Mutant]
+    def apply_selection: (Array[Mutant], ?reuse_filter: IncrementalFilter?) -> Array[Mutant]
 
     private
 
     def load_survivor_report: () -> SurvivorLoader::Report
-    def run_from_recipes: (SurvivorLoader::Report, Array[String]?) -> Array[Mutant]?
-    def recipe_fast_path_safe?: (SurvivorLoader::Report, Array[String]?) -> bool
+    def run_from_recipes: (SurvivorLoader::Report) -> Array[Mutant]?
+    def recipe_fast_path_safe?: (SurvivorLoader::Report) -> bool
     def recipe_selector_and_stubs: (Array[String], Hash[String, Hash[String, untyped]]) -> [SurvivorSelector, Array[Mutant]]
     def load_activation_recipes: (Array[String]) -> Hash[String, Hash[String, untyped]]?
     def build_stub_mutant: (String, Hash[String, untyped]) -> Mutant
     def stub_subject_from_recipe: (Hash[String, untyped]) -> Subject
     def recipe_location: (Hash[String, untyped]?) -> Hash[Symbol, untyped]
-    def finalize_survivor_split: (SurvivorSelector, Array[Mutant], Hash[Symbol, Array[Mutant]]) -> Array[Mutant]
+    def finalize_selection: (SurvivorSelector, Array[Mutant]) -> Array[Mutant]
     def dirty_worktree_changed_files: () -> Array[String]?
     def dirty_source_files?: (Array[String]?, ?git_sha: String?) -> bool
     def committed_changed_files: (String?) -> Array[String]
     def in_include_root?: (String, Array[String]) -> bool
     def normalize_path: (String) -> String
     def warn_survivor_drift: (SurvivorSelector) -> void
-    def build_survivor_stats: (SurvivorSelector, Array[Mutant], Hash[Symbol, Array[Mutant]]) -> Hash[Symbol, untyped]
-    def test_filter: (SurvivorLoader::Report, ?dirty_source_files: bool) -> SurvivorTestFilter
+    def build_survivor_stats: (SurvivorSelector, Array[Mutant]) -> Hash[Symbol, untyped]
   end
 
   class CoverageBootstrapper
@@ -1211,6 +1334,8 @@ module Henitai
     def coverage_fresh?: (Array[String], Configuration, Integration::Base, Array[String]?) -> bool
     def coverage_report_path: (Configuration) -> String
     def bootstrap_coverage: (Integration::Base, Configuration, ?Array[String]?) -> void
+    def flaky_detection?: (Configuration) -> bool
+    def run_flaky_baseline: (Integration::Base, Configuration, Array[String]) -> void
     def resolve_test_files: (Integration::Base, Array[String]?, Configuration) -> Array[String]
     def configured_test_excludes: (Configuration) -> Array[String]?
     def source_file_paths: (Array[String]) -> Array[String]
@@ -1237,5 +1362,60 @@ module RSpec
     class Runner
       def self.run: (Array[String]) -> Integer
     end
+
+  end
+end
+
+module Henitai
+  class TestFileOutcomes
+    REPORT_FILE_NAME: String
+
+    def self.report_path: (String) -> String
+    def self.default_report_path: () -> String
+    def self.load: (String) -> TestFileOutcomes
+    def initialize: () -> void
+    def record: (String, failed: bool) -> void
+    def record_unattributed_failure: () -> void
+    def unattributed_failures?: () -> bool
+    def failed_files: () -> Array[String]
+    def outcomes: () -> Hash[String, String]
+    def write: (String) -> void
+  end
+
+  class FlakyTestRegistry
+    FILE_NAME: String
+
+    def self.from_outcomes: (Hash[String, String], Hash[String, String]) -> FlakyTestRegistry
+    def self.load: (String) -> FlakyTestRegistry
+    def self.for_config: (untyped) -> FlakyTestRegistry
+    def self.for_reports_dir: (String?) -> FlakyTestRegistry
+    def initialize: (Array[String]) -> void
+    def all: () -> Array[String]
+    def flaky?: (String) -> bool
+    def empty?: () -> bool
+    def write: (String) -> void
+  end
+
+  class FlakyDetectionBaseline
+    def initialize: (reports_dir: String) -> void
+    def run: (integration: untyped, test_files: Array[String]) -> FlakyTestRegistry
+  end
+
+  module BootstrapErrorMessage
+    GENERIC: String
+
+    def self.build: (untyped) -> String
+  end
+
+  class FlakyRetryDecision
+    def initialize: (flaky_tests: FlakyTestRegistry, max_retries: Integer) -> void
+    def retry?: (status: Symbol, killing_tests: Array[String], attempts: Integer) -> bool
+  end
+
+  class FlakyMutantRetrier
+    attr_reader retried_count: Integer
+
+    def initialize: (flaky_tests: FlakyTestRegistry, max_retries: Integer) -> void
+    def run: (mutant: Mutant, integration: untyped, test_files: Array[String], timeout: Float) -> untyped
   end
 end