basic.rb 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893
  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 "TestMessage" do
  9. optional :optional_int32, :int32, 1
  10. optional :optional_int64, :int64, 2
  11. optional :optional_uint32, :uint32, 3
  12. optional :optional_uint64, :uint64, 4
  13. optional :optional_bool, :bool, 5
  14. optional :optional_float, :float, 6
  15. optional :optional_double, :double, 7
  16. optional :optional_string, :string, 8
  17. optional :optional_bytes, :bytes, 9
  18. optional :optional_msg, :message, 10, "TestMessage2"
  19. optional :optional_enum, :enum, 11, "TestEnum"
  20. repeated :repeated_int32, :int32, 12
  21. repeated :repeated_int64, :int64, 13
  22. repeated :repeated_uint32, :uint32, 14
  23. repeated :repeated_uint64, :uint64, 15
  24. repeated :repeated_bool, :bool, 16
  25. repeated :repeated_float, :float, 17
  26. repeated :repeated_double, :double, 18
  27. repeated :repeated_string, :string, 19
  28. repeated :repeated_bytes, :bytes, 20
  29. repeated :repeated_msg, :message, 21, "TestMessage2"
  30. repeated :repeated_enum, :enum, 22, "TestEnum"
  31. end
  32. add_message "TestMessage2" do
  33. optional :foo, :int32, 1
  34. end
  35. add_message "Recursive1" do
  36. optional :foo, :message, 1, "Recursive2"
  37. end
  38. add_message "Recursive2" do
  39. optional :foo, :message, 1, "Recursive1"
  40. end
  41. add_enum "TestEnum" do
  42. value :Default, 0
  43. value :A, 1
  44. value :B, 2
  45. value :C, 3
  46. end
  47. add_message "BadFieldNames" do
  48. optional :dup, :int32, 1
  49. optional :class, :int32, 2
  50. optional :"a.b", :int32, 3
  51. end
  52. add_message "MapMessage" do
  53. map :map_string_int32, :string, :int32, 1
  54. map :map_string_msg, :string, :message, 2, "TestMessage2"
  55. end
  56. add_message "MapMessageWireEquiv" do
  57. repeated :map_string_int32, :message, 1, "MapMessageWireEquiv_entry1"
  58. repeated :map_string_msg, :message, 2, "MapMessageWireEquiv_entry2"
  59. end
  60. add_message "MapMessageWireEquiv_entry1" do
  61. optional :key, :string, 1
  62. optional :value, :int32, 2
  63. end
  64. add_message "MapMessageWireEquiv_entry2" do
  65. optional :key, :string, 1
  66. optional :value, :message, 2, "TestMessage2"
  67. end
  68. end
  69. TestMessage = pool.lookup("TestMessage").msgclass
  70. TestMessage2 = pool.lookup("TestMessage2").msgclass
  71. Recursive1 = pool.lookup("Recursive1").msgclass
  72. Recursive2 = pool.lookup("Recursive2").msgclass
  73. TestEnum = pool.lookup("TestEnum").enummodule
  74. BadFieldNames = pool.lookup("BadFieldNames").msgclass
  75. MapMessage = pool.lookup("MapMessage").msgclass
  76. MapMessageWireEquiv = pool.lookup("MapMessageWireEquiv").msgclass
  77. MapMessageWireEquiv_entry1 =
  78. pool.lookup("MapMessageWireEquiv_entry1").msgclass
  79. MapMessageWireEquiv_entry2 =
  80. pool.lookup("MapMessageWireEquiv_entry2").msgclass
  81. # ------------ test cases ---------------
  82. class MessageContainerTest < Test::Unit::TestCase
  83. def test_defaults
  84. m = TestMessage.new
  85. assert m.optional_int32 == 0
  86. assert m.optional_int64 == 0
  87. assert m.optional_uint32 == 0
  88. assert m.optional_uint64 == 0
  89. assert m.optional_bool == false
  90. assert m.optional_float == 0.0
  91. assert m.optional_double == 0.0
  92. assert m.optional_string == ""
  93. assert m.optional_bytes == ""
  94. assert m.optional_msg == nil
  95. assert m.optional_enum == :Default
  96. end
  97. def test_setters
  98. m = TestMessage.new
  99. m.optional_int32 = -42
  100. assert m.optional_int32 == -42
  101. m.optional_int64 = -0x1_0000_0000
  102. assert m.optional_int64 == -0x1_0000_0000
  103. m.optional_uint32 = 0x9000_0000
  104. assert m.optional_uint32 == 0x9000_0000
  105. m.optional_uint64 = 0x9000_0000_0000_0000
  106. assert m.optional_uint64 == 0x9000_0000_0000_0000
  107. m.optional_bool = true
  108. assert m.optional_bool == true
  109. m.optional_float = 0.5
  110. assert m.optional_float == 0.5
  111. m.optional_double = 0.5
  112. m.optional_string = "hello"
  113. assert m.optional_string == "hello"
  114. m.optional_bytes = "world".encode!('ASCII-8BIT')
  115. assert m.optional_bytes == "world"
  116. m.optional_msg = TestMessage2.new(:foo => 42)
  117. assert m.optional_msg == TestMessage2.new(:foo => 42)
  118. end
  119. def test_ctor_args
  120. m = TestMessage.new(:optional_int32 => -42,
  121. :optional_msg => TestMessage2.new,
  122. :optional_enum => :C,
  123. :repeated_string => ["hello", "there", "world"])
  124. assert m.optional_int32 == -42
  125. assert m.optional_msg.class == TestMessage2
  126. assert m.repeated_string.length == 3
  127. assert m.optional_enum == :C
  128. assert m.repeated_string[0] == "hello"
  129. assert m.repeated_string[1] == "there"
  130. assert m.repeated_string[2] == "world"
  131. end
  132. def test_inspect
  133. m = TestMessage.new(:optional_int32 => -42,
  134. :optional_enum => :A,
  135. :optional_msg => TestMessage2.new,
  136. :repeated_string => ["hello", "there", "world"])
  137. 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: []>'
  138. assert m.inspect == expected
  139. end
  140. def test_hash
  141. m1 = TestMessage.new(:optional_int32 => 42)
  142. m2 = TestMessage.new(:optional_int32 => 102)
  143. assert m1.hash != 0
  144. assert m2.hash != 0
  145. # relying on the randomness here -- if hash function changes and we are
  146. # unlucky enough to get a collision, then change the values above.
  147. assert m1.hash != m2.hash
  148. end
  149. def test_type_errors
  150. m = TestMessage.new
  151. assert_raise TypeError do
  152. m.optional_int32 = "hello"
  153. end
  154. assert_raise TypeError do
  155. m.optional_string = 42
  156. end
  157. assert_raise TypeError do
  158. m.optional_string = nil
  159. end
  160. assert_raise TypeError do
  161. m.optional_bool = 42
  162. end
  163. assert_raise TypeError do
  164. m.optional_msg = TestMessage.new # expects TestMessage2
  165. end
  166. assert_raise TypeError do
  167. m.repeated_int32 = [] # needs RepeatedField
  168. end
  169. assert_raise TypeError do
  170. m.repeated_int32.push "hello"
  171. end
  172. assert_raise TypeError do
  173. m.repeated_msg.push TestMessage.new
  174. end
  175. end
  176. def test_string_encoding
  177. m = TestMessage.new
  178. # Assigning a normal (ASCII or UTF8) string to a bytes field, or
  179. # ASCII-8BIT to a string field, raises an error.
  180. assert_raise TypeError do
  181. m.optional_bytes = "Test string ASCII".encode!('ASCII')
  182. end
  183. assert_raise TypeError do
  184. m.optional_bytes = "Test string UTF-8 \u0100".encode!('UTF-8')
  185. end
  186. assert_raise TypeError do
  187. m.optional_string = ["FFFF"].pack('H*')
  188. end
  189. # "Ordinary" use case.
  190. m.optional_bytes = ["FFFF"].pack('H*')
  191. m.optional_string = "\u0100"
  192. # strings are mutable so we can do this, but serialize should catch it.
  193. m.optional_string = "asdf".encode!('UTF-8')
  194. m.optional_string.encode!('ASCII-8BIT')
  195. assert_raise TypeError do
  196. data = TestMessage.encode(m)
  197. end
  198. end
  199. def test_rptfield_int32
  200. l = Google::Protobuf::RepeatedField.new(:int32)
  201. assert l.count == 0
  202. l = Google::Protobuf::RepeatedField.new(:int32, [1, 2, 3])
  203. assert l.count == 3
  204. assert_equal [1, 2, 3], l
  205. assert_equal l, [1, 2, 3]
  206. l.push 4
  207. assert l == [1, 2, 3, 4]
  208. dst_list = []
  209. l.each { |val| dst_list.push val }
  210. assert dst_list == [1, 2, 3, 4]
  211. assert l.to_a == [1, 2, 3, 4]
  212. assert l[0] == 1
  213. assert l[3] == 4
  214. l[0] = 5
  215. assert l == [5, 2, 3, 4]
  216. l2 = l.dup
  217. assert l == l2
  218. assert l.object_id != l2.object_id
  219. l2.push 6
  220. assert l.count == 4
  221. assert l2.count == 5
  222. assert l.inspect == '[5, 2, 3, 4]'
  223. l.insert(7, 8, 9)
  224. assert l == [5, 2, 3, 4, 7, 8, 9]
  225. assert l.pop == 9
  226. assert l == [5, 2, 3, 4, 7, 8]
  227. assert_raise TypeError do
  228. m = TestMessage.new
  229. l.push m
  230. end
  231. m = TestMessage.new
  232. m.repeated_int32 = l
  233. assert m.repeated_int32 == [5, 2, 3, 4, 7, 8]
  234. assert m.repeated_int32.object_id == l.object_id
  235. l.push 42
  236. assert m.repeated_int32.pop == 42
  237. l3 = l + l.dup
  238. assert l3.count == l.count * 2
  239. l.count.times do |i|
  240. assert l3[i] == l[i]
  241. assert l3[l.count + i] == l[i]
  242. end
  243. l.clear
  244. assert l.count == 0
  245. l += [1, 2, 3, 4]
  246. l.replace([5, 6, 7, 8])
  247. assert l == [5, 6, 7, 8]
  248. l4 = Google::Protobuf::RepeatedField.new(:int32)
  249. l4[5] = 42
  250. assert l4 == [0, 0, 0, 0, 0, 42]
  251. l4 << 100
  252. assert l4 == [0, 0, 0, 0, 0, 42, 100]
  253. l4 << 101 << 102
  254. assert l4 == [0, 0, 0, 0, 0, 42, 100, 101, 102]
  255. end
  256. def test_rptfield_msg
  257. l = Google::Protobuf::RepeatedField.new(:message, TestMessage)
  258. l.push TestMessage.new
  259. assert l.count == 1
  260. assert_raise TypeError do
  261. l.push TestMessage2.new
  262. end
  263. assert_raise TypeError do
  264. l.push 42
  265. end
  266. l2 = l.dup
  267. assert l2[0] == l[0]
  268. assert l2[0].object_id == l[0].object_id
  269. l2 = Google::Protobuf.deep_copy(l)
  270. assert l2[0] == l[0]
  271. assert l2[0].object_id != l[0].object_id
  272. l3 = l + l2
  273. assert l3.count == 2
  274. assert l3[0] == l[0]
  275. assert l3[1] == l2[0]
  276. l3[0].optional_int32 = 1000
  277. assert l[0].optional_int32 == 1000
  278. new_msg = TestMessage.new(:optional_int32 => 200)
  279. l4 = l + [new_msg]
  280. assert l4.count == 2
  281. new_msg.optional_int32 = 1000
  282. assert l4[1].optional_int32 == 1000
  283. end
  284. def test_rptfield_enum
  285. l = Google::Protobuf::RepeatedField.new(:enum, TestEnum)
  286. l.push :A
  287. l.push :B
  288. l.push :C
  289. assert l.count == 3
  290. assert_raise RangeError do
  291. l.push :D
  292. end
  293. assert l[0] == :A
  294. l.push 4
  295. assert l[3] == 4
  296. end
  297. def test_rptfield_initialize
  298. assert_raise ArgumentError do
  299. l = Google::Protobuf::RepeatedField.new
  300. end
  301. assert_raise ArgumentError do
  302. l = Google::Protobuf::RepeatedField.new(:message)
  303. end
  304. assert_raise ArgumentError do
  305. l = Google::Protobuf::RepeatedField.new([1, 2, 3])
  306. end
  307. assert_raise ArgumentError do
  308. l = Google::Protobuf::RepeatedField.new(:message, [TestMessage2.new])
  309. end
  310. end
  311. def test_map_basic
  312. # allowed key types:
  313. # :int32, :int64, :uint32, :uint64, :bool, :string, :bytes.
  314. m = Google::Protobuf::Map.new(:string, :int32)
  315. m["asdf"] = 1
  316. assert m["asdf"] == 1
  317. m["jkl;"] = 42
  318. assert m == { "jkl;" => 42, "asdf" => 1 }
  319. assert m.has_key?("asdf")
  320. assert !m.has_key?("qwerty")
  321. assert m.length == 2
  322. m2 = m.dup
  323. assert m == m2
  324. assert m.hash != 0
  325. assert m.hash == m2.hash
  326. collected = {}
  327. m.each { |k,v| collected[v] = k }
  328. assert collected == { 42 => "jkl;", 1 => "asdf" }
  329. assert m.delete("asdf") == 1
  330. assert !m.has_key?("asdf")
  331. assert m["asdf"] == nil
  332. assert !m.has_key?("asdf")
  333. # We only assert on inspect value when there is one map entry because the
  334. # order in which elements appear is unspecified (depends on the internal
  335. # hash function). We don't want a brittle test.
  336. assert m.inspect == "{\"jkl;\" => 42}"
  337. assert m.keys == ["jkl;"]
  338. assert m.values == [42]
  339. m.clear
  340. assert m.length == 0
  341. assert m == {}
  342. assert_raise TypeError do
  343. m[1] = 1
  344. end
  345. assert_raise RangeError do
  346. m["asdf"] = 0x1_0000_0000
  347. end
  348. end
  349. def test_map_ctor
  350. m = Google::Protobuf::Map.new(:string, :int32,
  351. {"a" => 1, "b" => 2, "c" => 3})
  352. assert m == {"a" => 1, "c" => 3, "b" => 2}
  353. end
  354. def test_map_keytypes
  355. m = Google::Protobuf::Map.new(:int32, :int32)
  356. m[1] = 42
  357. m[-1] = 42
  358. assert_raise RangeError do
  359. m[0x8000_0000] = 1
  360. end
  361. assert_raise TypeError do
  362. m["asdf"] = 1
  363. end
  364. m = Google::Protobuf::Map.new(:int64, :int32)
  365. m[0x1000_0000_0000_0000] = 1
  366. assert_raise RangeError do
  367. m[0x1_0000_0000_0000_0000] = 1
  368. end
  369. assert_raise TypeError do
  370. m["asdf"] = 1
  371. end
  372. m = Google::Protobuf::Map.new(:uint32, :int32)
  373. m[0x8000_0000] = 1
  374. assert_raise RangeError do
  375. m[0x1_0000_0000] = 1
  376. end
  377. assert_raise RangeError do
  378. m[-1] = 1
  379. end
  380. m = Google::Protobuf::Map.new(:uint64, :int32)
  381. m[0x8000_0000_0000_0000] = 1
  382. assert_raise RangeError do
  383. m[0x1_0000_0000_0000_0000] = 1
  384. end
  385. assert_raise RangeError do
  386. m[-1] = 1
  387. end
  388. m = Google::Protobuf::Map.new(:bool, :int32)
  389. m[true] = 1
  390. m[false] = 2
  391. assert_raise TypeError do
  392. m[1] = 1
  393. end
  394. assert_raise TypeError do
  395. m["asdf"] = 1
  396. end
  397. m = Google::Protobuf::Map.new(:string, :int32)
  398. m["asdf"] = 1
  399. assert_raise TypeError do
  400. m[1] = 1
  401. end
  402. assert_raise TypeError do
  403. bytestring = ["FFFF"].pack("H*")
  404. m[bytestring] = 1
  405. end
  406. m = Google::Protobuf::Map.new(:bytes, :int32)
  407. bytestring = ["FFFF"].pack("H*")
  408. m[bytestring] = 1
  409. assert_raise TypeError do
  410. m["asdf"] = 1
  411. end
  412. assert_raise TypeError do
  413. m[1] = 1
  414. end
  415. end
  416. def test_map_msg_enum_valuetypes
  417. m = Google::Protobuf::Map.new(:string, :message, TestMessage)
  418. m["asdf"] = TestMessage.new
  419. assert_raise TypeError do
  420. m["jkl;"] = TestMessage2.new
  421. end
  422. m = Google::Protobuf::Map.new(
  423. :string, :message, TestMessage,
  424. { "a" => TestMessage.new(:optional_int32 => 42),
  425. "b" => TestMessage.new(:optional_int32 => 84) })
  426. assert m.length == 2
  427. assert m.values.map{|msg| msg.optional_int32}.sort == [42, 84]
  428. m = Google::Protobuf::Map.new(:string, :enum, TestEnum,
  429. { "x" => :A, "y" => :B, "z" => :C })
  430. assert m.length == 3
  431. assert m["z"] == :C
  432. m["z"] = 2
  433. assert m["z"] == :B
  434. m["z"] = 4
  435. assert m["z"] == 4
  436. assert_raise RangeError do
  437. m["z"] = :Z
  438. end
  439. assert_raise TypeError do
  440. m["z"] = "z"
  441. end
  442. end
  443. def test_map_dup_deep_copy
  444. m = Google::Protobuf::Map.new(
  445. :string, :message, TestMessage,
  446. { "a" => TestMessage.new(:optional_int32 => 42),
  447. "b" => TestMessage.new(:optional_int32 => 84) })
  448. m2 = m.dup
  449. assert m == m2
  450. assert m.object_id != m2.object_id
  451. assert m["a"].object_id == m2["a"].object_id
  452. assert m["b"].object_id == m2["b"].object_id
  453. m2 = Google::Protobuf.deep_copy(m)
  454. assert m == m2
  455. assert m.object_id != m2.object_id
  456. assert m["a"].object_id != m2["a"].object_id
  457. assert m["b"].object_id != m2["b"].object_id
  458. end
  459. def test_map_field
  460. m = MapMessage.new
  461. assert m.map_string_int32 == {}
  462. assert m.map_string_msg == {}
  463. m = MapMessage.new(
  464. :map_string_int32 => {"a" => 1, "b" => 2},
  465. :map_string_msg => {"a" => TestMessage2.new(:foo => 1),
  466. "b" => TestMessage2.new(:foo => 2)})
  467. assert m.map_string_int32.keys.sort == ["a", "b"]
  468. assert m.map_string_int32["a"] == 1
  469. assert m.map_string_msg["b"].foo == 2
  470. m.map_string_int32["c"] = 3
  471. assert m.map_string_int32["c"] == 3
  472. m.map_string_msg["c"] = TestMessage2.new(:foo => 3)
  473. assert m.map_string_msg["c"] == TestMessage2.new(:foo => 3)
  474. m.map_string_msg.delete("b")
  475. m.map_string_msg.delete("c")
  476. assert m.map_string_msg == { "a" => TestMessage2.new(:foo => 1) }
  477. assert_raise TypeError do
  478. m.map_string_msg["e"] = TestMessage.new # wrong value type
  479. end
  480. # ensure nothing was added by the above
  481. assert m.map_string_msg == { "a" => TestMessage2.new(:foo => 1) }
  482. m.map_string_int32 = Google::Protobuf::Map.new(:string, :int32)
  483. assert_raise TypeError do
  484. m.map_string_int32 = Google::Protobuf::Map.new(:string, :int64)
  485. end
  486. assert_raise TypeError do
  487. m.map_string_int32 = {}
  488. end
  489. assert_raise TypeError do
  490. m = MapMessage.new(:map_string_int32 => { 1 => "I am not a number" })
  491. end
  492. end
  493. def test_map_encode_decode
  494. m = MapMessage.new(
  495. :map_string_int32 => {"a" => 1, "b" => 2},
  496. :map_string_msg => {"a" => TestMessage2.new(:foo => 1),
  497. "b" => TestMessage2.new(:foo => 2)})
  498. m2 = MapMessage.decode(MapMessage.encode(m))
  499. assert m == m2
  500. m3 = MapMessageWireEquiv.decode(MapMessage.encode(m))
  501. assert m3.map_string_int32.length == 2
  502. kv = {}
  503. m3.map_string_int32.map { |msg| kv[msg.key] = msg.value }
  504. assert kv == {"a" => 1, "b" => 2}
  505. kv = {}
  506. m3.map_string_msg.map { |msg| kv[msg.key] = msg.value }
  507. assert kv == {"a" => TestMessage2.new(:foo => 1),
  508. "b" => TestMessage2.new(:foo => 2)}
  509. end
  510. def test_enum_field
  511. m = TestMessage.new
  512. assert m.optional_enum == :Default
  513. m.optional_enum = :A
  514. assert m.optional_enum == :A
  515. assert_raise RangeError do
  516. m.optional_enum = :ASDF
  517. end
  518. m.optional_enum = 1
  519. assert m.optional_enum == :A
  520. m.optional_enum = 100
  521. assert m.optional_enum == 100
  522. end
  523. def test_dup
  524. m = TestMessage.new
  525. m.optional_string = "hello"
  526. m.optional_int32 = 42
  527. m.repeated_msg.push TestMessage2.new(:foo => 100)
  528. m.repeated_msg.push TestMessage2.new(:foo => 200)
  529. m2 = m.dup
  530. assert m == m2
  531. m.optional_int32 += 1
  532. assert m != m2
  533. assert m.repeated_msg[0] == m2.repeated_msg[0]
  534. assert m.repeated_msg[0].object_id == m2.repeated_msg[0].object_id
  535. end
  536. def test_deep_copy
  537. m = TestMessage.new(:optional_int32 => 42,
  538. :repeated_msg => [TestMessage2.new(:foo => 100)])
  539. m2 = Google::Protobuf.deep_copy(m)
  540. assert m == m2
  541. assert m.repeated_msg == m2.repeated_msg
  542. assert m.repeated_msg.object_id != m2.repeated_msg.object_id
  543. assert m.repeated_msg[0].object_id != m2.repeated_msg[0].object_id
  544. end
  545. def test_enum_lookup
  546. assert TestEnum::A == 1
  547. assert TestEnum::B == 2
  548. assert TestEnum::C == 3
  549. assert TestEnum::lookup(1) == :A
  550. assert TestEnum::lookup(2) == :B
  551. assert TestEnum::lookup(3) == :C
  552. assert TestEnum::resolve(:A) == 1
  553. assert TestEnum::resolve(:B) == 2
  554. assert TestEnum::resolve(:C) == 3
  555. end
  556. def test_parse_serialize
  557. m = TestMessage.new(:optional_int32 => 42,
  558. :optional_string => "hello world",
  559. :optional_enum => :B,
  560. :repeated_string => ["a", "b", "c"],
  561. :repeated_int32 => [42, 43, 44],
  562. :repeated_enum => [:A, :B, :C, 100],
  563. :repeated_msg => [TestMessage2.new(:foo => 1),
  564. TestMessage2.new(:foo => 2)])
  565. data = TestMessage.encode m
  566. m2 = TestMessage.decode data
  567. assert m == m2
  568. data = Google::Protobuf.encode m
  569. m2 = Google::Protobuf.decode(TestMessage, data)
  570. assert m == m2
  571. end
  572. def test_def_errors
  573. s = Google::Protobuf::DescriptorPool.new
  574. assert_raise TypeError do
  575. s.build do
  576. # enum with no default (integer value 0)
  577. add_enum "MyEnum" do
  578. value :A, 1
  579. end
  580. end
  581. end
  582. assert_raise TypeError do
  583. s.build do
  584. # message with required field (unsupported in proto3)
  585. add_message "MyMessage" do
  586. required :foo, :int32, 1
  587. end
  588. end
  589. end
  590. end
  591. def test_corecursive
  592. # just be sure that we can instantiate types with corecursive field-type
  593. # references.
  594. m = Recursive1.new(:foo => Recursive2.new(:foo => Recursive1.new))
  595. assert Recursive1.descriptor.lookup("foo").subtype ==
  596. Recursive2.descriptor
  597. assert Recursive2.descriptor.lookup("foo").subtype ==
  598. Recursive1.descriptor
  599. serialized = Recursive1.encode(m)
  600. m2 = Recursive1.decode(serialized)
  601. assert m == m2
  602. end
  603. def test_serialize_cycle
  604. m = Recursive1.new(:foo => Recursive2.new)
  605. m.foo.foo = m
  606. assert_raise RuntimeError do
  607. serialized = Recursive1.encode(m)
  608. end
  609. end
  610. def test_bad_field_names
  611. m = BadFieldNames.new(:dup => 1, :class => 2)
  612. m2 = m.dup
  613. assert m == m2
  614. assert m['dup'] == 1
  615. assert m['class'] == 2
  616. m['dup'] = 3
  617. assert m['dup'] == 3
  618. m['a.b'] = 4
  619. assert m['a.b'] == 4
  620. end
  621. def test_int_ranges
  622. m = TestMessage.new
  623. m.optional_int32 = 0
  624. m.optional_int32 = -0x8000_0000
  625. m.optional_int32 = +0x7fff_ffff
  626. m.optional_int32 = 1.0
  627. m.optional_int32 = -1.0
  628. m.optional_int32 = 2e9
  629. assert_raise RangeError do
  630. m.optional_int32 = -0x8000_0001
  631. end
  632. assert_raise RangeError do
  633. m.optional_int32 = +0x8000_0000
  634. end
  635. assert_raise RangeError do
  636. m.optional_int32 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  637. end
  638. assert_raise RangeError do
  639. m.optional_int32 = 1e12
  640. end
  641. assert_raise RangeError do
  642. m.optional_int32 = 1.5
  643. end
  644. m.optional_uint32 = 0
  645. m.optional_uint32 = +0xffff_ffff
  646. m.optional_uint32 = 1.0
  647. m.optional_uint32 = 4e9
  648. assert_raise RangeError do
  649. m.optional_uint32 = -1
  650. end
  651. assert_raise RangeError do
  652. m.optional_uint32 = -1.5
  653. end
  654. assert_raise RangeError do
  655. m.optional_uint32 = -1.5e12
  656. end
  657. assert_raise RangeError do
  658. m.optional_uint32 = -0x1000_0000_0000_0000
  659. end
  660. assert_raise RangeError do
  661. m.optional_uint32 = +0x1_0000_0000
  662. end
  663. assert_raise RangeError do
  664. m.optional_uint32 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  665. end
  666. assert_raise RangeError do
  667. m.optional_uint32 = 1e12
  668. end
  669. assert_raise RangeError do
  670. m.optional_uint32 = 1.5
  671. end
  672. m.optional_int64 = 0
  673. m.optional_int64 = -0x8000_0000_0000_0000
  674. m.optional_int64 = +0x7fff_ffff_ffff_ffff
  675. m.optional_int64 = 1.0
  676. m.optional_int64 = -1.0
  677. m.optional_int64 = 8e18
  678. m.optional_int64 = -8e18
  679. assert_raise RangeError do
  680. m.optional_int64 = -0x8000_0000_0000_0001
  681. end
  682. assert_raise RangeError do
  683. m.optional_int64 = +0x8000_0000_0000_0000
  684. end
  685. assert_raise RangeError do
  686. m.optional_int64 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  687. end
  688. assert_raise RangeError do
  689. m.optional_int64 = 1e50
  690. end
  691. assert_raise RangeError do
  692. m.optional_int64 = 1.5
  693. end
  694. m.optional_uint64 = 0
  695. m.optional_uint64 = +0xffff_ffff_ffff_ffff
  696. m.optional_uint64 = 1.0
  697. m.optional_uint64 = 16e18
  698. assert_raise RangeError do
  699. m.optional_uint64 = -1
  700. end
  701. assert_raise RangeError do
  702. m.optional_uint64 = -1.5
  703. end
  704. assert_raise RangeError do
  705. m.optional_uint64 = -1.5e12
  706. end
  707. assert_raise RangeError do
  708. m.optional_uint64 = -0x1_0000_0000_0000_0000
  709. end
  710. assert_raise RangeError do
  711. m.optional_uint64 = +0x1_0000_0000_0000_0000
  712. end
  713. assert_raise RangeError do
  714. m.optional_uint64 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  715. end
  716. assert_raise RangeError do
  717. m.optional_uint64 = 1e50
  718. end
  719. assert_raise RangeError do
  720. m.optional_uint64 = 1.5
  721. end
  722. end
  723. def test_stress_test
  724. m = TestMessage.new
  725. m.optional_int32 = 42
  726. m.optional_int64 = 0x100000000
  727. m.optional_string = "hello world"
  728. 10.times do m.repeated_msg.push TestMessage2.new(:foo => 42) end
  729. 10.times do m.repeated_string.push "hello world" end
  730. data = TestMessage.encode(m)
  731. l = 0
  732. 10_000.times do
  733. m = TestMessage.decode(data)
  734. data_new = TestMessage.encode(m)
  735. assert data_new == data
  736. data = data_new
  737. end
  738. end
  739. def test_reflection
  740. m = TestMessage.new(:optional_int32 => 1234)
  741. msgdef = m.class.descriptor
  742. assert msgdef.class == Google::Protobuf::Descriptor
  743. assert msgdef.any? {|field| field.name == "optional_int32"}
  744. optional_int32 = msgdef.lookup "optional_int32"
  745. assert optional_int32.class == Google::Protobuf::FieldDescriptor
  746. assert optional_int32 != nil
  747. assert optional_int32.name == "optional_int32"
  748. assert optional_int32.type == :int32
  749. optional_int32.set(m, 5678)
  750. assert m.optional_int32 == 5678
  751. m.optional_int32 = 1000
  752. assert optional_int32.get(m) == 1000
  753. optional_msg = msgdef.lookup "optional_msg"
  754. assert optional_msg.subtype == TestMessage2.descriptor
  755. optional_msg.set(m, optional_msg.subtype.msgclass.new)
  756. assert msgdef.msgclass == TestMessage
  757. optional_enum = msgdef.lookup "optional_enum"
  758. assert optional_enum.subtype == TestEnum.descriptor
  759. assert optional_enum.subtype.class == Google::Protobuf::EnumDescriptor
  760. optional_enum.subtype.each do |k, v|
  761. # set with integer, check resolution to symbolic name
  762. optional_enum.set(m, v)
  763. assert optional_enum.get(m) == k
  764. end
  765. end
  766. def test_json
  767. m = TestMessage.new(:optional_int32 => 1234,
  768. :optional_int64 => -0x1_0000_0000,
  769. :optional_uint32 => 0x8000_0000,
  770. :optional_uint64 => 0xffff_ffff_ffff_ffff,
  771. :optional_bool => true,
  772. :optional_float => 1.0,
  773. :optional_double => -1e100,
  774. :optional_string => "Test string",
  775. :optional_bytes => ["FFFFFFFF"].pack('H*'),
  776. :optional_msg => TestMessage2.new(:foo => 42),
  777. :repeated_int32 => [1, 2, 3, 4],
  778. :repeated_string => ["a", "b", "c"],
  779. :repeated_bool => [true, false, true, false],
  780. :repeated_msg => [TestMessage2.new(:foo => 1),
  781. TestMessage2.new(:foo => 2)])
  782. json_text = TestMessage.encode_json(m)
  783. m2 = TestMessage.decode_json(json_text)
  784. assert m == m2
  785. end
  786. end
  787. end