basic.rb 37 KB

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