basic.rb 36 KB

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