diff --git a/tests/test_all_definition.py b/tests/test_all_definition.py new file mode 100644 index 00000000..61abb5f3 --- /dev/null +++ b/tests/test_all_definition.py @@ -0,0 +1,19 @@ +def test_all_definition(): + """ + Check that a compiled module defines __all__ with the right value. + + These modules have been chosen since they contain messages, services and enums. + """ + import tests.output_betterproto.enum as enum + import tests.output_betterproto.service as service + + assert service.__all__ == ( + "ThingType", + "DoThingRequest", + "DoThingResponse", + "GetThingRequest", + "GetThingResponse", + "TestStub", + "TestBase", + ) + assert enum.__all__ == ("Choice", "ArithmeticOperator", "Test")