Skip to content

Commit 4715158

Browse files
anandoleecopybara-github
authored andcommitted
Protobuf usage clean up
PiperOrigin-RevId: 554714188
1 parent da9f1c4 commit 4715158

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

pybind11_protobuf/tests/extension_test.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from absl.testing import absltest
1212
from absl.testing import parameterized
1313

14+
from google.protobuf import unknown_fields
1415
from google.protobuf.internal import api_implementation
1516
from pybind11_protobuf.tests import extension_in_other_file_in_deps_pb2
1617
from pybind11_protobuf.tests import extension_in_other_file_pb2
@@ -179,8 +180,8 @@ def test_reserialize_allow_unknown_inner(self):
179180
a = get_allow_unknown_inner(96)
180181
b = m.reserialize_allow_unknown_inner(a)
181182
if api_implementation.Type() == 'cpp':
182-
self.assertLen(b.UnknownFields(), 1)
183-
self.assertEqual(2001, b.UnknownFields()[0].field_number)
183+
self.assertLen(unknown_fields.UnknownFieldSet(b), 1)
184+
self.assertEqual(2001, unknown_fields.UnknownFieldSet(b)[0].field_number)
184185
else:
185186
b_value = b.Extensions[
186187
extension_in_other_file_pb2.AllowUnknownInnerExtension.hook].value
@@ -190,8 +191,9 @@ def test_reserialize_allow_unknown_outer(self):
190191
a = extension_pb2.AllowUnknownOuter(inner=get_allow_unknown_inner(97))
191192
b = m.reserialize_allow_unknown_outer(a)
192193
if api_implementation.Type() == 'cpp':
193-
self.assertLen(b.inner.UnknownFields(), 1)
194-
self.assertEqual(2001, b.inner.UnknownFields()[0].field_number)
194+
self.assertLen(unknown_fields.UnknownFieldSet(b.inner), 1)
195+
self.assertEqual(
196+
2001, unknown_fields.UnknownFieldSet(b.inner)[0].field_number)
195197
else:
196198
b_inner_value = b.inner.Extensions[
197199
extension_in_other_file_pb2.AllowUnknownInnerExtension.hook].value

0 commit comments

Comments
 (0)