/thirdparty/breakpad/third_party/protobuf/protobuf/python/google/protobuf/internal/test_util.py

http://github.com/tomahawk-player/tomahawk · Python · 635 lines · 451 code · 84 blank · 100 comment · 3 complexity · 6383ba2391045639b16ecd70c31f2bee MD5 · raw file

  1. # Protocol Buffers - Google's data interchange format
  2. # Copyright 2008 Google Inc. All rights reserved.
  3. # http://code.google.com/p/protobuf/
  4. #
  5. # Redistribution and use in source and binary forms, with or without
  6. # modification, are permitted provided that the following conditions are
  7. # met:
  8. #
  9. # * Redistributions of source code must retain the above copyright
  10. # notice, this list of conditions and the following disclaimer.
  11. # * Redistributions in binary form must reproduce the above
  12. # copyright notice, this list of conditions and the following disclaimer
  13. # in the documentation and/or other materials provided with the
  14. # distribution.
  15. # * Neither the name of Google Inc. nor the names of its
  16. # contributors may be used to endorse or promote products derived from
  17. # this software without specific prior written permission.
  18. #
  19. # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  20. # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  21. # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  22. # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  23. # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24. # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  25. # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  26. # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  27. # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  28. # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  29. # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  30. """Utilities for Python proto2 tests.
  31. This is intentionally modeled on C++ code in
  32. //google/protobuf/test_util.*.
  33. """
  34. __author__ = 'robinson@google.com (Will Robinson)'
  35. import os.path
  36. from google.protobuf import unittest_import_pb2
  37. from google.protobuf import unittest_pb2
  38. def SetAllFields(message):
  39. """Sets every field in the message to a unique value.
  40. Args:
  41. message: A unittest_pb2.TestAllTypes instance.
  42. """
  43. #
  44. # Optional fields.
  45. #
  46. message.optional_int32 = 101
  47. message.optional_int64 = 102
  48. message.optional_uint32 = 103
  49. message.optional_uint64 = 104
  50. message.optional_sint32 = 105
  51. message.optional_sint64 = 106
  52. message.optional_fixed32 = 107
  53. message.optional_fixed64 = 108
  54. message.optional_sfixed32 = 109
  55. message.optional_sfixed64 = 110
  56. message.optional_float = 111
  57. message.optional_double = 112
  58. message.optional_bool = True
  59. # TODO(robinson): Firmly spec out and test how
  60. # protos interact with unicode. One specific example:
  61. # what happens if we change the literal below to
  62. # u'115'? What *should* happen? Still some discussion
  63. # to finish with Kenton about bytes vs. strings
  64. # and forcing everything to be utf8. :-/
  65. message.optional_string = '115'
  66. message.optional_bytes = '116'
  67. message.optionalgroup.a = 117
  68. message.optional_nested_message.bb = 118
  69. message.optional_foreign_message.c = 119
  70. message.optional_import_message.d = 120
  71. message.optional_nested_enum = unittest_pb2.TestAllTypes.BAZ
  72. message.optional_foreign_enum = unittest_pb2.FOREIGN_BAZ
  73. message.optional_import_enum = unittest_import_pb2.IMPORT_BAZ
  74. message.optional_string_piece = '124'
  75. message.optional_cord = '125'
  76. #
  77. # Repeated fields.
  78. #
  79. message.repeated_int32.append(201)
  80. message.repeated_int64.append(202)
  81. message.repeated_uint32.append(203)
  82. message.repeated_uint64.append(204)
  83. message.repeated_sint32.append(205)
  84. message.repeated_sint64.append(206)
  85. message.repeated_fixed32.append(207)
  86. message.repeated_fixed64.append(208)
  87. message.repeated_sfixed32.append(209)
  88. message.repeated_sfixed64.append(210)
  89. message.repeated_float.append(211)
  90. message.repeated_double.append(212)
  91. message.repeated_bool.append(True)
  92. message.repeated_string.append('215')
  93. message.repeated_bytes.append('216')
  94. message.repeatedgroup.add().a = 217
  95. message.repeated_nested_message.add().bb = 218
  96. message.repeated_foreign_message.add().c = 219
  97. message.repeated_import_message.add().d = 220
  98. message.repeated_nested_enum.append(unittest_pb2.TestAllTypes.BAR)
  99. message.repeated_foreign_enum.append(unittest_pb2.FOREIGN_BAR)
  100. message.repeated_import_enum.append(unittest_import_pb2.IMPORT_BAR)
  101. message.repeated_string_piece.append('224')
  102. message.repeated_cord.append('225')
  103. # Add a second one of each field.
  104. message.repeated_int32.append(301)
  105. message.repeated_int64.append(302)
  106. message.repeated_uint32.append(303)
  107. message.repeated_uint64.append(304)
  108. message.repeated_sint32.append(305)
  109. message.repeated_sint64.append(306)
  110. message.repeated_fixed32.append(307)
  111. message.repeated_fixed64.append(308)
  112. message.repeated_sfixed32.append(309)
  113. message.repeated_sfixed64.append(310)
  114. message.repeated_float.append(311)
  115. message.repeated_double.append(312)
  116. message.repeated_bool.append(False)
  117. message.repeated_string.append('315')
  118. message.repeated_bytes.append('316')
  119. message.repeatedgroup.add().a = 317
  120. message.repeated_nested_message.add().bb = 318
  121. message.repeated_foreign_message.add().c = 319
  122. message.repeated_import_message.add().d = 320
  123. message.repeated_nested_enum.append(unittest_pb2.TestAllTypes.BAZ)
  124. message.repeated_foreign_enum.append(unittest_pb2.FOREIGN_BAZ)
  125. message.repeated_import_enum.append(unittest_import_pb2.IMPORT_BAZ)
  126. message.repeated_string_piece.append('324')
  127. message.repeated_cord.append('325')
  128. #
  129. # Fields that have defaults.
  130. #
  131. message.default_int32 = 401
  132. message.default_int64 = 402
  133. message.default_uint32 = 403
  134. message.default_uint64 = 404
  135. message.default_sint32 = 405
  136. message.default_sint64 = 406
  137. message.default_fixed32 = 407
  138. message.default_fixed64 = 408
  139. message.default_sfixed32 = 409
  140. message.default_sfixed64 = 410
  141. message.default_float = 411
  142. message.default_double = 412
  143. message.default_bool = False
  144. message.default_string = '415'
  145. message.default_bytes = '416'
  146. message.default_nested_enum = unittest_pb2.TestAllTypes.FOO
  147. message.default_foreign_enum = unittest_pb2.FOREIGN_FOO
  148. message.default_import_enum = unittest_import_pb2.IMPORT_FOO
  149. message.default_string_piece = '424'
  150. message.default_cord = '425'
  151. def SetAllExtensions(message):
  152. """Sets every extension in the message to a unique value.
  153. Args:
  154. message: A unittest_pb2.TestAllExtensions instance.
  155. """
  156. extensions = message.Extensions
  157. pb2 = unittest_pb2
  158. import_pb2 = unittest_import_pb2
  159. #
  160. # Optional fields.
  161. #
  162. extensions[pb2.optional_int32_extension] = 101
  163. extensions[pb2.optional_int64_extension] = 102
  164. extensions[pb2.optional_uint32_extension] = 103
  165. extensions[pb2.optional_uint64_extension] = 104
  166. extensions[pb2.optional_sint32_extension] = 105
  167. extensions[pb2.optional_sint64_extension] = 106
  168. extensions[pb2.optional_fixed32_extension] = 107
  169. extensions[pb2.optional_fixed64_extension] = 108
  170. extensions[pb2.optional_sfixed32_extension] = 109
  171. extensions[pb2.optional_sfixed64_extension] = 110
  172. extensions[pb2.optional_float_extension] = 111
  173. extensions[pb2.optional_double_extension] = 112
  174. extensions[pb2.optional_bool_extension] = True
  175. extensions[pb2.optional_string_extension] = '115'
  176. extensions[pb2.optional_bytes_extension] = '116'
  177. extensions[pb2.optionalgroup_extension].a = 117
  178. extensions[pb2.optional_nested_message_extension].bb = 118
  179. extensions[pb2.optional_foreign_message_extension].c = 119
  180. extensions[pb2.optional_import_message_extension].d = 120
  181. extensions[pb2.optional_nested_enum_extension] = pb2.TestAllTypes.BAZ
  182. extensions[pb2.optional_nested_enum_extension] = pb2.TestAllTypes.BAZ
  183. extensions[pb2.optional_foreign_enum_extension] = pb2.FOREIGN_BAZ
  184. extensions[pb2.optional_import_enum_extension] = import_pb2.IMPORT_BAZ
  185. extensions[pb2.optional_string_piece_extension] = '124'
  186. extensions[pb2.optional_cord_extension] = '125'
  187. #
  188. # Repeated fields.
  189. #
  190. extensions[pb2.repeated_int32_extension].append(201)
  191. extensions[pb2.repeated_int64_extension].append(202)
  192. extensions[pb2.repeated_uint32_extension].append(203)
  193. extensions[pb2.repeated_uint64_extension].append(204)
  194. extensions[pb2.repeated_sint32_extension].append(205)
  195. extensions[pb2.repeated_sint64_extension].append(206)
  196. extensions[pb2.repeated_fixed32_extension].append(207)
  197. extensions[pb2.repeated_fixed64_extension].append(208)
  198. extensions[pb2.repeated_sfixed32_extension].append(209)
  199. extensions[pb2.repeated_sfixed64_extension].append(210)
  200. extensions[pb2.repeated_float_extension].append(211)
  201. extensions[pb2.repeated_double_extension].append(212)
  202. extensions[pb2.repeated_bool_extension].append(True)
  203. extensions[pb2.repeated_string_extension].append('215')
  204. extensions[pb2.repeated_bytes_extension].append('216')
  205. extensions[pb2.repeatedgroup_extension].add().a = 217
  206. extensions[pb2.repeated_nested_message_extension].add().bb = 218
  207. extensions[pb2.repeated_foreign_message_extension].add().c = 219
  208. extensions[pb2.repeated_import_message_extension].add().d = 220
  209. extensions[pb2.repeated_nested_enum_extension].append(pb2.TestAllTypes.BAR)
  210. extensions[pb2.repeated_foreign_enum_extension].append(pb2.FOREIGN_BAR)
  211. extensions[pb2.repeated_import_enum_extension].append(import_pb2.IMPORT_BAR)
  212. extensions[pb2.repeated_string_piece_extension].append('224')
  213. extensions[pb2.repeated_cord_extension].append('225')
  214. # Append a second one of each field.
  215. extensions[pb2.repeated_int32_extension].append(301)
  216. extensions[pb2.repeated_int64_extension].append(302)
  217. extensions[pb2.repeated_uint32_extension].append(303)
  218. extensions[pb2.repeated_uint64_extension].append(304)
  219. extensions[pb2.repeated_sint32_extension].append(305)
  220. extensions[pb2.repeated_sint64_extension].append(306)
  221. extensions[pb2.repeated_fixed32_extension].append(307)
  222. extensions[pb2.repeated_fixed64_extension].append(308)
  223. extensions[pb2.repeated_sfixed32_extension].append(309)
  224. extensions[pb2.repeated_sfixed64_extension].append(310)
  225. extensions[pb2.repeated_float_extension].append(311)
  226. extensions[pb2.repeated_double_extension].append(312)
  227. extensions[pb2.repeated_bool_extension].append(False)
  228. extensions[pb2.repeated_string_extension].append('315')
  229. extensions[pb2.repeated_bytes_extension].append('316')
  230. extensions[pb2.repeatedgroup_extension].add().a = 317
  231. extensions[pb2.repeated_nested_message_extension].add().bb = 318
  232. extensions[pb2.repeated_foreign_message_extension].add().c = 319
  233. extensions[pb2.repeated_import_message_extension].add().d = 320
  234. extensions[pb2.repeated_nested_enum_extension].append(pb2.TestAllTypes.BAZ)
  235. extensions[pb2.repeated_foreign_enum_extension].append(pb2.FOREIGN_BAZ)
  236. extensions[pb2.repeated_import_enum_extension].append(import_pb2.IMPORT_BAZ)
  237. extensions[pb2.repeated_string_piece_extension].append('324')
  238. extensions[pb2.repeated_cord_extension].append('325')
  239. #
  240. # Fields with defaults.
  241. #
  242. extensions[pb2.default_int32_extension] = 401
  243. extensions[pb2.default_int64_extension] = 402
  244. extensions[pb2.default_uint32_extension] = 403
  245. extensions[pb2.default_uint64_extension] = 404
  246. extensions[pb2.default_sint32_extension] = 405
  247. extensions[pb2.default_sint64_extension] = 406
  248. extensions[pb2.default_fixed32_extension] = 407
  249. extensions[pb2.default_fixed64_extension] = 408
  250. extensions[pb2.default_sfixed32_extension] = 409
  251. extensions[pb2.default_sfixed64_extension] = 410
  252. extensions[pb2.default_float_extension] = 411
  253. extensions[pb2.default_double_extension] = 412
  254. extensions[pb2.default_bool_extension] = False
  255. extensions[pb2.default_string_extension] = '415'
  256. extensions[pb2.default_bytes_extension] = '416'
  257. extensions[pb2.default_nested_enum_extension] = pb2.TestAllTypes.FOO
  258. extensions[pb2.default_foreign_enum_extension] = pb2.FOREIGN_FOO
  259. extensions[pb2.default_import_enum_extension] = import_pb2.IMPORT_FOO
  260. extensions[pb2.default_string_piece_extension] = '424'
  261. extensions[pb2.default_cord_extension] = '425'
  262. def SetAllFieldsAndExtensions(message):
  263. """Sets every field and extension in the message to a unique value.
  264. Args:
  265. message: A unittest_pb2.TestAllExtensions message.
  266. """
  267. message.my_int = 1
  268. message.my_string = 'foo'
  269. message.my_float = 1.0
  270. message.Extensions[unittest_pb2.my_extension_int] = 23
  271. message.Extensions[unittest_pb2.my_extension_string] = 'bar'
  272. def ExpectAllFieldsAndExtensionsInOrder(serialized):
  273. """Ensures that serialized is the serialization we expect for a message
  274. filled with SetAllFieldsAndExtensions(). (Specifically, ensures that the
  275. serialization is in canonical, tag-number order).
  276. """
  277. my_extension_int = unittest_pb2.my_extension_int
  278. my_extension_string = unittest_pb2.my_extension_string
  279. expected_strings = []
  280. message = unittest_pb2.TestFieldOrderings()
  281. message.my_int = 1 # Field 1.
  282. expected_strings.append(message.SerializeToString())
  283. message.Clear()
  284. message.Extensions[my_extension_int] = 23 # Field 5.
  285. expected_strings.append(message.SerializeToString())
  286. message.Clear()
  287. message.my_string = 'foo' # Field 11.
  288. expected_strings.append(message.SerializeToString())
  289. message.Clear()
  290. message.Extensions[my_extension_string] = 'bar' # Field 50.
  291. expected_strings.append(message.SerializeToString())
  292. message.Clear()
  293. message.my_float = 1.0
  294. expected_strings.append(message.SerializeToString())
  295. message.Clear()
  296. expected = ''.join(expected_strings)
  297. if expected != serialized:
  298. raise ValueError('Expected %r, found %r' % (expected, serialized))
  299. def ExpectAllFieldsSet(test_case, message):
  300. """Check all fields for correct values have after Set*Fields() is called."""
  301. test_case.assertTrue(message.HasField('optional_int32'))
  302. test_case.assertTrue(message.HasField('optional_int64'))
  303. test_case.assertTrue(message.HasField('optional_uint32'))
  304. test_case.assertTrue(message.HasField('optional_uint64'))
  305. test_case.assertTrue(message.HasField('optional_sint32'))
  306. test_case.assertTrue(message.HasField('optional_sint64'))
  307. test_case.assertTrue(message.HasField('optional_fixed32'))
  308. test_case.assertTrue(message.HasField('optional_fixed64'))
  309. test_case.assertTrue(message.HasField('optional_sfixed32'))
  310. test_case.assertTrue(message.HasField('optional_sfixed64'))
  311. test_case.assertTrue(message.HasField('optional_float'))
  312. test_case.assertTrue(message.HasField('optional_double'))
  313. test_case.assertTrue(message.HasField('optional_bool'))
  314. test_case.assertTrue(message.HasField('optional_string'))
  315. test_case.assertTrue(message.HasField('optional_bytes'))
  316. test_case.assertTrue(message.HasField('optionalgroup'))
  317. test_case.assertTrue(message.HasField('optional_nested_message'))
  318. test_case.assertTrue(message.HasField('optional_foreign_message'))
  319. test_case.assertTrue(message.HasField('optional_import_message'))
  320. test_case.assertTrue(message.optionalgroup.HasField('a'))
  321. test_case.assertTrue(message.optional_nested_message.HasField('bb'))
  322. test_case.assertTrue(message.optional_foreign_message.HasField('c'))
  323. test_case.assertTrue(message.optional_import_message.HasField('d'))
  324. test_case.assertTrue(message.HasField('optional_nested_enum'))
  325. test_case.assertTrue(message.HasField('optional_foreign_enum'))
  326. test_case.assertTrue(message.HasField('optional_import_enum'))
  327. test_case.assertTrue(message.HasField('optional_string_piece'))
  328. test_case.assertTrue(message.HasField('optional_cord'))
  329. test_case.assertEqual(101, message.optional_int32)
  330. test_case.assertEqual(102, message.optional_int64)
  331. test_case.assertEqual(103, message.optional_uint32)
  332. test_case.assertEqual(104, message.optional_uint64)
  333. test_case.assertEqual(105, message.optional_sint32)
  334. test_case.assertEqual(106, message.optional_sint64)
  335. test_case.assertEqual(107, message.optional_fixed32)
  336. test_case.assertEqual(108, message.optional_fixed64)
  337. test_case.assertEqual(109, message.optional_sfixed32)
  338. test_case.assertEqual(110, message.optional_sfixed64)
  339. test_case.assertEqual(111, message.optional_float)
  340. test_case.assertEqual(112, message.optional_double)
  341. test_case.assertEqual(True, message.optional_bool)
  342. test_case.assertEqual('115', message.optional_string)
  343. test_case.assertEqual('116', message.optional_bytes)
  344. test_case.assertEqual(117, message.optionalgroup.a)
  345. test_case.assertEqual(118, message.optional_nested_message.bb)
  346. test_case.assertEqual(119, message.optional_foreign_message.c)
  347. test_case.assertEqual(120, message.optional_import_message.d)
  348. test_case.assertEqual(unittest_pb2.TestAllTypes.BAZ,
  349. message.optional_nested_enum)
  350. test_case.assertEqual(unittest_pb2.FOREIGN_BAZ,
  351. message.optional_foreign_enum)
  352. test_case.assertEqual(unittest_import_pb2.IMPORT_BAZ,
  353. message.optional_import_enum)
  354. # -----------------------------------------------------------------
  355. test_case.assertEqual(2, len(message.repeated_int32))
  356. test_case.assertEqual(2, len(message.repeated_int64))
  357. test_case.assertEqual(2, len(message.repeated_uint32))
  358. test_case.assertEqual(2, len(message.repeated_uint64))
  359. test_case.assertEqual(2, len(message.repeated_sint32))
  360. test_case.assertEqual(2, len(message.repeated_sint64))
  361. test_case.assertEqual(2, len(message.repeated_fixed32))
  362. test_case.assertEqual(2, len(message.repeated_fixed64))
  363. test_case.assertEqual(2, len(message.repeated_sfixed32))
  364. test_case.assertEqual(2, len(message.repeated_sfixed64))
  365. test_case.assertEqual(2, len(message.repeated_float))
  366. test_case.assertEqual(2, len(message.repeated_double))
  367. test_case.assertEqual(2, len(message.repeated_bool))
  368. test_case.assertEqual(2, len(message.repeated_string))
  369. test_case.assertEqual(2, len(message.repeated_bytes))
  370. test_case.assertEqual(2, len(message.repeatedgroup))
  371. test_case.assertEqual(2, len(message.repeated_nested_message))
  372. test_case.assertEqual(2, len(message.repeated_foreign_message))
  373. test_case.assertEqual(2, len(message.repeated_import_message))
  374. test_case.assertEqual(2, len(message.repeated_nested_enum))
  375. test_case.assertEqual(2, len(message.repeated_foreign_enum))
  376. test_case.assertEqual(2, len(message.repeated_import_enum))
  377. test_case.assertEqual(2, len(message.repeated_string_piece))
  378. test_case.assertEqual(2, len(message.repeated_cord))
  379. test_case.assertEqual(201, message.repeated_int32[0])
  380. test_case.assertEqual(202, message.repeated_int64[0])
  381. test_case.assertEqual(203, message.repeated_uint32[0])
  382. test_case.assertEqual(204, message.repeated_uint64[0])
  383. test_case.assertEqual(205, message.repeated_sint32[0])
  384. test_case.assertEqual(206, message.repeated_sint64[0])
  385. test_case.assertEqual(207, message.repeated_fixed32[0])
  386. test_case.assertEqual(208, message.repeated_fixed64[0])
  387. test_case.assertEqual(209, message.repeated_sfixed32[0])
  388. test_case.assertEqual(210, message.repeated_sfixed64[0])
  389. test_case.assertEqual(211, message.repeated_float[0])
  390. test_case.assertEqual(212, message.repeated_double[0])
  391. test_case.assertEqual(True, message.repeated_bool[0])
  392. test_case.assertEqual('215', message.repeated_string[0])
  393. test_case.assertEqual('216', message.repeated_bytes[0])
  394. test_case.assertEqual(217, message.repeatedgroup[0].a)
  395. test_case.assertEqual(218, message.repeated_nested_message[0].bb)
  396. test_case.assertEqual(219, message.repeated_foreign_message[0].c)
  397. test_case.assertEqual(220, message.repeated_import_message[0].d)
  398. test_case.assertEqual(unittest_pb2.TestAllTypes.BAR,
  399. message.repeated_nested_enum[0])
  400. test_case.assertEqual(unittest_pb2.FOREIGN_BAR,
  401. message.repeated_foreign_enum[0])
  402. test_case.assertEqual(unittest_import_pb2.IMPORT_BAR,
  403. message.repeated_import_enum[0])
  404. test_case.assertEqual(301, message.repeated_int32[1])
  405. test_case.assertEqual(302, message.repeated_int64[1])
  406. test_case.assertEqual(303, message.repeated_uint32[1])
  407. test_case.assertEqual(304, message.repeated_uint64[1])
  408. test_case.assertEqual(305, message.repeated_sint32[1])
  409. test_case.assertEqual(306, message.repeated_sint64[1])
  410. test_case.assertEqual(307, message.repeated_fixed32[1])
  411. test_case.assertEqual(308, message.repeated_fixed64[1])
  412. test_case.assertEqual(309, message.repeated_sfixed32[1])
  413. test_case.assertEqual(310, message.repeated_sfixed64[1])
  414. test_case.assertEqual(311, message.repeated_float[1])
  415. test_case.assertEqual(312, message.repeated_double[1])
  416. test_case.assertEqual(False, message.repeated_bool[1])
  417. test_case.assertEqual('315', message.repeated_string[1])
  418. test_case.assertEqual('316', message.repeated_bytes[1])
  419. test_case.assertEqual(317, message.repeatedgroup[1].a)
  420. test_case.assertEqual(318, message.repeated_nested_message[1].bb)
  421. test_case.assertEqual(319, message.repeated_foreign_message[1].c)
  422. test_case.assertEqual(320, message.repeated_import_message[1].d)
  423. test_case.assertEqual(unittest_pb2.TestAllTypes.BAZ,
  424. message.repeated_nested_enum[1])
  425. test_case.assertEqual(unittest_pb2.FOREIGN_BAZ,
  426. message.repeated_foreign_enum[1])
  427. test_case.assertEqual(unittest_import_pb2.IMPORT_BAZ,
  428. message.repeated_import_enum[1])
  429. # -----------------------------------------------------------------
  430. test_case.assertTrue(message.HasField('default_int32'))
  431. test_case.assertTrue(message.HasField('default_int64'))
  432. test_case.assertTrue(message.HasField('default_uint32'))
  433. test_case.assertTrue(message.HasField('default_uint64'))
  434. test_case.assertTrue(message.HasField('default_sint32'))
  435. test_case.assertTrue(message.HasField('default_sint64'))
  436. test_case.assertTrue(message.HasField('default_fixed32'))
  437. test_case.assertTrue(message.HasField('default_fixed64'))
  438. test_case.assertTrue(message.HasField('default_sfixed32'))
  439. test_case.assertTrue(message.HasField('default_sfixed64'))
  440. test_case.assertTrue(message.HasField('default_float'))
  441. test_case.assertTrue(message.HasField('default_double'))
  442. test_case.assertTrue(message.HasField('default_bool'))
  443. test_case.assertTrue(message.HasField('default_string'))
  444. test_case.assertTrue(message.HasField('default_bytes'))
  445. test_case.assertTrue(message.HasField('default_nested_enum'))
  446. test_case.assertTrue(message.HasField('default_foreign_enum'))
  447. test_case.assertTrue(message.HasField('default_import_enum'))
  448. test_case.assertEqual(401, message.default_int32)
  449. test_case.assertEqual(402, message.default_int64)
  450. test_case.assertEqual(403, message.default_uint32)
  451. test_case.assertEqual(404, message.default_uint64)
  452. test_case.assertEqual(405, message.default_sint32)
  453. test_case.assertEqual(406, message.default_sint64)
  454. test_case.assertEqual(407, message.default_fixed32)
  455. test_case.assertEqual(408, message.default_fixed64)
  456. test_case.assertEqual(409, message.default_sfixed32)
  457. test_case.assertEqual(410, message.default_sfixed64)
  458. test_case.assertEqual(411, message.default_float)
  459. test_case.assertEqual(412, message.default_double)
  460. test_case.assertEqual(False, message.default_bool)
  461. test_case.assertEqual('415', message.default_string)
  462. test_case.assertEqual('416', message.default_bytes)
  463. test_case.assertEqual(unittest_pb2.TestAllTypes.FOO,
  464. message.default_nested_enum)
  465. test_case.assertEqual(unittest_pb2.FOREIGN_FOO,
  466. message.default_foreign_enum)
  467. test_case.assertEqual(unittest_import_pb2.IMPORT_FOO,
  468. message.default_import_enum)
  469. def GoldenFile(filename):
  470. """Finds the given golden file and returns a file object representing it."""
  471. # Search up the directory tree looking for the C++ protobuf source code.
  472. path = '.'
  473. while os.path.exists(path):
  474. if os.path.exists(os.path.join(path, 'src/google/protobuf')):
  475. # Found it. Load the golden file from the testdata directory.
  476. full_path = os.path.join(path, 'src/google/protobuf/testdata', filename)
  477. return open(full_path, 'rb')
  478. path = os.path.join(path, '..')
  479. raise RuntimeError(
  480. 'Could not find golden files. This test must be run from within the '
  481. 'protobuf source package so that it can read test data files from the '
  482. 'C++ source tree.')
  483. def SetAllPackedFields(message):
  484. """Sets every field in the message to a unique value.
  485. Args:
  486. message: A unittest_pb2.TestPackedTypes instance.
  487. """
  488. message.packed_int32.extend([601, 701])
  489. message.packed_int64.extend([602, 702])
  490. message.packed_uint32.extend([603, 703])
  491. message.packed_uint64.extend([604, 704])
  492. message.packed_sint32.extend([605, 705])
  493. message.packed_sint64.extend([606, 706])
  494. message.packed_fixed32.extend([607, 707])
  495. message.packed_fixed64.extend([608, 708])
  496. message.packed_sfixed32.extend([609, 709])
  497. message.packed_sfixed64.extend([610, 710])
  498. message.packed_float.extend([611.0, 711.0])
  499. message.packed_double.extend([612.0, 712.0])
  500. message.packed_bool.extend([True, False])
  501. message.packed_enum.extend([unittest_pb2.FOREIGN_BAR,
  502. unittest_pb2.FOREIGN_BAZ])
  503. def SetAllPackedExtensions(message):
  504. """Sets every extension in the message to a unique value.
  505. Args:
  506. message: A unittest_pb2.TestPackedExtensions instance.
  507. """
  508. extensions = message.Extensions
  509. pb2 = unittest_pb2
  510. extensions[pb2.packed_int32_extension].extend([601, 701])
  511. extensions[pb2.packed_int64_extension].extend([602, 702])
  512. extensions[pb2.packed_uint32_extension].extend([603, 703])
  513. extensions[pb2.packed_uint64_extension].extend([604, 704])
  514. extensions[pb2.packed_sint32_extension].extend([605, 705])
  515. extensions[pb2.packed_sint64_extension].extend([606, 706])
  516. extensions[pb2.packed_fixed32_extension].extend([607, 707])
  517. extensions[pb2.packed_fixed64_extension].extend([608, 708])
  518. extensions[pb2.packed_sfixed32_extension].extend([609, 709])
  519. extensions[pb2.packed_sfixed64_extension].extend([610, 710])
  520. extensions[pb2.packed_float_extension].extend([611.0, 711.0])
  521. extensions[pb2.packed_double_extension].extend([612.0, 712.0])
  522. extensions[pb2.packed_bool_extension].extend([True, False])
  523. extensions[pb2.packed_enum_extension].extend([unittest_pb2.FOREIGN_BAR,
  524. unittest_pb2.FOREIGN_BAZ])
  525. def SetAllUnpackedFields(message):
  526. """Sets every field in the message to a unique value.
  527. Args:
  528. message: A unittest_pb2.TestUnpackedTypes instance.
  529. """
  530. message.unpacked_int32.extend([601, 701])
  531. message.unpacked_int64.extend([602, 702])
  532. message.unpacked_uint32.extend([603, 703])
  533. message.unpacked_uint64.extend([604, 704])
  534. message.unpacked_sint32.extend([605, 705])
  535. message.unpacked_sint64.extend([606, 706])
  536. message.unpacked_fixed32.extend([607, 707])
  537. message.unpacked_fixed64.extend([608, 708])
  538. message.unpacked_sfixed32.extend([609, 709])
  539. message.unpacked_sfixed64.extend([610, 710])
  540. message.unpacked_float.extend([611.0, 711.0])
  541. message.unpacked_double.extend([612.0, 712.0])
  542. message.unpacked_bool.extend([True, False])
  543. message.unpacked_enum.extend([unittest_pb2.FOREIGN_BAR,
  544. unittest_pb2.FOREIGN_BAZ])