add getnewargs_ex to Enum, fix pickling (#663)

This commit is contained in:
benedikt-bartscher
2025-03-30 23:48:58 +02:00
committed by GitHub
parent 6a65ca94bc
commit 08e8a68893
2 changed files with 16 additions and 0 deletions

View File

@@ -57,6 +57,11 @@ class Complex(betterproto.Message):
)
class BetterprotoEnum(betterproto.Enum):
UNSPECIFIED = 0
ONE = 1
def complex_msg():
return Complex(
foo_str="yep",
@@ -201,3 +206,11 @@ def test_message_can_be_cached():
.string_value
== "world"
)
def test_pickle_enum():
enum = BetterprotoEnum.ONE
assert unpickled(enum) == enum
enum = BetterprotoEnum.UNSPECIFIED
assert unpickled(enum) == enum