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

66.22

/src/python/pants/backend/codegen/protobuf/python/rules_integration_test.py

# Copyright 2020 Pants project contributors (see CONTRIBUTORS.md).
# Licensed under the Apache License, Version 2.0 (see LICENSE).

from __future__ import annotations

from textwrap import dedent

import pytest

from pants.backend.codegen.protobuf.python import additional_fields
from pants.backend.codegen.protobuf.python.python_protobuf_subsystem import PythonProtobufMypyPlugin
from pants.backend.codegen.protobuf.python.python_protobuf_subsystem import (
    rules as protobuf_subsystem_rules,
)
from pants.backend.codegen.protobuf.python.rules import GeneratePythonFromProtobufRequest
from pants.backend.codegen.protobuf.python.rules import rules as protobuf_rules
from pants.backend.codegen.protobuf.target_types import (
    ProtobufSourceField,
    ProtobufSourcesGeneratorTarget,
)
from pants.backend.codegen.protobuf.target_types import rules as target_types_rules
from pants.backend.python.dependency_inference import module_mapper
from pants.core.util_rules import stripped_source_files
from pants.engine.addresses import Address
from pants.engine.target import GeneratedSources, HydratedSources, HydrateSourcesRequest
from pants.source.source_root import NoSourceRootError
from pants.testutil.python_interpreter_selection import all_major_minor_python_versions
from pants.testutil.rule_runner import QueryRule, RuleRunner, engine_error
from pants.util.resources import read_sibling_resource

GRPC_PROTO_STANZA = """
syntax = "proto3";

package dir1;

// The greeter service definition.
service Greeter {
  // Sends a greeting
  rpc SayHello (HelloRequest) returns (HelloReply) {}
}

// The request message containing the user's name.
message HelloRequest {
  string name = 1;
}

// The response message containing the greetings
message HelloReply {
  string message = 1;
}
"""


@pytest.fixture
def rule_runner() -> RuleRunner:
    return RuleRunner(
        rules=[
            *protobuf_rules(),
            *protobuf_subsystem_rules(),
            *additional_fields.rules(),
            *stripped_source_files.rules(),
            *target_types_rules(),
            *module_mapper.rules(),
            QueryRule(HydratedSources, [HydrateSourcesRequest]),
            QueryRule(GeneratedSources, [GeneratePythonFromProtobufRequest]),
        ],
        target_types=[ProtobufSourcesGeneratorTarget],
    )


def assert_files_generated(
    rule_runner: RuleRunner,
    address: Address,
    *,
    expected_files: list[str],
    source_roots: list[str],
    mypy: bool = False,
    extra_args: list[str] | None = None,
) -> None:
    args = [
        f"--source-root-patterns={repr(source_roots)}",
        "--no-python-protobuf-infer-runtime-dependency",
        *(extra_args or ()),
    ]
    if mypy:
        args.append("--python-protobuf-mypy-plugin")
    rule_runner.set_options(args, env_inherit={"PATH", "PYENV_ROOT", "HOME"})
    tgt = rule_runner.get_target(address)
    protocol_sources = rule_runner.request(
        HydratedSources, [HydrateSourcesRequest(tgt[ProtobufSourceField])]
    )
    generated_sources = rule_runner.request(
        GeneratedSources,
        [GeneratePythonFromProtobufRequest(protocol_sources.snapshot, tgt)],
    )
    assert set(generated_sources.snapshot.files) == set(expected_files)


def test_generates_python(rule_runner: RuleRunner) -> None:
    # This tests a few things:
    #  * We generate the correct file names.
    #  * Protobuf files can import other protobuf files, and those can import others
    #    (transitive dependencies). We'll only generate the requested target, though.
    #  * We can handle multiple source roots, which need to be preserved in the final output.
    rule_runner.write_files(
        {
            "src/protobuf/dir1/f.proto": dedent(
                """\
                syntax = "proto3";

                package dir1;

                message Person {
                  string name = 1;
                  int32 id = 2;
                  string email = 3;
                }
                """
            ),
            "src/protobuf/dir1/f2.proto": dedent(
                """\
                syntax = "proto3";

                package dir1;
                """
            ),
            "src/protobuf/dir1/BUILD": "protobuf_sources()",
            "src/protobuf/dir2/f.proto": dedent(
                """\
                syntax = "proto3";

                package dir2;

                import "dir1/f.proto";
                """
            ),
            "src/protobuf/dir2/BUILD": dedent(
                """\
                protobuf_sources(dependencies=['src/protobuf/dir1'],
                python_source_root='src/python')
                """
            ),
            # Test another source root.
            "tests/protobuf/test_protos/f.proto": dedent(
                """\
                syntax = "proto3";

                package test_protos;

                import "dir2/f.proto";
                """
            ),
            "tests/protobuf/test_protos/BUILD": (
                "protobuf_sources(dependencies=['src/protobuf/dir2'])"
            ),
        }
    )

    def assert_gen(addr: Address, expected: str) -> None:
        assert_files_generated(
            rule_runner,
            addr,
            source_roots=["src/python", "/src/protobuf", "/tests/protobuf"],
            expected_files=[expected],
        )

    assert_gen(
        Address("src/protobuf/dir1", relative_file_path="f.proto"), "src/protobuf/dir1/f_pb2.py"
    )
    assert_gen(
        Address("src/protobuf/dir1", relative_file_path="f2.proto"), "src/protobuf/dir1/f2_pb2.py"
    )
    assert_gen(
        Address("src/protobuf/dir2", relative_file_path="f.proto"), "src/python/dir2/f_pb2.py"
    )
    assert_gen(
        Address("tests/protobuf/test_protos", relative_file_path="f.proto"),
        "tests/protobuf/test_protos/f_pb2.py",
    )


def test_top_level_proto_root(rule_runner: RuleRunner) -> None:
    rule_runner.write_files(
        {
            "protos/f.proto": dedent(
                """\
                syntax = "proto3";

                package protos;
                """
            ),
            "protos/BUILD": "protobuf_sources()",
        }
    )
    assert_files_generated(
        rule_runner,
        Address("protos", relative_file_path="f.proto"),
        source_roots=["/"],
        expected_files=["protos/f_pb2.py"],
    )


def test_top_level_python_source_root(rule_runner: RuleRunner) -> None:
    rule_runner.write_files(
        {
            "src/proto/protos/f.proto": dedent(
                """\
                syntax = "proto3";

                package protos;
                """
            ),
            "src/proto/protos/BUILD": "protobuf_sources(python_source_root='.')",
        }
    )
    assert_files_generated(
        rule_runner,
        Address("src/proto/protos", relative_file_path="f.proto"),
        source_roots=["/", "src/proto"],
        expected_files=["protos/f_pb2.py"],
    )


def test_bad_python_source_root(rule_runner: RuleRunner) -> None:
    rule_runner.write_files(
        {
            "src/protobuf/dir1/f.proto": dedent(
                """\
                syntax = "proto3";

                package dir1;
                """
            ),
            "src/protobuf/dir1/BUILD": "protobuf_sources(python_source_root='notasourceroot')",
        }
    )
    with engine_error(NoSourceRootError):
        assert_files_generated(
            rule_runner,
            Address("src/protobuf/dir1", relative_file_path="f.proto"),
            source_roots=["src/protobuf"],
            expected_files=[],
        )


@pytest.mark.platform_specific_behavior
@pytest.mark.parametrize(
    "major_minor_interpreter",
    all_major_minor_python_versions(PythonProtobufMypyPlugin.default_interpreter_constraints),
)
def test_generate_type_stubs(rule_runner: RuleRunner, major_minor_interpreter: str) -> None:
    rule_runner.write_files(
        {
            "src/protobuf/dir1/f.proto": dedent(
                """\
                syntax = "proto3";

                package dir1;

                message Person {
                  string name = 1;
                  int32 id = 2;
                  string email = 3;
                }
                """
            ),
            "src/protobuf/dir1/BUILD": "protobuf_sources()",
        }
    )
    assert_files_generated(
        rule_runner,
        Address("src/protobuf/dir1", relative_file_path="f.proto"),
        source_roots=["src/protobuf"],
        extra_args=[
            "--python-protobuf-generate-type-stubs",
            f"--mypy-protobuf-interpreter-constraints=['=={major_minor_interpreter}.*']",
        ],
        expected_files=["src/protobuf/dir1/f_pb2.py", "src/protobuf/dir1/f_pb2.pyi"],
    )


@pytest.mark.platform_specific_behavior
@pytest.mark.parametrize(
    "major_minor_interpreter",
    all_major_minor_python_versions(PythonProtobufMypyPlugin.default_interpreter_constraints),
)
def test_mypy_plugin(rule_runner: RuleRunner, major_minor_interpreter: str) -> None:
    rule_runner.write_files(
        {
            "src/protobuf/dir1/f.proto": dedent(
                """\
                syntax = "proto3";

                package dir1;

                message Person {
                  string name = 1;
                  int32 id = 2;
                  string email = 3;
                }
                """
            ),
            "src/protobuf/dir1/BUILD": "protobuf_sources()",
        }
    )
    assert_files_generated(
        rule_runner,
        Address("src/protobuf/dir1", relative_file_path="f.proto"),
        source_roots=["src/protobuf"],
        extra_args=[
            "--python-protobuf-mypy-plugin",
            f"--mypy-protobuf-interpreter-constraints=['=={major_minor_interpreter}.*']",
        ],
        expected_files=["src/protobuf/dir1/f_pb2.py", "src/protobuf/dir1/f_pb2.pyi"],
    )


def test_grpc(rule_runner: RuleRunner) -> None:
    rule_runner.write_files(
        {
            "src/protobuf/dir1/f.proto": dedent(GRPC_PROTO_STANZA),
            "src/protobuf/dir1/BUILD": "protobuf_sources(grpc=True)",
        }
    )
    assert_files_generated(
        rule_runner,
        Address("src/protobuf/dir1", relative_file_path="f.proto"),
        source_roots=["src/protobuf"],
        expected_files=["src/protobuf/dir1/f_pb2.py", "src/protobuf/dir1/f_pb2_grpc.py"],
    )


def test_grpc_mypy_plugin(rule_runner: RuleRunner) -> None:
    rule_runner.write_files(
        {
            "src/protobuf/dir1/f.proto": dedent(GRPC_PROTO_STANZA),
            "src/protobuf/dir1/BUILD": "protobuf_sources(grpc=True)",
        }
    )
    assert_files_generated(
        rule_runner,
        Address("src/protobuf/dir1", relative_file_path="f.proto"),
        source_roots=["src/protobuf"],
        mypy=True,
        expected_files=[
            "src/protobuf/dir1/f_pb2.py",
            "src/protobuf/dir1/f_pb2.pyi",
            "src/protobuf/dir1/f_pb2_grpc.py",
            "src/protobuf/dir1/f_pb2_grpc.pyi",
        ],
    )


def test_grpc_pre_v2_mypy_plugin(rule_runner: RuleRunner) -> None:
    rule_runner.write_files(
        {
            "src/protobuf/dir1/f.proto": dedent(GRPC_PROTO_STANZA),
            "src/protobuf/dir1/BUILD": "protobuf_sources(grpc=True)",
            "mypy-protobuf.lock": read_sibling_resource(
                __name__, "test_grpc_pre_v2_mypy_plugin.lock"
            ),
        }
    )
    assert_files_generated(
        rule_runner,
        Address("src/protobuf/dir1", relative_file_path="f.proto"),
        source_roots=["src/protobuf"],
        extra_args=[
            "--python-protobuf-mypy-plugin",
            "--python-resolves={'mypy-protobuf':'mypy-protobuf.lock'}",
            "--mypy-protobuf-install-from-resolve=mypy-protobuf",
        ],
        expected_files=[
            "src/protobuf/dir1/f_pb2.py",
            "src/protobuf/dir1/f_pb2.pyi",
            "src/protobuf/dir1/f_pb2_grpc.py",
        ],
    )


def test_grpclib_plugin(rule_runner: RuleRunner) -> None:
    rule_runner.write_files(
        {
            "src/protobuf/dir1/f.proto": dedent(GRPC_PROTO_STANZA),
            "src/protobuf/dir1/BUILD": "protobuf_sources(grpc=True)",
        }
    )
    assert_files_generated(
        rule_runner,
        Address("src/protobuf/dir1", relative_file_path="f.proto"),
        source_roots=["src/protobuf"],
        extra_args=[
            "--python-protobuf-grpclib-plugin",
            "--no-python-protobuf-grpcio-plugin",
        ],
        expected_files=[
            "src/protobuf/dir1/f_pb2.py",
            "src/protobuf/dir1/f_grpc.py",
        ],
    )


def test_all_plugins(rule_runner: RuleRunner) -> None:
    rule_runner.write_files(
        {
            "src/protobuf/dir1/f.proto": dedent(GRPC_PROTO_STANZA),
            "src/protobuf/dir1/BUILD": "protobuf_sources(grpc=True)",
        }
    )
    assert_files_generated(
        rule_runner,
        Address("src/protobuf/dir1", relative_file_path="f.proto"),
        source_roots=["src/protobuf"],
        extra_args=[
            "--python-protobuf-grpclib-plugin",
            "--python-protobuf-grpcio-plugin",
            "--python-protobuf-mypy-plugin",
        ],
        expected_files=[
            "src/protobuf/dir1/f_pb2.py",
            "src/protobuf/dir1/f_pb2.pyi",
            "src/protobuf/dir1/f_pb2_grpc.py",
            "src/protobuf/dir1/f_pb2_grpc.pyi",
            "src/protobuf/dir1/f_grpc.py",
        ],
    )

1	# Copyright 2020 Pants project contributors (see CONTRIBUTORS.md).
2	# Licensed under the Apache License, Version 2.0 (see LICENSE).
3
4	from __future__ import annotations	3✔
5
6	from textwrap import dedent	3✔
7
8	import pytest	3✔
9
10	from pants.backend.codegen.protobuf.python import additional_fields	3✔
11	from pants.backend.codegen.protobuf.python.python_protobuf_subsystem import PythonProtobufMypyPlugin	3✔
12	from pants.backend.codegen.protobuf.python.python_protobuf_subsystem import (	3✔
13	rules as protobuf_subsystem_rules,
14	)
15	from pants.backend.codegen.protobuf.python.rules import GeneratePythonFromProtobufRequest	3✔
16	from pants.backend.codegen.protobuf.python.rules import rules as protobuf_rules	3✔
17	from pants.backend.codegen.protobuf.target_types import (	3✔
18	ProtobufSourceField,
19	ProtobufSourcesGeneratorTarget,
20	)
21	from pants.backend.codegen.protobuf.target_types import rules as target_types_rules	3✔
22	from pants.backend.python.dependency_inference import module_mapper	3✔
23	from pants.core.util_rules import stripped_source_files	3✔
24	from pants.engine.addresses import Address	3✔
25	from pants.engine.target import GeneratedSources, HydratedSources, HydrateSourcesRequest	3✔
26	from pants.source.source_root import NoSourceRootError	3✔
27	from pants.testutil.python_interpreter_selection import all_major_minor_python_versions	3✔
28	from pants.testutil.rule_runner import QueryRule, RuleRunner, engine_error	3✔
29	from pants.util.resources import read_sibling_resource	3✔
30
31	GRPC_PROTO_STANZA = """	3✔
32	syntax = "proto3";
33
34	package dir1;
35
36	// The greeter service definition.
37	service Greeter {
38	// Sends a greeting
39	rpc SayHello (HelloRequest) returns (HelloReply) {}
40	}
41
42	// The request message containing the user's name.
43	message HelloRequest {
44	string name = 1;
45	}
46
47	// The response message containing the greetings
48	message HelloReply {
49	string message = 1;
50	}
51	"""
52
53
54	@pytest.fixture	3✔
55	def rule_runner() -> RuleRunner:	3✔
56	return RuleRunner(	3✔
57	rules=[
58	*protobuf_rules(),
59	*protobuf_subsystem_rules(),
60	*additional_fields.rules(),
61	*stripped_source_files.rules(),
62	*target_types_rules(),
63	*module_mapper.rules(),
64	QueryRule(HydratedSources, [HydrateSourcesRequest]),
65	QueryRule(GeneratedSources, [GeneratePythonFromProtobufRequest]),
66	],
67	target_types=[ProtobufSourcesGeneratorTarget],
68	)
69
70
71	def assert_files_generated(	3✔
72	rule_runner: RuleRunner,
73	address: Address,
74	*,
75	expected_files: list[str],
76	source_roots: list[str],
77	mypy: bool = False,
78	extra_args: list[str] \| None = None,
79	) -> None:
80	args = [	3✔
81	f"--source-root-patterns={repr(source_roots)}",
82	"--no-python-protobuf-infer-runtime-dependency",
83	*(extra_args or ()),
84	]
85	if mypy:	3✔
86	args.append("--python-protobuf-mypy-plugin")	×
87	rule_runner.set_options(args, env_inherit={"PATH", "PYENV_ROOT", "HOME"})	3✔
88	tgt = rule_runner.get_target(address)	3✔
89	protocol_sources = rule_runner.request(	3✔
90	HydratedSources, [HydrateSourcesRequest(tgt[ProtobufSourceField])]
91	)
92	generated_sources = rule_runner.request(	3✔
93	GeneratedSources,
94	[GeneratePythonFromProtobufRequest(protocol_sources.snapshot, tgt)],
95	)
96	assert set(generated_sources.snapshot.files) == set(expected_files)	3✔
97
98
99	def test_generates_python(rule_runner: RuleRunner) -> None:	3✔
100	# This tests a few things:
101	# * We generate the correct file names.
102	# * Protobuf files can import other protobuf files, and those can import others
103	# (transitive dependencies). We'll only generate the requested target, though.
104	# * We can handle multiple source roots, which need to be preserved in the final output.
105	rule_runner.write_files(	×
106	{
107	"src/protobuf/dir1/f.proto": dedent(
108	"""\
109	syntax = "proto3";
110
111	package dir1;
112
113	message Person {
114	string name = 1;
115	int32 id = 2;
116	string email = 3;
117	}
118	"""
119	),
120	"src/protobuf/dir1/f2.proto": dedent(
121	"""\
122	syntax = "proto3";
123
124	package dir1;
125	"""
126	),
127	"src/protobuf/dir1/BUILD": "protobuf_sources()",
128	"src/protobuf/dir2/f.proto": dedent(
129	"""\
130	syntax = "proto3";
131
132	package dir2;
133
134	import "dir1/f.proto";
135	"""
136	),
137	"src/protobuf/dir2/BUILD": dedent(
138	"""\
139	protobuf_sources(dependencies=['src/protobuf/dir1'],
140	python_source_root='src/python')
141	"""
142	),
143	# Test another source root.
144	"tests/protobuf/test_protos/f.proto": dedent(
145	"""\
146	syntax = "proto3";
147
148	package test_protos;
149
150	import "dir2/f.proto";
151	"""
152	),
153	"tests/protobuf/test_protos/BUILD": (
154	"protobuf_sources(dependencies=['src/protobuf/dir2'])"
155	),
156	}
157	)
158
159	def assert_gen(addr: Address, expected: str) -> None:	×
160	assert_files_generated(	×
161	rule_runner,
162	addr,
163	source_roots=["src/python", "/src/protobuf", "/tests/protobuf"],
164	expected_files=[expected],
165	)
166
167	assert_gen(	×
168	Address("src/protobuf/dir1", relative_file_path="f.proto"), "src/protobuf/dir1/f_pb2.py"
169	)
170	assert_gen(	×
171	Address("src/protobuf/dir1", relative_file_path="f2.proto"), "src/protobuf/dir1/f2_pb2.py"
172	)
173	assert_gen(	×
174	Address("src/protobuf/dir2", relative_file_path="f.proto"), "src/python/dir2/f_pb2.py"
175	)
176	assert_gen(	×
177	Address("tests/protobuf/test_protos", relative_file_path="f.proto"),
178	"tests/protobuf/test_protos/f_pb2.py",
179	)
180
181
182	def test_top_level_proto_root(rule_runner: RuleRunner) -> None:	3✔
183	rule_runner.write_files(	×
184	{
185	"protos/f.proto": dedent(
186	"""\
187	syntax = "proto3";
188
189	package protos;
190	"""
191	),
192	"protos/BUILD": "protobuf_sources()",
193	}
194	)
195	assert_files_generated(	×
196	rule_runner,
197	Address("protos", relative_file_path="f.proto"),
198	source_roots=["/"],
199	expected_files=["protos/f_pb2.py"],
200	)
201
202
203	def test_top_level_python_source_root(rule_runner: RuleRunner) -> None:	3✔
204	rule_runner.write_files(	×
205	{
206	"src/proto/protos/f.proto": dedent(
207	"""\
208	syntax = "proto3";
209
210	package protos;
211	"""
212	),
213	"src/proto/protos/BUILD": "protobuf_sources(python_source_root='.')",
214	}
215	)
216	assert_files_generated(	×
217	rule_runner,
218	Address("src/proto/protos", relative_file_path="f.proto"),
219	source_roots=["/", "src/proto"],
220	expected_files=["protos/f_pb2.py"],
221	)
222
223
224	def test_bad_python_source_root(rule_runner: RuleRunner) -> None:	3✔
225	rule_runner.write_files(	×
226	{
227	"src/protobuf/dir1/f.proto": dedent(
228	"""\
229	syntax = "proto3";
230
231	package dir1;
232	"""
233	),
234	"src/protobuf/dir1/BUILD": "protobuf_sources(python_source_root='notasourceroot')",
235	}
236	)
237	with engine_error(NoSourceRootError):	×
238	assert_files_generated(	×
239	rule_runner,
240	Address("src/protobuf/dir1", relative_file_path="f.proto"),
241	source_roots=["src/protobuf"],
242	expected_files=[],
243	)
244
245
246	@pytest.mark.platform_specific_behavior	3✔
247	@pytest.mark.parametrize(	3✔
248	"major_minor_interpreter",
249	all_major_minor_python_versions(PythonProtobufMypyPlugin.default_interpreter_constraints),
250	)
251	def test_generate_type_stubs(rule_runner: RuleRunner, major_minor_interpreter: str) -> None:	3✔
252	rule_runner.write_files(	3✔
253	{
254	"src/protobuf/dir1/f.proto": dedent(
255	"""\
256	syntax = "proto3";
257
258	package dir1;
259
260	message Person {
261	string name = 1;
262	int32 id = 2;
263	string email = 3;
264	}
265	"""
266	),
267	"src/protobuf/dir1/BUILD": "protobuf_sources()",
268	}
269	)
270	assert_files_generated(	3✔
271	rule_runner,
272	Address("src/protobuf/dir1", relative_file_path="f.proto"),
273	source_roots=["src/protobuf"],
274	extra_args=[
275	"--python-protobuf-generate-type-stubs",
276	f"--mypy-protobuf-interpreter-constraints=['=={major_minor_interpreter}.*']",
277	],
278	expected_files=["src/protobuf/dir1/f_pb2.py", "src/protobuf/dir1/f_pb2.pyi"],
279	)
280
281
282	@pytest.mark.platform_specific_behavior	3✔
283	@pytest.mark.parametrize(	3✔
284	"major_minor_interpreter",
285	all_major_minor_python_versions(PythonProtobufMypyPlugin.default_interpreter_constraints),
286	)
287	def test_mypy_plugin(rule_runner: RuleRunner, major_minor_interpreter: str) -> None:	3✔
288	rule_runner.write_files(	3✔
289	{
290	"src/protobuf/dir1/f.proto": dedent(
291	"""\
292	syntax = "proto3";
293
294	package dir1;
295
296	message Person {
297	string name = 1;
298	int32 id = 2;
299	string email = 3;
300	}
301	"""
302	),
303	"src/protobuf/dir1/BUILD": "protobuf_sources()",
304	}
305	)
306	assert_files_generated(	3✔
307	rule_runner,
308	Address("src/protobuf/dir1", relative_file_path="f.proto"),
309	source_roots=["src/protobuf"],
310	extra_args=[
311	"--python-protobuf-mypy-plugin",
312	f"--mypy-protobuf-interpreter-constraints=['=={major_minor_interpreter}.*']",
313	],
314	expected_files=["src/protobuf/dir1/f_pb2.py", "src/protobuf/dir1/f_pb2.pyi"],
315	)
316
317
318	def test_grpc(rule_runner: RuleRunner) -> None:	3✔
319	rule_runner.write_files(	×
320	{
321	"src/protobuf/dir1/f.proto": dedent(GRPC_PROTO_STANZA),
322	"src/protobuf/dir1/BUILD": "protobuf_sources(grpc=True)",
323	}
324	)
325	assert_files_generated(	×
326	rule_runner,
327	Address("src/protobuf/dir1", relative_file_path="f.proto"),
328	source_roots=["src/protobuf"],
329	expected_files=["src/protobuf/dir1/f_pb2.py", "src/protobuf/dir1/f_pb2_grpc.py"],
330	)
331
332
333	def test_grpc_mypy_plugin(rule_runner: RuleRunner) -> None:	3✔
334	rule_runner.write_files(	×
335	{
336	"src/protobuf/dir1/f.proto": dedent(GRPC_PROTO_STANZA),
337	"src/protobuf/dir1/BUILD": "protobuf_sources(grpc=True)",
338	}
339	)
340	assert_files_generated(	×
341	rule_runner,
342	Address("src/protobuf/dir1", relative_file_path="f.proto"),
343	source_roots=["src/protobuf"],
344	mypy=True,
345	expected_files=[
346	"src/protobuf/dir1/f_pb2.py",
347	"src/protobuf/dir1/f_pb2.pyi",
348	"src/protobuf/dir1/f_pb2_grpc.py",
349	"src/protobuf/dir1/f_pb2_grpc.pyi",
350	],
351	)
352
353
354	def test_grpc_pre_v2_mypy_plugin(rule_runner: RuleRunner) -> None:	3✔
355	rule_runner.write_files(	×
356	{
357	"src/protobuf/dir1/f.proto": dedent(GRPC_PROTO_STANZA),
358	"src/protobuf/dir1/BUILD": "protobuf_sources(grpc=True)",
359	"mypy-protobuf.lock": read_sibling_resource(
360	__name__, "test_grpc_pre_v2_mypy_plugin.lock"
361	),
362	}
363	)
364	assert_files_generated(	×
365	rule_runner,
366	Address("src/protobuf/dir1", relative_file_path="f.proto"),
367	source_roots=["src/protobuf"],
368	extra_args=[
369	"--python-protobuf-mypy-plugin",
370	"--python-resolves={'mypy-protobuf':'mypy-protobuf.lock'}",
371	"--mypy-protobuf-install-from-resolve=mypy-protobuf",
372	],
373	expected_files=[
374	"src/protobuf/dir1/f_pb2.py",
375	"src/protobuf/dir1/f_pb2.pyi",
376	"src/protobuf/dir1/f_pb2_grpc.py",
377	],
378	)
379
380
381	def test_grpclib_plugin(rule_runner: RuleRunner) -> None:	3✔
382	rule_runner.write_files(	×
383	{
384	"src/protobuf/dir1/f.proto": dedent(GRPC_PROTO_STANZA),
385	"src/protobuf/dir1/BUILD": "protobuf_sources(grpc=True)",
386	}
387	)
388	assert_files_generated(	×
389	rule_runner,
390	Address("src/protobuf/dir1", relative_file_path="f.proto"),
391	source_roots=["src/protobuf"],
392	extra_args=[
393	"--python-protobuf-grpclib-plugin",
394	"--no-python-protobuf-grpcio-plugin",
395	],
396	expected_files=[
397	"src/protobuf/dir1/f_pb2.py",
398	"src/protobuf/dir1/f_grpc.py",
399	],
400	)
401
402
403	def test_all_plugins(rule_runner: RuleRunner) -> None:	3✔
404	rule_runner.write_files(	×
405	{
406	"src/protobuf/dir1/f.proto": dedent(GRPC_PROTO_STANZA),
407	"src/protobuf/dir1/BUILD": "protobuf_sources(grpc=True)",
408	}
409	)
410	assert_files_generated(	×
411	rule_runner,
412	Address("src/protobuf/dir1", relative_file_path="f.proto"),
413	source_roots=["src/protobuf"],
414	extra_args=[
415	"--python-protobuf-grpclib-plugin",
416	"--python-protobuf-grpcio-plugin",
417	"--python-protobuf-mypy-plugin",
418	],
419	expected_files=[
420	"src/protobuf/dir1/f_pb2.py",
421	"src/protobuf/dir1/f_pb2.pyi",
422	"src/protobuf/dir1/f_pb2_grpc.py",
423	"src/protobuf/dir1/f_pb2_grpc.pyi",
424	"src/protobuf/dir1/f_grpc.py",
425	],
426	)

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