def test_from_any_pb_success():
in_message = date_pb2.Date(year=1990)
in_message_any = any_pb2.Any()
in_message_any.Pack(in_message)
out_message = protobuf_helpers.from_any_pb(date_pb2.Date, in_message_any)
assert in_message == out_message
def test_success(self):
from google.protobuf import any_pb2
from google.type import date_pb2
in_message = date_pb2.Date(year=1990)
in_message_any = any_pb2.Any()
in_message_any.Pack(in_message)
out_message = self._call_fut(date_pb2.Date, in_message_any)
self.assertEqual(in_message, out_message)
def test_failure(self, ):
from google.protobuf import any_pb2
from google.type import date_pb2
from google.type import timeofday_pb2
in_message = any_pb2.Any()
in_message.Pack(date_pb2.Date(year=1990))
with self.assertRaises(TypeError):
self._call_fut(timeofday_pb2.TimeOfDay, in_message)
def test_from_any_pb_wrapped_success():
# Declare a message class conforming to wrapped messages.
class WrappedDate(object):
def __init__(self, **kwargs):
self._pb = date_pb2.Date(**kwargs)
def __eq__(self, other):
return self._pb == other
@classmethod
def pb(cls, msg):
return msg._pb
# Run the same test as `test_from_any_pb_success`, but using the
# wrapped class.
in_message = date_pb2.Date(year=1990)
in_message_any = any_pb2.Any()
in_message_any.Pack(in_message)
out_message = protobuf_helpers.from_any_pb(WrappedDate, in_message_any)
assert out_message == in_message
def test_from_any_pb_failure():
in_message = any_pb2.Any()
in_message.Pack(date_pb2.Date(year=1990))
with pytest.raises(TypeError):
protobuf_helpers.from_any_pb(timeofday_pb2.TimeOfDay, in_message)
def __init__(self, **kwargs):
self._pb = date_pb2.Date(**kwargs)
