basic.rb 37 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276
  1. #!/usr/bin/ruby
  2. require 'google/protobuf'
  3. require 'test/unit'
  4. # ------------- generated code --------------
  5. module BasicTest
  6. pool = Google::Protobuf::DescriptorPool.new
  7. pool.build do
  8. add_message "Foo" do
  9. optional :bar, :message, 1, "Bar"
  10. repeated :baz, :message, 2, "Baz"
  11. end
  12. add_message "Bar" do
  13. optional :msg, :string, 1
  14. end
  15. add_message "Baz" do
  16. optional :msg, :string, 1
  17. end
  18. add_message "TestMessage" do
  19. optional :optional_int32, :int32, 1
  20. optional :optional_int64, :int64, 2
  21. optional :optional_uint32, :uint32, 3
  22. optional :optional_uint64, :uint64, 4
  23. optional :optional_bool, :bool, 5
  24. optional :optional_float, :float, 6
  25. optional :optional_double, :double, 7
  26. optional :optional_string, :string, 8
  27. optional :optional_bytes, :bytes, 9
  28. optional :optional_msg, :message, 10, "TestMessage2"
  29. optional :optional_enum, :enum, 11, "TestEnum"
  30. repeated :repeated_int32, :int32, 12
  31. repeated :repeated_int64, :int64, 13
  32. repeated :repeated_uint32, :uint32, 14
  33. repeated :repeated_uint64, :uint64, 15
  34. repeated :repeated_bool, :bool, 16
  35. repeated :repeated_float, :float, 17
  36. repeated :repeated_double, :double, 18
  37. repeated :repeated_string, :string, 19
  38. repeated :repeated_bytes, :bytes, 20
  39. repeated :repeated_msg, :message, 21, "TestMessage2"
  40. repeated :repeated_enum, :enum, 22, "TestEnum"
  41. end
  42. add_message "TestMessage2" do
  43. optional :foo, :int32, 1
  44. end
  45. add_message "Recursive1" do
  46. optional :foo, :message, 1, "Recursive2"
  47. end
  48. add_message "Recursive2" do
  49. optional :foo, :message, 1, "Recursive1"
  50. end
  51. add_enum "TestEnum" do
  52. value :Default, 0
  53. value :A, 1
  54. value :B, 2
  55. value :C, 3
  56. end
  57. add_message "BadFieldNames" do
  58. optional :dup, :int32, 1
  59. optional :class, :int32, 2
  60. end
  61. add_message "MapMessage" do
  62. map :map_string_int32, :string, :int32, 1
  63. map :map_string_msg, :string, :message, 2, "TestMessage2"
  64. end
  65. add_message "MapMessageWireEquiv" do
  66. repeated :map_string_int32, :message, 1, "MapMessageWireEquiv_entry1"
  67. repeated :map_string_msg, :message, 2, "MapMessageWireEquiv_entry2"
  68. end
  69. add_message "MapMessageWireEquiv_entry1" do
  70. optional :key, :string, 1
  71. optional :value, :int32, 2
  72. end
  73. add_message "MapMessageWireEquiv_entry2" do
  74. optional :key, :string, 1
  75. optional :value, :message, 2, "TestMessage2"
  76. end
  77. add_message "OneofMessage" do
  78. oneof :my_oneof do
  79. optional :a, :string, 1
  80. optional :b, :int32, 2
  81. optional :c, :message, 3, "TestMessage2"
  82. optional :d, :enum, 4, "TestEnum"
  83. end
  84. end
  85. end
  86. Foo = pool.lookup("Foo").msgclass
  87. Bar = pool.lookup("Bar").msgclass
  88. Baz = pool.lookup("Baz").msgclass
  89. TestMessage = pool.lookup("TestMessage").msgclass
  90. TestMessage2 = pool.lookup("TestMessage2").msgclass
  91. Recursive1 = pool.lookup("Recursive1").msgclass
  92. Recursive2 = pool.lookup("Recursive2").msgclass
  93. TestEnum = pool.lookup("TestEnum").enummodule
  94. BadFieldNames = pool.lookup("BadFieldNames").msgclass
  95. MapMessage = pool.lookup("MapMessage").msgclass
  96. MapMessageWireEquiv = pool.lookup("MapMessageWireEquiv").msgclass
  97. MapMessageWireEquiv_entry1 =
  98. pool.lookup("MapMessageWireEquiv_entry1").msgclass
  99. MapMessageWireEquiv_entry2 =
  100. pool.lookup("MapMessageWireEquiv_entry2").msgclass
  101. OneofMessage = pool.lookup("OneofMessage").msgclass
  102. # ------------ test cases ---------------
  103. class MessageContainerTest < Test::Unit::TestCase
  104. def test_defaults
  105. m = TestMessage.new
  106. assert m.optional_int32 == 0
  107. assert m.optional_int64 == 0
  108. assert m.optional_uint32 == 0
  109. assert m.optional_uint64 == 0
  110. assert m.optional_bool == false
  111. assert m.optional_float == 0.0
  112. assert m.optional_double == 0.0
  113. assert m.optional_string == ""
  114. assert m.optional_bytes == ""
  115. assert m.optional_msg == nil
  116. assert m.optional_enum == :Default
  117. end
  118. def test_setters
  119. m = TestMessage.new
  120. m.optional_int32 = -42
  121. assert m.optional_int32 == -42
  122. m.optional_int64 = -0x1_0000_0000
  123. assert m.optional_int64 == -0x1_0000_0000
  124. m.optional_uint32 = 0x9000_0000
  125. assert m.optional_uint32 == 0x9000_0000
  126. m.optional_uint64 = 0x9000_0000_0000_0000
  127. assert m.optional_uint64 == 0x9000_0000_0000_0000
  128. m.optional_bool = true
  129. assert m.optional_bool == true
  130. m.optional_float = 0.5
  131. assert m.optional_float == 0.5
  132. m.optional_double = 0.5
  133. m.optional_string = "hello"
  134. assert m.optional_string == "hello"
  135. m.optional_bytes = "world".encode!('ASCII-8BIT')
  136. assert m.optional_bytes == "world"
  137. m.optional_msg = TestMessage2.new(:foo => 42)
  138. assert m.optional_msg == TestMessage2.new(:foo => 42)
  139. m.optional_msg = nil
  140. assert m.optional_msg == nil
  141. end
  142. def test_ctor_args
  143. m = TestMessage.new(:optional_int32 => -42,
  144. :optional_msg => TestMessage2.new,
  145. :optional_enum => :C,
  146. :repeated_string => ["hello", "there", "world"])
  147. assert m.optional_int32 == -42
  148. assert m.optional_msg.class == TestMessage2
  149. assert m.repeated_string.length == 3
  150. assert m.optional_enum == :C
  151. assert m.repeated_string[0] == "hello"
  152. assert m.repeated_string[1] == "there"
  153. assert m.repeated_string[2] == "world"
  154. end
  155. def test_inspect
  156. m = TestMessage.new(:optional_int32 => -42,
  157. :optional_enum => :A,
  158. :optional_msg => TestMessage2.new,
  159. :repeated_string => ["hello", "there", "world"])
  160. expected = '<BasicTest::TestMessage: optional_int32: -42, optional_int64: 0, optional_uint32: 0, optional_uint64: 0, optional_bool: false, optional_float: 0.0, optional_double: 0.0, optional_string: "", optional_bytes: "", optional_msg: <BasicTest::TestMessage2: foo: 0>, optional_enum: :A, repeated_int32: [], repeated_int64: [], repeated_uint32: [], repeated_uint64: [], repeated_bool: [], repeated_float: [], repeated_double: [], repeated_string: ["hello", "there", "world"], repeated_bytes: [], repeated_msg: [], repeated_enum: []>'
  161. assert_equal expected, m.inspect
  162. end
  163. def test_hash
  164. m1 = TestMessage.new(:optional_int32 => 42)
  165. m2 = TestMessage.new(:optional_int32 => 102)
  166. assert m1.hash != 0
  167. assert m2.hash != 0
  168. # relying on the randomness here -- if hash function changes and we are
  169. # unlucky enough to get a collision, then change the values above.
  170. assert m1.hash != m2.hash
  171. end
  172. def test_unknown_field_errors
  173. e = assert_raise NoMethodError do
  174. TestMessage.new.hello
  175. end
  176. assert_match(/hello/, e.message)
  177. e = assert_raise NoMethodError do
  178. TestMessage.new.hello = "world"
  179. end
  180. assert_match(/hello/, e.message)
  181. end
  182. def test_initialization_map_errors
  183. e = assert_raise ArgumentError do
  184. TestMessage.new(:hello => "world")
  185. end
  186. assert_match(/hello/, e.message)
  187. e = assert_raise ArgumentError do
  188. MapMessage.new(:map_string_int32 => "hello")
  189. end
  190. assert_equal e.message, "Expected Hash object as initializer value for map field 'map_string_int32' (given String)."
  191. e = assert_raise ArgumentError do
  192. TestMessage.new(:repeated_uint32 => "hello")
  193. end
  194. assert_equal e.message, "Expected array as initializer value for repeated field 'repeated_uint32' (given String)."
  195. end
  196. def test_type_errors
  197. m = TestMessage.new
  198. # Use rescue to allow subclasses of error
  199. success = false
  200. begin
  201. m.optional_int32 = "hello"
  202. rescue TypeError
  203. success = true
  204. end
  205. assert(success)
  206. success = false
  207. begin
  208. m.optional_string = nil
  209. rescue TypeError
  210. success = true
  211. end
  212. assert(success)
  213. success = false
  214. begin
  215. m.optional_bool = 42
  216. rescue TypeError
  217. success = true
  218. end
  219. assert(success)
  220. success = false
  221. begin
  222. m.optional_msg = TestMessage.new # expects TestMessage2
  223. rescue TypeError
  224. success = true
  225. end
  226. assert(success)
  227. success = false
  228. begin
  229. m.repeated_int32 = [] # needs RepeatedField
  230. rescue TypeError
  231. success = true
  232. end
  233. assert(success)
  234. success = false
  235. begin
  236. m.repeated_msg.push TestMessage.new
  237. rescue TypeError
  238. success = true
  239. end
  240. assert(success)
  241. end
  242. def test_string_encoding
  243. m = TestMessage.new
  244. # Assigning a normal (ASCII or UTF8) string to a bytes field, or
  245. # ASCII-8BIT to a string field will convert to the proper encoding.
  246. m.optional_bytes = "Test string ASCII".encode!('ASCII')
  247. assert m.optional_bytes.frozen?
  248. assert_equal Encoding::ASCII_8BIT, m.optional_bytes.encoding
  249. assert_equal "Test string ASCII", m.optional_bytes
  250. assert_raise Encoding::UndefinedConversionError do
  251. m.optional_bytes = "Test string UTF-8 \u0100".encode!('UTF-8')
  252. end
  253. assert_raise Encoding::UndefinedConversionError do
  254. m.optional_string = ["FFFF"].pack('H*')
  255. end
  256. # "Ordinary" use case.
  257. m.optional_bytes = ["FFFF"].pack('H*')
  258. m.optional_string = "\u0100"
  259. # strings are immutable so we can't do this, but serialize should catch it.
  260. m.optional_string = "asdf".encode!('UTF-8')
  261. assert_raise do
  262. m.optional_string.encode!('ASCII-8BIT')
  263. end
  264. end
  265. def test_rptfield_int32
  266. l = Google::Protobuf::RepeatedField.new(:int32)
  267. assert l.count == 0
  268. l = Google::Protobuf::RepeatedField.new(:int32, [1, 2, 3])
  269. assert l.count == 3
  270. assert_equal [1, 2, 3], l
  271. assert_equal l, [1, 2, 3]
  272. l.push 4
  273. assert l == [1, 2, 3, 4]
  274. dst_list = []
  275. l.each { |val| dst_list.push val }
  276. assert dst_list == [1, 2, 3, 4]
  277. assert l.to_a == [1, 2, 3, 4]
  278. assert l[0] == 1
  279. assert l[3] == 4
  280. l[0] = 5
  281. assert l == [5, 2, 3, 4]
  282. l2 = l.dup
  283. assert l == l2
  284. assert l.object_id != l2.object_id
  285. l2.push 6
  286. assert l.count == 4
  287. assert l2.count == 5
  288. assert l.inspect == '[5, 2, 3, 4]'
  289. l.concat([7, 8, 9])
  290. assert l == [5, 2, 3, 4, 7, 8, 9]
  291. assert l.pop == 9
  292. assert l == [5, 2, 3, 4, 7, 8]
  293. success = false
  294. begin
  295. m = TestMessage.new
  296. l.push m
  297. rescue TypeError
  298. success = true
  299. end
  300. assert(success)
  301. m = TestMessage.new
  302. m.repeated_int32 = l
  303. assert m.repeated_int32 == [5, 2, 3, 4, 7, 8]
  304. assert m.repeated_int32.object_id == l.object_id
  305. l.push 42
  306. assert m.repeated_int32.pop == 42
  307. l3 = l + l.dup
  308. assert l3.count == l.count * 2
  309. l.count.times do |i|
  310. assert l3[i] == l[i]
  311. assert l3[l.count + i] == l[i]
  312. end
  313. l.clear
  314. assert l.count == 0
  315. l += [1, 2, 3, 4]
  316. l.replace([5, 6, 7, 8])
  317. assert l == [5, 6, 7, 8]
  318. l4 = Google::Protobuf::RepeatedField.new(:int32)
  319. l4[5] = 42
  320. assert l4 == [0, 0, 0, 0, 0, 42]
  321. l4 << 100
  322. assert l4 == [0, 0, 0, 0, 0, 42, 100]
  323. l4 << 101 << 102
  324. assert l4 == [0, 0, 0, 0, 0, 42, 100, 101, 102]
  325. end
  326. def test_parent_rptfield
  327. #make sure we set the RepeatedField and can add to it
  328. m = TestMessage.new
  329. assert m.repeated_string == []
  330. m.repeated_string << 'ok'
  331. m.repeated_string.push('ok2')
  332. assert m.repeated_string == ['ok', 'ok2']
  333. m.repeated_string += ['ok3']
  334. assert m.repeated_string == ['ok', 'ok2', 'ok3']
  335. end
  336. def test_rptfield_msg
  337. l = Google::Protobuf::RepeatedField.new(:message, TestMessage)
  338. l.push TestMessage.new
  339. assert l.count == 1
  340. success = false
  341. begin
  342. l.push TestMessage2.new
  343. rescue TypeError
  344. success = true
  345. end
  346. assert(success)
  347. success = false
  348. begin
  349. l.push 42
  350. rescue TypeError
  351. success = true
  352. end
  353. assert(success)
  354. l2 = l.dup
  355. assert l2[0] == l[0]
  356. assert l2[0].object_id == l[0].object_id
  357. l2 = Google::Protobuf.deep_copy(l)
  358. assert l2[0] == l[0]
  359. assert l2[0].object_id != l[0].object_id
  360. l3 = l + l2
  361. assert l3.count == 2
  362. assert l3[0] == l[0]
  363. assert l3[1] == l2[0]
  364. l3[0].optional_int32 = 1000
  365. assert l[0].optional_int32 == 1000
  366. new_msg = TestMessage.new(:optional_int32 => 200)
  367. l4 = l + [new_msg]
  368. assert l4.count == 2
  369. new_msg.optional_int32 = 1000
  370. assert l4[1].optional_int32 == 1000
  371. end
  372. def test_rptfield_enum
  373. l = Google::Protobuf::RepeatedField.new(:enum, TestEnum)
  374. l.push :A
  375. l.push :B
  376. l.push :C
  377. assert l.count == 3
  378. assert_raise RangeError do
  379. l.push :D
  380. end
  381. assert l[0] == :A
  382. l.push 4
  383. assert l[3] == 4
  384. end
  385. def test_rptfield_initialize
  386. assert_raise ArgumentError do
  387. l = Google::Protobuf::RepeatedField.new
  388. end
  389. assert_raise ArgumentError do
  390. l = Google::Protobuf::RepeatedField.new(:message)
  391. end
  392. assert_raise ArgumentError do
  393. l = Google::Protobuf::RepeatedField.new([1, 2, 3])
  394. end
  395. assert_raise ArgumentError do
  396. l = Google::Protobuf::RepeatedField.new(:message, [TestMessage2.new])
  397. end
  398. end
  399. def test_rptfield_array_ducktyping
  400. l = Google::Protobuf::RepeatedField.new(:int32)
  401. length_methods = %w(count length size)
  402. length_methods.each do |lm|
  403. assert l.send(lm) == 0
  404. end
  405. # out of bounds returns a nil
  406. assert l[0] == nil
  407. assert l[1] == nil
  408. assert l[-1] == nil
  409. l.push 4
  410. length_methods.each do |lm|
  411. assert l.send(lm) == 1
  412. end
  413. assert l[0] == 4
  414. assert l[1] == nil
  415. assert l[-1] == 4
  416. assert l[-2] == nil
  417. l.push 2
  418. length_methods.each do |lm|
  419. assert l.send(lm) == 2
  420. end
  421. assert l[0] == 4
  422. assert l[1] == 2
  423. assert l[2] == nil
  424. assert l[-1] == 2
  425. assert l[-2] == 4
  426. assert l[-3] == nil
  427. #adding out of scope will backfill with empty objects
  428. end
  429. def test_map_basic
  430. # allowed key types:
  431. # :int32, :int64, :uint32, :uint64, :bool, :string, :bytes.
  432. m = Google::Protobuf::Map.new(:string, :int32)
  433. m["asdf"] = 1
  434. assert m["asdf"] == 1
  435. m["jkl;"] = 42
  436. assert m == { "jkl;" => 42, "asdf" => 1 }
  437. assert m.has_key?("asdf")
  438. assert !m.has_key?("qwerty")
  439. assert m.length == 2
  440. m2 = m.dup
  441. assert m == m2
  442. assert m.hash != 0
  443. assert m.hash == m2.hash
  444. collected = {}
  445. m.each { |k,v| collected[v] = k }
  446. assert collected == { 42 => "jkl;", 1 => "asdf" }
  447. assert m.delete("asdf") == 1
  448. assert !m.has_key?("asdf")
  449. assert m["asdf"] == nil
  450. assert !m.has_key?("asdf")
  451. # We only assert on inspect value when there is one map entry because the
  452. # order in which elements appear is unspecified (depends on the internal
  453. # hash function). We don't want a brittle test.
  454. assert m.inspect == "{\"jkl;\"=>42}"
  455. assert m.keys == ["jkl;"]
  456. assert m.values == [42]
  457. m.clear
  458. assert m.length == 0
  459. assert m == {}
  460. success = false
  461. begin
  462. m[1] = 1
  463. rescue TypeError
  464. success = true
  465. end
  466. assert(success)
  467. assert_raise RangeError do
  468. m["asdf"] = 0x1_0000_0000
  469. end
  470. end
  471. def test_map_ctor
  472. m = Google::Protobuf::Map.new(:string, :int32,
  473. {"a" => 1, "b" => 2, "c" => 3})
  474. assert m == {"a" => 1, "c" => 3, "b" => 2}
  475. end
  476. def test_map_keytypes
  477. m = Google::Protobuf::Map.new(:int32, :int32)
  478. m[1] = 42
  479. m[-1] = 42
  480. assert_raise RangeError do
  481. m[0x8000_0000] = 1
  482. end
  483. success = false
  484. begin
  485. m["asdf"] = 1
  486. rescue TypeError
  487. success = true
  488. end
  489. assert(success)
  490. m = Google::Protobuf::Map.new(:int64, :int32)
  491. m[0x1000_0000_0000_0000] = 1
  492. assert_raise RangeError do
  493. m[0x1_0000_0000_0000_0000] = 1
  494. end
  495. success = false
  496. begin
  497. m["asdf"] = 1
  498. rescue TypeError
  499. success = true
  500. end
  501. assert(success)
  502. m = Google::Protobuf::Map.new(:uint32, :int32)
  503. m[0x8000_0000] = 1
  504. assert_raise RangeError do
  505. m[0x1_0000_0000] = 1
  506. end
  507. assert_raise RangeError do
  508. m[-1] = 1
  509. end
  510. m = Google::Protobuf::Map.new(:uint64, :int32)
  511. m[0x8000_0000_0000_0000] = 1
  512. assert_raise RangeError do
  513. m[0x1_0000_0000_0000_0000] = 1
  514. end
  515. assert_raise RangeError do
  516. m[-1] = 1
  517. end
  518. m = Google::Protobuf::Map.new(:bool, :int32)
  519. m[true] = 1
  520. m[false] = 2
  521. success = false
  522. begin
  523. m[1] = 1
  524. rescue TypeError
  525. success = true
  526. end
  527. assert(success)
  528. success = false
  529. begin
  530. m["asdf"] = 1
  531. rescue TypeError
  532. success = true
  533. end
  534. assert(success)
  535. m = Google::Protobuf::Map.new(:string, :int32)
  536. m["asdf"] = 1
  537. success = false
  538. begin
  539. m[1] = 1
  540. rescue TypeError
  541. success = true
  542. end
  543. assert(success)
  544. assert_raise Encoding::UndefinedConversionError do
  545. bytestring = ["FFFF"].pack("H*")
  546. m[bytestring] = 1
  547. end
  548. m = Google::Protobuf::Map.new(:bytes, :int32)
  549. bytestring = ["FFFF"].pack("H*")
  550. m[bytestring] = 1
  551. # Allowed -- we will automatically convert to ASCII-8BIT.
  552. m["asdf"] = 1
  553. success = false
  554. begin
  555. m[1] = 1
  556. rescue TypeError
  557. success = true
  558. end
  559. assert(success)
  560. end
  561. def test_map_msg_enum_valuetypes
  562. m = Google::Protobuf::Map.new(:string, :message, TestMessage)
  563. m["asdf"] = TestMessage.new
  564. success = false
  565. begin
  566. m["jkl;"] = TestMessage2.new
  567. rescue TypeError
  568. success = true
  569. end
  570. assert(success)
  571. m = Google::Protobuf::Map.new(
  572. :string, :message, TestMessage,
  573. { "a" => TestMessage.new(:optional_int32 => 42),
  574. "b" => TestMessage.new(:optional_int32 => 84) })
  575. assert m.length == 2
  576. assert m.values.map{|msg| msg.optional_int32}.sort == [42, 84]
  577. m = Google::Protobuf::Map.new(:string, :enum, TestEnum,
  578. { "x" => :A, "y" => :B, "z" => :C })
  579. assert m.length == 3
  580. assert m["z"] == :C
  581. m["z"] = 2
  582. assert m["z"] == :B
  583. m["z"] = 4
  584. assert m["z"] == 4
  585. assert_raise RangeError do
  586. m["z"] = :Z
  587. end
  588. assert_raise RangeError do
  589. m["z"] = "z"
  590. end
  591. end
  592. def test_map_dup_deep_copy
  593. m = Google::Protobuf::Map.new(
  594. :string, :message, TestMessage,
  595. { "a" => TestMessage.new(:optional_int32 => 42),
  596. "b" => TestMessage.new(:optional_int32 => 84) })
  597. m2 = m.dup
  598. assert m == m2
  599. assert m.object_id != m2.object_id
  600. assert m["a"].object_id == m2["a"].object_id
  601. assert m["b"].object_id == m2["b"].object_id
  602. m2 = Google::Protobuf.deep_copy(m)
  603. assert m == m2
  604. assert m.object_id != m2.object_id
  605. assert m["a"].object_id != m2["a"].object_id
  606. assert m["b"].object_id != m2["b"].object_id
  607. end
  608. def test_map_field
  609. m = MapMessage.new
  610. assert m.map_string_int32 == {}
  611. assert m.map_string_msg == {}
  612. m = MapMessage.new(
  613. :map_string_int32 => {"a" => 1, "b" => 2},
  614. :map_string_msg => {"a" => TestMessage2.new(:foo => 1),
  615. "b" => TestMessage2.new(:foo => 2)})
  616. assert m.map_string_int32.keys.sort == ["a", "b"]
  617. assert m.map_string_int32["a"] == 1
  618. assert m.map_string_msg["b"].foo == 2
  619. m.map_string_int32["c"] = 3
  620. assert m.map_string_int32["c"] == 3
  621. m.map_string_msg["c"] = TestMessage2.new(:foo => 3)
  622. assert m.map_string_msg["c"] == TestMessage2.new(:foo => 3)
  623. m.map_string_msg.delete("b")
  624. m.map_string_msg.delete("c")
  625. assert m.map_string_msg == { "a" => TestMessage2.new(:foo => 1) }
  626. success = false
  627. begin
  628. m.map_string_msg["e"] = TestMessage.new # wrong value type
  629. rescue TypeError
  630. success = true
  631. end
  632. assert(success)
  633. # ensure nothing was added by the above
  634. assert m.map_string_msg == { "a" => TestMessage2.new(:foo => 1) }
  635. m.map_string_int32 = Google::Protobuf::Map.new(:string, :int32)
  636. success = false
  637. begin
  638. m.map_string_int32 = Google::Protobuf::Map.new(:string, :int64)
  639. rescue TypeError
  640. success = true
  641. end
  642. assert(success)
  643. success = false
  644. begin
  645. m.map_string_int32 = {}
  646. rescue TypeError
  647. success = true
  648. end
  649. assert(success)
  650. success = false
  651. begin
  652. m = MapMessage.new(:map_string_int32 => { 1 => "I am not a number" })
  653. rescue TypeError
  654. success = true
  655. end
  656. assert(success)
  657. end
  658. def test_map_encode_decode
  659. m = MapMessage.new(
  660. :map_string_int32 => {"a" => 1, "b" => 2},
  661. :map_string_msg => {"a" => TestMessage2.new(:foo => 1),
  662. "b" => TestMessage2.new(:foo => 2)})
  663. m2 = MapMessage.decode(MapMessage.encode(m))
  664. assert m == m2
  665. m3 = MapMessageWireEquiv.decode(MapMessage.encode(m))
  666. assert m3.map_string_int32.length == 2
  667. kv = {}
  668. m3.map_string_int32.map { |msg| kv[msg.key] = msg.value }
  669. assert kv == {"a" => 1, "b" => 2}
  670. kv = {}
  671. m3.map_string_msg.map { |msg| kv[msg.key] = msg.value }
  672. assert kv == {"a" => TestMessage2.new(:foo => 1),
  673. "b" => TestMessage2.new(:foo => 2)}
  674. end
  675. def test_oneof_descriptors
  676. d = OneofMessage.descriptor
  677. o = d.lookup_oneof("my_oneof")
  678. assert o != nil
  679. assert o.class == Google::Protobuf::OneofDescriptor
  680. assert o.name == "my_oneof"
  681. oneof_count = 0
  682. d.each_oneof{ |oneof|
  683. oneof_count += 1
  684. assert oneof == o
  685. }
  686. assert oneof_count == 1
  687. assert o.count == 4
  688. field_names = o.map{|f| f.name}.sort
  689. assert field_names == ["a", "b", "c", "d"]
  690. end
  691. def test_oneof
  692. d = OneofMessage.new
  693. assert d.a == ""
  694. assert d.b == 0
  695. assert d.c == nil
  696. assert d.d == :Default
  697. assert d.my_oneof == nil
  698. d.a = "hi"
  699. assert d.a == "hi"
  700. assert d.b == 0
  701. assert d.c == nil
  702. assert d.d == :Default
  703. assert d.my_oneof == :a
  704. d.b = 42
  705. assert d.a == ""
  706. assert d.b == 42
  707. assert d.c == nil
  708. assert d.d == :Default
  709. assert d.my_oneof == :b
  710. d.c = TestMessage2.new(:foo => 100)
  711. assert d.a == ""
  712. assert d.b == 0
  713. assert d.c.foo == 100
  714. assert d.d == :Default
  715. assert d.my_oneof == :c
  716. d.d = :C
  717. assert d.a == ""
  718. assert d.b == 0
  719. assert d.c == nil
  720. assert d.d == :C
  721. assert d.my_oneof == :d
  722. d2 = OneofMessage.decode(OneofMessage.encode(d))
  723. assert d2 == d
  724. encoded_field_a = OneofMessage.encode(OneofMessage.new(:a => "string"))
  725. encoded_field_b = OneofMessage.encode(OneofMessage.new(:b => 1000))
  726. encoded_field_c = OneofMessage.encode(
  727. OneofMessage.new(:c => TestMessage2.new(:foo => 1)))
  728. encoded_field_d = OneofMessage.encode(OneofMessage.new(:d => :B))
  729. d3 = OneofMessage.decode(
  730. encoded_field_c + encoded_field_a + encoded_field_d)
  731. assert d3.a == ""
  732. assert d3.b == 0
  733. assert d3.c == nil
  734. assert d3.d == :B
  735. d4 = OneofMessage.decode(
  736. encoded_field_c + encoded_field_a + encoded_field_d +
  737. encoded_field_c)
  738. assert d4.a == ""
  739. assert d4.b == 0
  740. assert d4.c.foo == 1
  741. assert d4.d == :Default
  742. d5 = OneofMessage.new(:a => "hello")
  743. assert d5.a == "hello"
  744. d5.a = nil
  745. assert d5.a == ""
  746. assert OneofMessage.encode(d5) == ''
  747. assert d5.my_oneof == nil
  748. end
  749. def test_enum_field
  750. m = TestMessage.new
  751. assert m.optional_enum == :Default
  752. m.optional_enum = :A
  753. assert m.optional_enum == :A
  754. assert_raise RangeError do
  755. m.optional_enum = :ASDF
  756. end
  757. m.optional_enum = 1
  758. assert m.optional_enum == :A
  759. m.optional_enum = 100
  760. assert m.optional_enum == 100
  761. end
  762. def test_dup
  763. m = TestMessage.new
  764. m.optional_string = "hello"
  765. m.optional_int32 = 42
  766. tm1 = TestMessage2.new(:foo => 100)
  767. tm2 = TestMessage2.new(:foo => 200)
  768. m.repeated_msg.push tm1
  769. assert m.repeated_msg[-1] == tm1
  770. m.repeated_msg.push tm2
  771. assert m.repeated_msg[-1] == tm2
  772. m2 = m.dup
  773. assert m == m2
  774. m.optional_int32 += 1
  775. assert m != m2
  776. assert m.repeated_msg[0] == m2.repeated_msg[0]
  777. assert m.repeated_msg[0].object_id == m2.repeated_msg[0].object_id
  778. end
  779. def test_deep_copy
  780. m = TestMessage.new(:optional_int32 => 42,
  781. :repeated_msg => [TestMessage2.new(:foo => 100)])
  782. m2 = Google::Protobuf.deep_copy(m)
  783. assert m == m2
  784. assert m.repeated_msg == m2.repeated_msg
  785. assert m.repeated_msg.object_id != m2.repeated_msg.object_id
  786. assert m.repeated_msg[0].object_id != m2.repeated_msg[0].object_id
  787. end
  788. def test_eq
  789. m = TestMessage.new(:optional_int32 => 42,
  790. :repeated_int32 => [1, 2, 3])
  791. m2 = TestMessage.new(:optional_int32 => 43,
  792. :repeated_int32 => [1, 2, 3])
  793. assert m != m2
  794. end
  795. def test_enum_lookup
  796. assert TestEnum::A == 1
  797. assert TestEnum::B == 2
  798. assert TestEnum::C == 3
  799. assert TestEnum::lookup(1) == :A
  800. assert TestEnum::lookup(2) == :B
  801. assert TestEnum::lookup(3) == :C
  802. assert TestEnum::resolve(:A) == 1
  803. assert TestEnum::resolve(:B) == 2
  804. assert TestEnum::resolve(:C) == 3
  805. end
  806. def test_parse_serialize
  807. m = TestMessage.new(:optional_int32 => 42,
  808. :optional_string => "hello world",
  809. :optional_enum => :B,
  810. :repeated_string => ["a", "b", "c"],
  811. :repeated_int32 => [42, 43, 44],
  812. :repeated_enum => [:A, :B, :C, 100],
  813. :repeated_msg => [TestMessage2.new(:foo => 1),
  814. TestMessage2.new(:foo => 2)])
  815. data = TestMessage.encode m
  816. m2 = TestMessage.decode data
  817. assert m == m2
  818. data = Google::Protobuf.encode m
  819. m2 = Google::Protobuf.decode(TestMessage, data)
  820. assert m == m2
  821. end
  822. def test_encode_decode_helpers
  823. m = TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  824. assert_equal 'foo', m.optional_string
  825. assert_equal ['bar1', 'bar2'], m.repeated_string
  826. json = m.to_json
  827. m2 = TestMessage.decode_json(json)
  828. assert_equal 'foo', m2.optional_string
  829. assert_equal ['bar1', 'bar2'], m2.repeated_string
  830. if RUBY_PLATFORM != "java"
  831. assert m2.optional_string.frozen?
  832. assert m2.repeated_string[0].frozen?
  833. end
  834. proto = m.to_proto
  835. m2 = TestMessage.decode(proto)
  836. assert_equal 'foo', m2.optional_string
  837. assert_equal ['bar1', 'bar2'], m2.repeated_string
  838. end
  839. def test_protobuf_encode_decode_helpers
  840. m = TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  841. encoded_msg = Google::Protobuf.encode(m)
  842. assert_equal m.to_proto, encoded_msg
  843. decoded_msg = Google::Protobuf.decode(TestMessage, encoded_msg)
  844. assert_equal TestMessage.decode(m.to_proto), decoded_msg
  845. end
  846. def test_protobuf_encode_decode_json_helpers
  847. m = TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  848. encoded_msg = Google::Protobuf.encode_json(m)
  849. assert_equal m.to_json, encoded_msg
  850. decoded_msg = Google::Protobuf.decode_json(TestMessage, encoded_msg)
  851. assert_equal TestMessage.decode_json(m.to_json), decoded_msg
  852. end
  853. def test_to_h
  854. m = TestMessage.new(:optional_bool => true, :optional_double => -10.100001, :optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  855. expected_result = {
  856. :optional_bool=>true,
  857. :optional_bytes=>"",
  858. :optional_double=>-10.100001,
  859. :optional_enum=>:Default,
  860. :optional_float=>0.0,
  861. :optional_int32=>0,
  862. :optional_int64=>0,
  863. :optional_msg=>nil,
  864. :optional_string=>"foo",
  865. :optional_uint32=>0,
  866. :optional_uint64=>0,
  867. :repeated_bool=>[],
  868. :repeated_bytes=>[],
  869. :repeated_double=>[],
  870. :repeated_enum=>[],
  871. :repeated_float=>[],
  872. :repeated_int32=>[],
  873. :repeated_int64=>[],
  874. :repeated_msg=>[],
  875. :repeated_string=>["bar1", "bar2"],
  876. :repeated_uint32=>[],
  877. :repeated_uint64=>[]
  878. }
  879. assert_equal expected_result, m.to_h
  880. end
  881. def test_def_errors
  882. s = Google::Protobuf::DescriptorPool.new
  883. success = false
  884. begin
  885. s.build do
  886. # enum with no default (integer value 0)
  887. add_enum "MyEnum" do
  888. value :A, 1
  889. end
  890. end
  891. rescue TypeError
  892. success = true
  893. end
  894. assert(success)
  895. success = false
  896. begin
  897. s.build do
  898. # message with required field (unsupported in proto3)
  899. add_message "MyMessage" do
  900. required :foo, :int32, 1
  901. end
  902. end
  903. rescue TypeError
  904. success = true
  905. end
  906. assert(success)
  907. end
  908. def test_corecursive
  909. # just be sure that we can instantiate types with corecursive field-type
  910. # references.
  911. m = Recursive1.new(:foo => Recursive2.new(:foo => Recursive1.new))
  912. assert Recursive1.descriptor.lookup("foo").subtype ==
  913. Recursive2.descriptor
  914. assert Recursive2.descriptor.lookup("foo").subtype ==
  915. Recursive1.descriptor
  916. serialized = Recursive1.encode(m)
  917. m2 = Recursive1.decode(serialized)
  918. assert m == m2
  919. end
  920. def test_serialize_cycle
  921. m = Recursive1.new(:foo => Recursive2.new)
  922. m.foo.foo = m
  923. assert_raise RuntimeError do
  924. serialized = Recursive1.encode(m)
  925. end
  926. end
  927. def test_bad_field_names
  928. m = BadFieldNames.new(:dup => 1, :class => 2)
  929. m2 = m.dup
  930. assert m == m2
  931. assert m['dup'] == 1
  932. assert m['class'] == 2
  933. m['dup'] = 3
  934. assert m['dup'] == 3
  935. end
  936. def test_int_ranges
  937. m = TestMessage.new
  938. m.optional_int32 = 0
  939. m.optional_int32 = -0x8000_0000
  940. m.optional_int32 = +0x7fff_ffff
  941. m.optional_int32 = 1.0
  942. m.optional_int32 = -1.0
  943. m.optional_int32 = 2e9
  944. assert_raise RangeError do
  945. m.optional_int32 = -0x8000_0001
  946. end
  947. assert_raise RangeError do
  948. m.optional_int32 = +0x8000_0000
  949. end
  950. assert_raise RangeError do
  951. m.optional_int32 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  952. end
  953. assert_raise RangeError do
  954. m.optional_int32 = 1e12
  955. end
  956. assert_raise RangeError do
  957. m.optional_int32 = 1.5
  958. end
  959. m.optional_uint32 = 0
  960. m.optional_uint32 = +0xffff_ffff
  961. m.optional_uint32 = 1.0
  962. m.optional_uint32 = 4e9
  963. assert_raise RangeError do
  964. m.optional_uint32 = -1
  965. end
  966. assert_raise RangeError do
  967. m.optional_uint32 = -1.5
  968. end
  969. assert_raise RangeError do
  970. m.optional_uint32 = -1.5e12
  971. end
  972. assert_raise RangeError do
  973. m.optional_uint32 = -0x1000_0000_0000_0000
  974. end
  975. assert_raise RangeError do
  976. m.optional_uint32 = +0x1_0000_0000
  977. end
  978. assert_raise RangeError do
  979. m.optional_uint32 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  980. end
  981. assert_raise RangeError do
  982. m.optional_uint32 = 1e12
  983. end
  984. assert_raise RangeError do
  985. m.optional_uint32 = 1.5
  986. end
  987. m.optional_int64 = 0
  988. m.optional_int64 = -0x8000_0000_0000_0000
  989. m.optional_int64 = +0x7fff_ffff_ffff_ffff
  990. m.optional_int64 = 1.0
  991. m.optional_int64 = -1.0
  992. m.optional_int64 = 8e18
  993. m.optional_int64 = -8e18
  994. assert_raise RangeError do
  995. m.optional_int64 = -0x8000_0000_0000_0001
  996. end
  997. assert_raise RangeError do
  998. m.optional_int64 = +0x8000_0000_0000_0000
  999. end
  1000. assert_raise RangeError do
  1001. m.optional_int64 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  1002. end
  1003. assert_raise RangeError do
  1004. m.optional_int64 = 1e50
  1005. end
  1006. assert_raise RangeError do
  1007. m.optional_int64 = 1.5
  1008. end
  1009. m.optional_uint64 = 0
  1010. m.optional_uint64 = +0xffff_ffff_ffff_ffff
  1011. m.optional_uint64 = 1.0
  1012. m.optional_uint64 = 16e18
  1013. assert_raise RangeError do
  1014. m.optional_uint64 = -1
  1015. end
  1016. assert_raise RangeError do
  1017. m.optional_uint64 = -1.5
  1018. end
  1019. assert_raise RangeError do
  1020. m.optional_uint64 = -1.5e12
  1021. end
  1022. assert_raise RangeError do
  1023. m.optional_uint64 = -0x1_0000_0000_0000_0000
  1024. end
  1025. assert_raise RangeError do
  1026. m.optional_uint64 = +0x1_0000_0000_0000_0000
  1027. end
  1028. assert_raise RangeError do
  1029. m.optional_uint64 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  1030. end
  1031. assert_raise RangeError do
  1032. m.optional_uint64 = 1e50
  1033. end
  1034. assert_raise RangeError do
  1035. m.optional_uint64 = 1.5
  1036. end
  1037. end
  1038. def test_stress_test
  1039. m = TestMessage.new
  1040. m.optional_int32 = 42
  1041. m.optional_int64 = 0x100000000
  1042. m.optional_string = "hello world"
  1043. 10.times do m.repeated_msg.push TestMessage2.new(:foo => 42) end
  1044. 10.times do m.repeated_string.push "hello world" end
  1045. data = TestMessage.encode(m)
  1046. l = 0
  1047. 10_000.times do
  1048. m = TestMessage.decode(data)
  1049. data_new = TestMessage.encode(m)
  1050. assert data_new == data
  1051. data = data_new
  1052. end
  1053. end
  1054. def test_reflection
  1055. m = TestMessage.new(:optional_int32 => 1234)
  1056. msgdef = m.class.descriptor
  1057. assert msgdef.class == Google::Protobuf::Descriptor
  1058. assert msgdef.any? {|field| field.name == "optional_int32"}
  1059. optional_int32 = msgdef.lookup "optional_int32"
  1060. assert optional_int32.class == Google::Protobuf::FieldDescriptor
  1061. assert optional_int32 != nil
  1062. assert optional_int32.name == "optional_int32"
  1063. assert optional_int32.type == :int32
  1064. optional_int32.set(m, 5678)
  1065. assert m.optional_int32 == 5678
  1066. m.optional_int32 = 1000
  1067. assert optional_int32.get(m) == 1000
  1068. optional_msg = msgdef.lookup "optional_msg"
  1069. assert optional_msg.subtype == TestMessage2.descriptor
  1070. optional_msg.set(m, optional_msg.subtype.msgclass.new)
  1071. assert msgdef.msgclass == TestMessage
  1072. optional_enum = msgdef.lookup "optional_enum"
  1073. assert optional_enum.subtype == TestEnum.descriptor
  1074. assert optional_enum.subtype.class == Google::Protobuf::EnumDescriptor
  1075. optional_enum.subtype.each do |k, v|
  1076. # set with integer, check resolution to symbolic name
  1077. optional_enum.set(m, v)
  1078. assert optional_enum.get(m) == k
  1079. end
  1080. end
  1081. def test_json
  1082. # TODO: Fix JSON in JRuby version.
  1083. return if RUBY_PLATFORM == "java"
  1084. m = TestMessage.new(:optional_int32 => 1234,
  1085. :optional_int64 => -0x1_0000_0000,
  1086. :optional_uint32 => 0x8000_0000,
  1087. :optional_uint64 => 0xffff_ffff_ffff_ffff,
  1088. :optional_bool => true,
  1089. :optional_float => 1.0,
  1090. :optional_double => -1e100,
  1091. :optional_string => "Test string",
  1092. :optional_bytes => ["FFFFFFFF"].pack('H*'),
  1093. :optional_msg => TestMessage2.new(:foo => 42),
  1094. :repeated_int32 => [1, 2, 3, 4],
  1095. :repeated_string => ["a", "b", "c"],
  1096. :repeated_bool => [true, false, true, false],
  1097. :repeated_msg => [TestMessage2.new(:foo => 1),
  1098. TestMessage2.new(:foo => 2)])
  1099. json_text = TestMessage.encode_json(m)
  1100. m2 = TestMessage.decode_json(json_text)
  1101. assert m == m2
  1102. # Crash case from GitHub issue 283.
  1103. bar = Bar.new(msg: "bar")
  1104. baz1 = Baz.new(msg: "baz")
  1105. baz2 = Baz.new(msg: "quux")
  1106. Foo.encode_json(Foo.new)
  1107. Foo.encode_json(Foo.new(bar: bar))
  1108. Foo.encode_json(Foo.new(bar: bar, baz: [baz1, baz2]))
  1109. end
  1110. def test_json_maps
  1111. # TODO: Fix JSON in JRuby version.
  1112. return if RUBY_PLATFORM == "java"
  1113. m = MapMessage.new(:map_string_int32 => {"a" => 1})
  1114. expected = '{"mapStringInt32":{"a":1},"mapStringMsg":{}}'
  1115. expected_preserve = '{"map_string_int32":{"a":1},"map_string_msg":{}}'
  1116. assert_equal expected, MapMessage.encode_json(m, :emit_defaults => true)
  1117. json = MapMessage.encode_json(m, :preserve_proto_fieldnames => true, :emit_defaults => true)
  1118. assert_equal expected_preserve, json
  1119. m2 = MapMessage.decode_json(MapMessage.encode_json(m))
  1120. assert m == m2
  1121. end
  1122. end
  1123. end