basic.rb 32 KB

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