18252174847

Committed 05 Oct 2025 01:36AM UTC coverage: 43.382% (-36.9%) from 80.261%

Build # 18252174847

Build Type

push

github

Committed by

web-flow

Commit Message

run tests on mac arm (#22717)

Just doing the minimal to pull forward the x86_64 pattern.

ref #20993

Run Details

25776 of 59416 relevant lines covered (43.38%)

1.3 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

/src/python/pants/backend/codegen/thrift/jvm_symbol_mapper.py

# Copyright 2023 Pants project contributors (see CONTRIBUTORS.md).
# Licensed under the Apache License, Version 2.0 (see LICENSE).
from __future__ import annotations

from collections import defaultdict
from collections.abc import Mapping
from dataclasses import dataclass
from typing import DefaultDict

from pants.backend.codegen.thrift.target_types import AllThriftTargets, ThriftSourceField
from pants.backend.codegen.thrift.thrift_parser import ParsedThriftRequest, parse_thrift_file
from pants.engine.addresses import Address
from pants.engine.rules import collect_rules, concurrently, rule
from pants.jvm.dependency_inference.artifact_mapper import MutableTrieNode
from pants.jvm.dependency_inference.symbol_mapper import SymbolMap
from pants.jvm.subsystems import JvmSubsystem
from pants.jvm.target_types import JvmResolveField
from pants.util.ordered_set import OrderedSet

_ResolveName = str


@dataclass(frozen=True)
class FirstPartyThriftJvmMappingRequest:
    extra_lang_ids: tuple[str, ...] = ()
    extra_namespace_directives: tuple[str, ...] = ()


@rule
async def map_first_party_thrift_targets_to_jvm_symbols(
    request: FirstPartyThriftJvmMappingRequest, thrift_targets: AllThriftTargets, jvm: JvmSubsystem
) -> SymbolMap:
    jvm_thrift_targets = [tgt for tgt in thrift_targets if tgt.has_field(JvmResolveField)]

    parsed_thrift_sources = await concurrently(
        parse_thrift_file(
            ParsedThriftRequest(
                sources_field=tgt[ThriftSourceField],
                extra_namespace_directives=request.extra_namespace_directives,
            )
        )
        for tgt in jvm_thrift_targets
    )

    package_symbols: DefaultDict[tuple[_ResolveName, str], OrderedSet[Address]] = defaultdict(
        OrderedSet
    )
    lang_ids = {"java", *request.extra_lang_ids}
    for target, parsed_thrift in zip(jvm_thrift_targets, parsed_thrift_sources):
        for lang_id in lang_ids:
            package_name = parsed_thrift.namespaces.get(lang_id)
            if not package_name:
                continue

            resolve = target[JvmResolveField].normalized_value(jvm)
            package_symbols[(resolve, package_name)].add(target.address)

    mapping: Mapping[str, MutableTrieNode] = defaultdict(MutableTrieNode)
    for (resolve, package_name), addresses in package_symbols.items():
        mapping[resolve].insert(package_name, addresses, first_party=True, recursive=True)

    return SymbolMap((resolve, node.frozen()) for resolve, node in mapping.items())


def rules():
    return collect_rules()

1	# Copyright 2023 Pants project contributors (see CONTRIBUTORS.md).
2	# Licensed under the Apache License, Version 2.0 (see LICENSE).
3	from __future__ import annotations	×
4
5	from collections import defaultdict	×
6	from collections.abc import Mapping	×
7	from dataclasses import dataclass	×
8	from typing import DefaultDict	×
9
10	from pants.backend.codegen.thrift.target_types import AllThriftTargets, ThriftSourceField	×
11	from pants.backend.codegen.thrift.thrift_parser import ParsedThriftRequest, parse_thrift_file	×
12	from pants.engine.addresses import Address	×
13	from pants.engine.rules import collect_rules, concurrently, rule	×
14	from pants.jvm.dependency_inference.artifact_mapper import MutableTrieNode	×
15	from pants.jvm.dependency_inference.symbol_mapper import SymbolMap	×
16	from pants.jvm.subsystems import JvmSubsystem	×
17	from pants.jvm.target_types import JvmResolveField	×
18	from pants.util.ordered_set import OrderedSet	×
19
20	_ResolveName = str	×
21
22
23	@dataclass(frozen=True)	×
24	class FirstPartyThriftJvmMappingRequest:	×
25	extra_lang_ids: tuple[str, ...] = ()	×
26	extra_namespace_directives: tuple[str, ...] = ()	×
27
28
29	@rule	×
30	async def map_first_party_thrift_targets_to_jvm_symbols(	×
31	request: FirstPartyThriftJvmMappingRequest, thrift_targets: AllThriftTargets, jvm: JvmSubsystem
32	) -> SymbolMap:
33	jvm_thrift_targets = [tgt for tgt in thrift_targets if tgt.has_field(JvmResolveField)]	×
34
35	parsed_thrift_sources = await concurrently(	×
36	parse_thrift_file(
37	ParsedThriftRequest(
38	sources_field=tgt[ThriftSourceField],
39	extra_namespace_directives=request.extra_namespace_directives,
40	)
41	)
42	for tgt in jvm_thrift_targets
43	)
44
45	package_symbols: DefaultDict[tuple[_ResolveName, str], OrderedSet[Address]] = defaultdict(	×
46	OrderedSet
47	)
48	lang_ids = {"java", *request.extra_lang_ids}	×
49	for target, parsed_thrift in zip(jvm_thrift_targets, parsed_thrift_sources):	×
50	for lang_id in lang_ids:	×
51	package_name = parsed_thrift.namespaces.get(lang_id)	×
52	if not package_name:	×
53	continue	×
54
55	resolve = target[JvmResolveField].normalized_value(jvm)	×
56	package_symbols[(resolve, package_name)].add(target.address)	×
57
58	mapping: Mapping[str, MutableTrieNode] = defaultdict(MutableTrieNode)	×
59	for (resolve, package_name), addresses in package_symbols.items():	×
60	mapping[resolve].insert(package_name, addresses, first_party=True, recursive=True)	×
61
62	return SymbolMap((resolve, node.frozen()) for resolve, node in mapping.items())	×
63
64
65	def rules():	×
66	return collect_rules()	×

pantsbuild / pants / 18252174847

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous