basic.rb 30 KB

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