common_tests.rb 50 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551
  1. require 'google/protobuf/wrappers_pb.rb'
  2. # Defines tests which are common between proto2 and proto3 syntax.
  3. #
  4. # Requires that the proto messages are exactly the same in proto2 and proto3 syntax
  5. # and that the including class should define a 'proto_module' method which returns
  6. # the enclosing module of the proto message classes.
  7. require 'bigdecimal'
  8. module CommonTests
  9. # Ruby 2.5 changed to raise FrozenError instead of RuntimeError
  10. FrozenErrorType = Gem::Version.new(RUBY_VERSION) < Gem::Version.new('2.5') ? RuntimeError : FrozenError
  11. def test_defaults
  12. m = proto_module::TestMessage.new
  13. assert m.optional_int32 == 0
  14. assert m.optional_int64 == 0
  15. assert m.optional_uint32 == 0
  16. assert m.optional_uint64 == 0
  17. assert m.optional_bool == false
  18. assert m.optional_float == 0.0
  19. assert m.optional_double == 0.0
  20. assert m.optional_string == ""
  21. assert m.optional_bytes == ""
  22. assert m.optional_msg == nil
  23. assert m.optional_enum == :Default
  24. end
  25. def test_setters
  26. m = proto_module::TestMessage.new
  27. m.optional_int32 = -42
  28. assert m.optional_int32 == -42
  29. m.optional_int64 = -0x1_0000_0000
  30. assert m.optional_int64 == -0x1_0000_0000
  31. m.optional_uint32 = 0x9000_0000
  32. assert m.optional_uint32 == 0x9000_0000
  33. m.optional_uint64 = 0x9000_0000_0000_0000
  34. assert m.optional_uint64 == 0x9000_0000_0000_0000
  35. m.optional_bool = true
  36. assert m.optional_bool == true
  37. m.optional_float = 0.5
  38. assert m.optional_float == 0.5
  39. m.optional_double = 0.5
  40. assert m.optional_double == 0.5
  41. m.optional_string = "hello"
  42. assert m.optional_string == "hello"
  43. m.optional_string = :hello
  44. assert m.optional_string == "hello"
  45. m.optional_bytes = "world".encode!('ASCII-8BIT')
  46. assert m.optional_bytes == "world"
  47. m.optional_msg = proto_module::TestMessage2.new(:foo => 42)
  48. assert m.optional_msg == proto_module::TestMessage2.new(:foo => 42)
  49. m.optional_msg = nil
  50. assert m.optional_msg == nil
  51. m.optional_enum = :C
  52. assert m.optional_enum == :C
  53. m.optional_enum = 'C'
  54. assert m.optional_enum == :C
  55. end
  56. def test_ctor_args
  57. m = proto_module::TestMessage.new(:optional_int32 => -42,
  58. :optional_msg => proto_module::TestMessage2.new,
  59. :optional_enum => :C,
  60. :repeated_string => ["hello", "there", "world"])
  61. assert m.optional_int32 == -42
  62. assert m.optional_msg.class == proto_module::TestMessage2
  63. assert m.repeated_string.length == 3
  64. assert m.optional_enum == :C
  65. assert m.repeated_string[0] == "hello"
  66. assert m.repeated_string[1] == "there"
  67. assert m.repeated_string[2] == "world"
  68. end
  69. def test_ctor_string_symbol_args
  70. m = proto_module::TestMessage.new(:optional_enum => 'C', :repeated_enum => ['A', 'B'])
  71. assert_equal :C, m.optional_enum
  72. assert_equal [:A, :B], m.repeated_enum
  73. m = proto_module::TestMessage.new(:optional_string => :foo, :repeated_string => [:foo, :bar])
  74. assert_equal 'foo', m.optional_string
  75. assert_equal ['foo', 'bar'], m.repeated_string
  76. end
  77. def test_ctor_nil_args
  78. m = proto_module::TestMessage.new(:optional_enum => nil, :optional_int32 => nil, :optional_string => nil, :optional_msg => nil)
  79. assert_equal :Default, m.optional_enum
  80. assert_equal 0, m.optional_int32
  81. assert_equal "", m.optional_string
  82. assert_nil m.optional_msg
  83. end
  84. def test_embeddedmsg_hash_init
  85. m = proto_module::TestEmbeddedMessageParent.new(
  86. :child_msg => {sub_child: {optional_int32: 1}},
  87. :number => 2,
  88. :repeated_msg => [{sub_child: {optional_int32: 3}}],
  89. :repeated_number => [10, 20, 30])
  90. assert_equal 2, m.number
  91. assert_equal [10, 20, 30], m.repeated_number
  92. assert_not_nil m.child_msg
  93. assert_not_nil m.child_msg.sub_child
  94. assert_equal m.child_msg.sub_child.optional_int32, 1
  95. assert_not_nil m.repeated_msg
  96. assert_equal 1, m.repeated_msg.length
  97. assert_equal 3, m.repeated_msg.first.sub_child.optional_int32
  98. end
  99. def test_inspect_eq_to_s
  100. m = proto_module::TestMessage.new(
  101. :optional_int32 => -42,
  102. :optional_enum => :A,
  103. :optional_msg => proto_module::TestMessage2.new,
  104. :repeated_string => ["hello", "there", "world"])
  105. expected = "<#{proto_module}::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: <#{proto_module}::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: []>"
  106. assert_equal expected, m.inspect
  107. assert_equal expected, m.to_s
  108. m = proto_module::OneofMessage.new(:b => -42)
  109. expected = "<#{proto_module}::OneofMessage: a: \"\", b: -42, c: nil, d: :Default>"
  110. assert_equal expected, m.inspect
  111. assert_equal expected, m.to_s
  112. end
  113. def test_hash
  114. m1 = proto_module::TestMessage.new(:optional_int32 => 42)
  115. m2 = proto_module::TestMessage.new(:optional_int32 => 102, repeated_string: ['please', 'work', 'ok?'])
  116. m3 = proto_module::TestMessage.new(:optional_int32 => 102, repeated_string: ['please', 'work', 'ok?'])
  117. assert m1.hash != 0
  118. assert m2.hash != 0
  119. assert m3.hash != 0
  120. # relying on the randomness here -- if hash function changes and we are
  121. # unlucky enough to get a collision, then change the values above.
  122. assert m1.hash != m2.hash
  123. assert_equal m2.hash, m3.hash
  124. end
  125. def test_unknown_field_errors
  126. e = assert_raise NoMethodError do
  127. proto_module::TestMessage.new.hello
  128. end
  129. assert_match(/hello/, e.message)
  130. e = assert_raise NoMethodError do
  131. proto_module::TestMessage.new.hello = "world"
  132. end
  133. assert_match(/hello/, e.message)
  134. end
  135. def test_type_errors
  136. m = proto_module::TestMessage.new
  137. e = assert_raise Google::Protobuf::TypeError do
  138. m.optional_int32 = "hello"
  139. end
  140. # Google::Protobuf::TypeError should inherit from TypeError for backwards compatibility
  141. # TODO: This can be removed when we can safely migrate to Google::Protobuf::TypeError
  142. assert e.is_a?(::TypeError)
  143. assert_raise Google::Protobuf::TypeError do
  144. m.optional_string = 42
  145. end
  146. assert_raise Google::Protobuf::TypeError do
  147. m.optional_string = nil
  148. end
  149. assert_raise Google::Protobuf::TypeError do
  150. m.optional_bool = 42
  151. end
  152. assert_raise Google::Protobuf::TypeError do
  153. m.optional_msg = proto_module::TestMessage.new # expects TestMessage2
  154. end
  155. assert_raise Google::Protobuf::TypeError do
  156. m.repeated_int32 = [] # needs RepeatedField
  157. end
  158. assert_raise Google::Protobuf::TypeError do
  159. m.repeated_int32.push "hello"
  160. end
  161. assert_raise Google::Protobuf::TypeError do
  162. m.repeated_msg.push proto_module::TestMessage.new
  163. end
  164. end
  165. def test_string_encoding
  166. m = proto_module::TestMessage.new
  167. # Assigning a normal (ASCII or UTF8) string to a bytes field, or
  168. # ASCII-8BIT to a string field will convert to the proper encoding.
  169. m.optional_bytes = "Test string ASCII".encode!('ASCII')
  170. assert m.optional_bytes.frozen?
  171. assert_equal Encoding::ASCII_8BIT, m.optional_bytes.encoding
  172. assert_equal "Test string ASCII", m.optional_bytes
  173. assert_raise Encoding::UndefinedConversionError do
  174. m.optional_bytes = "Test string UTF-8 \u0100".encode!('UTF-8')
  175. end
  176. assert_raise Encoding::UndefinedConversionError do
  177. m.optional_string = ["FFFF"].pack('H*')
  178. end
  179. # "Ordinary" use case.
  180. m.optional_bytes = ["FFFF"].pack('H*')
  181. m.optional_string = "\u0100"
  182. # strings are immutable so we can't do this, but serialize should catch it.
  183. m.optional_string = "asdf".encode!('UTF-8')
  184. assert_raise(FrozenErrorType) { m.optional_string.encode!('ASCII-8BIT') }
  185. end
  186. def test_rptfield_int32
  187. l = Google::Protobuf::RepeatedField.new(:int32)
  188. assert l.count == 0
  189. l = Google::Protobuf::RepeatedField.new(:int32, [1, 2, 3])
  190. assert l.count == 3
  191. assert_equal [1, 2, 3], l
  192. assert_equal l, [1, 2, 3]
  193. l.push 4
  194. assert l == [1, 2, 3, 4]
  195. dst_list = []
  196. l.each { |val| dst_list.push val }
  197. assert dst_list == [1, 2, 3, 4]
  198. assert l.to_a == [1, 2, 3, 4]
  199. assert l[0] == 1
  200. assert l[3] == 4
  201. l[0] = 5
  202. assert l == [5, 2, 3, 4]
  203. l2 = l.dup
  204. assert l == l2
  205. assert l.object_id != l2.object_id
  206. l2.push 6
  207. assert l.count == 4
  208. assert l2.count == 5
  209. assert l.inspect == '[5, 2, 3, 4]'
  210. l.concat([7, 8, 9])
  211. assert l == [5, 2, 3, 4, 7, 8, 9]
  212. assert l.pop == 9
  213. assert l == [5, 2, 3, 4, 7, 8]
  214. assert_raise Google::Protobuf::TypeError do
  215. m = proto_module::TestMessage.new
  216. l.push m
  217. end
  218. m = proto_module::TestMessage.new
  219. m.repeated_int32 = l
  220. assert m.repeated_int32 == [5, 2, 3, 4, 7, 8]
  221. assert m.repeated_int32.object_id == l.object_id
  222. l.push 42
  223. assert m.repeated_int32.pop == 42
  224. l3 = l + l.dup
  225. assert l3.count == l.count * 2
  226. l.count.times do |i|
  227. assert l3[i] == l[i]
  228. assert l3[l.count + i] == l[i]
  229. end
  230. l.clear
  231. assert l.count == 0
  232. l += [1, 2, 3, 4]
  233. l.replace([5, 6, 7, 8])
  234. assert l == [5, 6, 7, 8]
  235. l4 = Google::Protobuf::RepeatedField.new(:int32)
  236. l4[5] = 42
  237. assert l4 == [0, 0, 0, 0, 0, 42]
  238. l4 << 100
  239. assert l4 == [0, 0, 0, 0, 0, 42, 100]
  240. l4 << 101 << 102
  241. assert l4 == [0, 0, 0, 0, 0, 42, 100, 101, 102]
  242. end
  243. def test_parent_rptfield
  244. #make sure we set the RepeatedField and can add to it
  245. m = proto_module::TestMessage.new
  246. assert m.repeated_string == []
  247. m.repeated_string << 'ok'
  248. m.repeated_string.push('ok2')
  249. assert m.repeated_string == ['ok', 'ok2']
  250. m.repeated_string += ['ok3']
  251. assert m.repeated_string == ['ok', 'ok2', 'ok3']
  252. end
  253. def test_rptfield_msg
  254. l = Google::Protobuf::RepeatedField.new(:message, proto_module::TestMessage)
  255. l.push proto_module::TestMessage.new
  256. assert l.count == 1
  257. assert_raise Google::Protobuf::TypeError do
  258. l.push proto_module::TestMessage2.new
  259. end
  260. assert_raise Google::Protobuf::TypeError do
  261. l.push 42
  262. end
  263. l2 = l.dup
  264. assert l2[0] == l[0]
  265. assert l2[0].object_id == l[0].object_id
  266. l2 = Google::Protobuf.deep_copy(l)
  267. assert l2[0] == l[0]
  268. assert l2[0].object_id != l[0].object_id
  269. l3 = l + l2
  270. assert l3.count == 2
  271. assert l3[0] == l[0]
  272. assert l3[1] == l2[0]
  273. l3[0].optional_int32 = 1000
  274. assert l[0].optional_int32 == 1000
  275. new_msg = proto_module::TestMessage.new(:optional_int32 => 200)
  276. l4 = l + [new_msg]
  277. assert l4.count == 2
  278. new_msg.optional_int32 = 1000
  279. assert l4[1].optional_int32 == 1000
  280. end
  281. def test_rptfield_enum
  282. l = Google::Protobuf::RepeatedField.new(:enum, proto_module::TestEnum)
  283. l.push :A
  284. l.push :B
  285. l.push :C
  286. assert l.count == 3
  287. assert_raise RangeError do
  288. l.push :D
  289. end
  290. assert l[0] == :A
  291. l.push 4
  292. assert l[3] == 4
  293. end
  294. def test_rptfield_initialize
  295. assert_raise ArgumentError do
  296. Google::Protobuf::RepeatedField.new
  297. end
  298. assert_raise ArgumentError do
  299. Google::Protobuf::RepeatedField.new(:message)
  300. end
  301. assert_raise ArgumentError do
  302. Google::Protobuf::RepeatedField.new([1, 2, 3])
  303. end
  304. assert_raise ArgumentError do
  305. Google::Protobuf::RepeatedField.new(:message, [proto_module::TestMessage2.new])
  306. end
  307. end
  308. def test_rptfield_array_ducktyping
  309. l = Google::Protobuf::RepeatedField.new(:int32)
  310. length_methods = %w(count length size)
  311. length_methods.each do |lm|
  312. assert l.send(lm) == 0
  313. end
  314. # out of bounds returns a nil
  315. assert l[0] == nil
  316. assert l[1] == nil
  317. assert l[-1] == nil
  318. l.push 4
  319. length_methods.each do |lm|
  320. assert l.send(lm) == 1
  321. end
  322. assert l[0] == 4
  323. assert l[1] == nil
  324. assert l[-1] == 4
  325. assert l[-2] == nil
  326. l.push 2
  327. length_methods.each do |lm|
  328. assert l.send(lm) == 2
  329. end
  330. assert l[0] == 4
  331. assert l[1] == 2
  332. assert l[2] == nil
  333. assert l[-1] == 2
  334. assert l[-2] == 4
  335. assert l[-3] == nil
  336. #adding out of scope will backfill with empty objects
  337. end
  338. def test_map_basic
  339. # allowed key types:
  340. # :int32, :int64, :uint32, :uint64, :bool, :string, :bytes.
  341. m = Google::Protobuf::Map.new(:string, :int32)
  342. m["asdf"] = 1
  343. assert m["asdf"] == 1
  344. m["jkl;"] = 42
  345. assert m == { "jkl;" => 42, "asdf" => 1 }
  346. assert m.has_key?("asdf")
  347. assert !m.has_key?("qwerty")
  348. assert m.length == 2
  349. m2 = m.dup
  350. assert_equal m, m2
  351. assert m.hash != 0
  352. assert_equal m.hash, m2.hash
  353. collected = {}
  354. m.each { |k,v| collected[v] = k }
  355. assert collected == { 42 => "jkl;", 1 => "asdf" }
  356. assert m.delete("asdf") == 1
  357. assert !m.has_key?("asdf")
  358. assert m["asdf"] == nil
  359. assert !m.has_key?("asdf")
  360. # We only assert on inspect value when there is one map entry because the
  361. # order in which elements appear is unspecified (depends on the internal
  362. # hash function). We don't want a brittle test.
  363. assert m.inspect == "{\"jkl;\"=>42}"
  364. assert m.keys == ["jkl;"]
  365. assert m.values == [42]
  366. m.clear
  367. assert m.length == 0
  368. assert m == {}
  369. assert_raise TypeError do
  370. m[1] = 1
  371. end
  372. assert_raise RangeError do
  373. m["asdf"] = 0x1_0000_0000
  374. end
  375. end
  376. def test_map_ctor
  377. m = Google::Protobuf::Map.new(:string, :int32,
  378. {"a" => 1, "b" => 2, "c" => 3})
  379. assert m == {"a" => 1, "c" => 3, "b" => 2}
  380. end
  381. def test_map_keytypes
  382. m = Google::Protobuf::Map.new(:int32, :int32)
  383. m[1] = 42
  384. m[-1] = 42
  385. assert_raise RangeError do
  386. m[0x8000_0000] = 1
  387. end
  388. assert_raise Google::Protobuf::TypeError do
  389. m["asdf"] = 1
  390. end
  391. m = Google::Protobuf::Map.new(:int64, :int32)
  392. m[0x1000_0000_0000_0000] = 1
  393. assert_raise RangeError do
  394. m[0x1_0000_0000_0000_0000] = 1
  395. end
  396. assert_raise Google::Protobuf::TypeError do
  397. m["asdf"] = 1
  398. end
  399. m = Google::Protobuf::Map.new(:uint32, :int32)
  400. m[0x8000_0000] = 1
  401. assert_raise RangeError do
  402. m[0x1_0000_0000] = 1
  403. end
  404. assert_raise RangeError do
  405. m[-1] = 1
  406. end
  407. m = Google::Protobuf::Map.new(:uint64, :int32)
  408. m[0x8000_0000_0000_0000] = 1
  409. assert_raise RangeError do
  410. m[0x1_0000_0000_0000_0000] = 1
  411. end
  412. assert_raise RangeError do
  413. m[-1] = 1
  414. end
  415. m = Google::Protobuf::Map.new(:bool, :int32)
  416. m[true] = 1
  417. m[false] = 2
  418. assert_raise Google::Protobuf::TypeError do
  419. m[1] = 1
  420. end
  421. assert_raise Google::Protobuf::TypeError do
  422. m["asdf"] = 1
  423. end
  424. m = Google::Protobuf::Map.new(:string, :int32)
  425. m["asdf"] = 1
  426. assert_raise TypeError do
  427. m[1] = 1
  428. end
  429. assert_raise Encoding::UndefinedConversionError do
  430. bytestring = ["FFFF"].pack("H*")
  431. m[bytestring] = 1
  432. end
  433. m = Google::Protobuf::Map.new(:bytes, :int32)
  434. bytestring = ["FFFF"].pack("H*")
  435. m[bytestring] = 1
  436. # Allowed -- we will automatically convert to ASCII-8BIT.
  437. m["asdf"] = 1
  438. assert_raise TypeError do
  439. m[1] = 1
  440. end
  441. end
  442. def test_map_msg_enum_valuetypes
  443. m = Google::Protobuf::Map.new(:string, :message, proto_module::TestMessage)
  444. m["asdf"] = proto_module::TestMessage.new
  445. assert_raise Google::Protobuf::TypeError do
  446. m["jkl;"] = proto_module::TestMessage2.new
  447. end
  448. m = Google::Protobuf::Map.new(
  449. :string, :message, proto_module::TestMessage,
  450. { "a" => proto_module::TestMessage.new(:optional_int32 => 42),
  451. "b" => proto_module::TestMessage.new(:optional_int32 => 84) })
  452. assert m.length == 2
  453. assert m.values.map{|msg| msg.optional_int32}.sort == [42, 84]
  454. m = Google::Protobuf::Map.new(:string, :enum, proto_module::TestEnum,
  455. { "x" => :A, "y" => :B, "z" => :C })
  456. assert m.length == 3
  457. assert m["z"] == :C
  458. m["z"] = 2
  459. assert m["z"] == :B
  460. m["z"] = 4
  461. assert m["z"] == 4
  462. assert_raise RangeError do
  463. m["z"] = :Z
  464. end
  465. assert_raise RangeError do
  466. m["z"] = "z"
  467. end
  468. end
  469. def test_map_dup_deep_copy
  470. m = Google::Protobuf::Map.new(
  471. :string, :message, proto_module::TestMessage,
  472. { "a" => proto_module::TestMessage.new(:optional_int32 => 42),
  473. "b" => proto_module::TestMessage.new(:optional_int32 => 84) })
  474. m2 = m.dup
  475. assert m == m2
  476. assert m.object_id != m2.object_id
  477. assert m["a"].object_id == m2["a"].object_id
  478. assert m["b"].object_id == m2["b"].object_id
  479. m2 = Google::Protobuf.deep_copy(m)
  480. assert m == m2
  481. assert m.object_id != m2.object_id
  482. assert m["a"].object_id != m2["a"].object_id
  483. assert m["b"].object_id != m2["b"].object_id
  484. end
  485. def test_oneof_descriptors
  486. d = proto_module::OneofMessage.descriptor
  487. o = d.lookup_oneof("my_oneof")
  488. assert o != nil
  489. assert o.class == Google::Protobuf::OneofDescriptor
  490. assert o.name == "my_oneof"
  491. oneof_count = 0
  492. d.each_oneof{ |oneof|
  493. oneof_count += 1
  494. assert oneof == o
  495. }
  496. assert oneof_count == 1
  497. assert o.count == 4
  498. field_names = o.map{|f| f.name}.sort
  499. assert field_names == ["a", "b", "c", "d"]
  500. end
  501. def test_oneof
  502. d = proto_module::OneofMessage.new
  503. assert d.a == ""
  504. assert d.b == 0
  505. assert d.c == nil
  506. assert d.d == :Default
  507. assert d.my_oneof == nil
  508. d.a = "hi"
  509. assert d.a == "hi"
  510. assert d.b == 0
  511. assert d.c == nil
  512. assert d.d == :Default
  513. assert d.my_oneof == :a
  514. d.b = 42
  515. assert d.a == ""
  516. assert d.b == 42
  517. assert d.c == nil
  518. assert d.d == :Default
  519. assert d.my_oneof == :b
  520. d.c = proto_module::TestMessage2.new(:foo => 100)
  521. assert d.a == ""
  522. assert d.b == 0
  523. assert d.c.foo == 100
  524. assert d.d == :Default
  525. assert d.my_oneof == :c
  526. d.d = :C
  527. assert d.a == ""
  528. assert d.b == 0
  529. assert d.c == nil
  530. assert d.d == :C
  531. assert d.my_oneof == :d
  532. d2 = proto_module::OneofMessage.decode(proto_module::OneofMessage.encode(d))
  533. assert d2 == d
  534. encoded_field_a = proto_module::OneofMessage.encode(proto_module::OneofMessage.new(:a => "string"))
  535. encoded_field_b = proto_module::OneofMessage.encode(proto_module::OneofMessage.new(:b => 1000))
  536. encoded_field_c = proto_module::OneofMessage.encode(
  537. proto_module::OneofMessage.new(:c => proto_module::TestMessage2.new(:foo => 1)))
  538. encoded_field_d = proto_module::OneofMessage.encode(proto_module::OneofMessage.new(:d => :B))
  539. d3 = proto_module::OneofMessage.decode(
  540. encoded_field_c + encoded_field_a + encoded_field_b + encoded_field_d)
  541. assert d3.a == ""
  542. assert d3.b == 0
  543. assert d3.c == nil
  544. assert d3.d == :B
  545. d4 = proto_module::OneofMessage.decode(
  546. encoded_field_c + encoded_field_a + encoded_field_b + encoded_field_d +
  547. encoded_field_c)
  548. assert d4.a == ""
  549. assert d4.b == 0
  550. assert d4.c.foo == 1
  551. assert d4.d == :Default
  552. d5 = proto_module::OneofMessage.new(:a => "hello")
  553. assert d5.a == "hello"
  554. d5.a = nil
  555. assert d5.a == ""
  556. assert proto_module::OneofMessage.encode(d5) == ''
  557. assert d5.my_oneof == nil
  558. end
  559. def test_enum_field
  560. m = proto_module::TestMessage.new
  561. assert m.optional_enum == :Default
  562. m.optional_enum = :A
  563. assert m.optional_enum == :A
  564. assert_raise RangeError do
  565. m.optional_enum = :ASDF
  566. end
  567. m.optional_enum = 1
  568. assert m.optional_enum == :A
  569. m.optional_enum = 100
  570. assert m.optional_enum == 100
  571. end
  572. def test_dup
  573. m = proto_module::TestMessage.new
  574. m.optional_string = "hello"
  575. m.optional_int32 = 42
  576. tm1 = proto_module::TestMessage2.new(:foo => 100)
  577. tm2 = proto_module::TestMessage2.new(:foo => 200)
  578. m.repeated_msg.push tm1
  579. assert m.repeated_msg[-1] == tm1
  580. m.repeated_msg.push tm2
  581. assert m.repeated_msg[-1] == tm2
  582. m2 = m.dup
  583. assert m == m2
  584. m.optional_int32 += 1
  585. assert m != m2
  586. assert m.repeated_msg[0] == m2.repeated_msg[0]
  587. assert m.repeated_msg[0].object_id == m2.repeated_msg[0].object_id
  588. end
  589. def test_deep_copy
  590. m = proto_module::TestMessage.new(:optional_int32 => 42,
  591. :repeated_msg => [proto_module::TestMessage2.new(:foo => 100)])
  592. m2 = Google::Protobuf.deep_copy(m)
  593. assert m == m2
  594. assert m.repeated_msg == m2.repeated_msg
  595. assert m.repeated_msg.object_id != m2.repeated_msg.object_id
  596. assert m.repeated_msg[0].object_id != m2.repeated_msg[0].object_id
  597. end
  598. def test_eq
  599. m = proto_module::TestMessage.new(:optional_int32 => 42,
  600. :repeated_int32 => [1, 2, 3])
  601. m2 = proto_module::TestMessage.new(:optional_int32 => 43,
  602. :repeated_int32 => [1, 2, 3])
  603. assert m != m2
  604. end
  605. def test_enum_lookup
  606. assert proto_module::TestEnum::A == 1
  607. assert proto_module::TestEnum::B == 2
  608. assert proto_module::TestEnum::C == 3
  609. assert proto_module::TestEnum::lookup(1) == :A
  610. assert proto_module::TestEnum::lookup(2) == :B
  611. assert proto_module::TestEnum::lookup(3) == :C
  612. assert proto_module::TestEnum::resolve(:A) == 1
  613. assert proto_module::TestEnum::resolve(:B) == 2
  614. assert proto_module::TestEnum::resolve(:C) == 3
  615. end
  616. def test_enum_const_get_helpers
  617. m = proto_module::TestMessage.new
  618. assert_equal proto_module::TestEnum::Default, m.optional_enum_const
  619. assert_equal proto_module::TestEnum.const_get(:Default), m.optional_enum_const
  620. m = proto_module::TestMessage.new({optional_enum: proto_module::TestEnum::A})
  621. assert_equal proto_module::TestEnum::A, m.optional_enum_const
  622. assert_equal proto_module::TestEnum.const_get(:A), m.optional_enum_const
  623. m = proto_module::TestMessage.new({optional_enum: proto_module::TestEnum::B})
  624. assert_equal proto_module::TestEnum::B, m.optional_enum_const
  625. assert_equal proto_module::TestEnum.const_get(:B), m.optional_enum_const
  626. m = proto_module::TestMessage.new({optional_enum: proto_module::TestEnum::C})
  627. assert_equal proto_module::TestEnum::C, m.optional_enum_const
  628. assert_equal proto_module::TestEnum.const_get(:C), m.optional_enum_const
  629. m = proto_module::TestMessage2.new({foo: 2})
  630. assert_equal 2, m.foo
  631. assert_raise(NoMethodError) { m.foo_ }
  632. assert_raise(NoMethodError) { m.foo_X }
  633. assert_raise(NoMethodError) { m.foo_XX }
  634. assert_raise(NoMethodError) { m.foo_XXX }
  635. assert_raise(NoMethodError) { m.foo_XXXX }
  636. assert_raise(NoMethodError) { m.foo_XXXXX }
  637. assert_raise(NoMethodError) { m.foo_XXXXXX }
  638. m = proto_module::Enumer.new({optional_enum: :B})
  639. assert_equal :B, m.optional_enum
  640. assert_raise(NoMethodError) { m.optional_enum_ }
  641. assert_raise(NoMethodError) { m.optional_enum_X }
  642. assert_raise(NoMethodError) { m.optional_enum_XX }
  643. assert_raise(NoMethodError) { m.optional_enum_XXX }
  644. assert_raise(NoMethodError) { m.optional_enum_XXXX }
  645. assert_raise(NoMethodError) { m.optional_enum_XXXXX }
  646. assert_raise(NoMethodError) { m.optional_enum_XXXXXX }
  647. end
  648. def test_enum_getter
  649. m = proto_module::Enumer.new(:optional_enum => :B, :repeated_enum => [:A, :C])
  650. assert_equal :B, m.optional_enum
  651. assert_equal 2, m.optional_enum_const
  652. assert_equal proto_module::TestEnum::B, m.optional_enum_const
  653. assert_equal [:A, :C], m.repeated_enum
  654. assert_equal [1, 3], m.repeated_enum_const
  655. assert_equal [proto_module::TestEnum::A, proto_module::TestEnum::C], m.repeated_enum_const
  656. end
  657. def test_enum_getter_oneof
  658. m = proto_module::Enumer.new(:const => :C)
  659. assert_equal :C, m.const
  660. assert_equal 3, m.const_const
  661. assert_equal proto_module::TestEnum::C, m.const_const
  662. end
  663. def test_enum_getter_only_enums
  664. m = proto_module::Enumer.new(:optional_enum => :B, :a_const => 'thing')
  665. assert_equal 'thing', m.a_const
  666. assert_equal :B, m.optional_enum
  667. assert_raise(NoMethodError) { m.a }
  668. assert_raise(NoMethodError) { m.a_const_const }
  669. end
  670. def test_repeated_push
  671. m = proto_module::TestMessage.new
  672. m.repeated_string += ['one']
  673. m.repeated_string += %w[two three]
  674. assert_equal %w[one two three], m.repeated_string
  675. m.repeated_string.push *['four', 'five']
  676. assert_equal %w[one two three four five], m.repeated_string
  677. m.repeated_string.push 'six', 'seven'
  678. assert_equal %w[one two three four five six seven], m.repeated_string
  679. m = proto_module::TestMessage.new
  680. m.repeated_msg += [proto_module::TestMessage2.new(:foo => 1), proto_module::TestMessage2.new(:foo => 2)]
  681. m.repeated_msg += [proto_module::TestMessage2.new(:foo => 3)]
  682. m.repeated_msg.push proto_module::TestMessage2.new(:foo => 4), proto_module::TestMessage2.new(:foo => 5)
  683. assert_equal [1, 2, 3, 4, 5], m.repeated_msg.map {|x| x.foo}
  684. end
  685. def test_parse_serialize
  686. m = proto_module::TestMessage.new(:optional_int32 => 42,
  687. :optional_string => "hello world",
  688. :optional_enum => :B,
  689. :repeated_string => ["a", "b", "c"],
  690. :repeated_int32 => [42, 43, 44],
  691. :repeated_enum => [:A, :B, :C, 100],
  692. :repeated_msg => [proto_module::TestMessage2.new(:foo => 1),
  693. proto_module::TestMessage2.new(:foo => 2)])
  694. data = proto_module::TestMessage.encode m
  695. m2 = proto_module::TestMessage.decode data
  696. assert m == m2
  697. data = Google::Protobuf.encode m
  698. m2 = Google::Protobuf.decode(proto_module::TestMessage, data)
  699. assert m == m2
  700. end
  701. def test_encode_decode_helpers
  702. m = proto_module::TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  703. assert_equal 'foo', m.optional_string
  704. assert_equal ['bar1', 'bar2'], m.repeated_string
  705. json = m.to_json
  706. m2 = proto_module::TestMessage.decode_json(json)
  707. assert_equal 'foo', m2.optional_string
  708. assert_equal ['bar1', 'bar2'], m2.repeated_string
  709. if RUBY_PLATFORM != "java"
  710. assert m2.optional_string.frozen?
  711. assert m2.repeated_string[0].frozen?
  712. end
  713. proto = m.to_proto
  714. m2 = proto_module::TestMessage.decode(proto)
  715. assert_equal 'foo', m2.optional_string
  716. assert_equal ['bar1', 'bar2'], m2.repeated_string
  717. end
  718. def test_protobuf_encode_decode_helpers
  719. m = proto_module::TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  720. encoded_msg = Google::Protobuf.encode(m)
  721. assert_equal m.to_proto, encoded_msg
  722. decoded_msg = Google::Protobuf.decode(proto_module::TestMessage, encoded_msg)
  723. assert_equal proto_module::TestMessage.decode(m.to_proto), decoded_msg
  724. end
  725. def test_protobuf_encode_decode_json_helpers
  726. m = proto_module::TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  727. encoded_msg = Google::Protobuf.encode_json(m)
  728. assert_equal m.to_json, encoded_msg
  729. decoded_msg = Google::Protobuf.decode_json(proto_module::TestMessage, encoded_msg)
  730. assert_equal proto_module::TestMessage.decode_json(m.to_json), decoded_msg
  731. end
  732. def test_def_errors
  733. s = Google::Protobuf::DescriptorPool.new
  734. assert_raise Google::Protobuf::TypeError do
  735. s.build do
  736. # enum with no default (integer value 0)
  737. add_enum "MyEnum" do
  738. value :A, 1
  739. end
  740. end
  741. end
  742. assert_raise Google::Protobuf::TypeError do
  743. s.build do
  744. # message with required field (unsupported in proto3)
  745. add_message "MyMessage" do
  746. required :foo, :int32, 1
  747. end
  748. end
  749. end
  750. end
  751. def test_corecursive
  752. # just be sure that we can instantiate types with corecursive field-type
  753. # references.
  754. m = proto_module::Recursive1.new(:foo => proto_module::Recursive2.new(:foo => proto_module::Recursive1.new))
  755. assert proto_module::Recursive1.descriptor.lookup("foo").subtype ==
  756. proto_module::Recursive2.descriptor
  757. assert proto_module::Recursive2.descriptor.lookup("foo").subtype ==
  758. proto_module::Recursive1.descriptor
  759. serialized = proto_module::Recursive1.encode(m)
  760. m2 = proto_module::Recursive1.decode(serialized)
  761. assert m == m2
  762. end
  763. def test_serialize_cycle
  764. m = proto_module::Recursive1.new(:foo => proto_module::Recursive2.new)
  765. m.foo.foo = m
  766. assert_raise RuntimeError do
  767. proto_module::Recursive1.encode(m)
  768. end
  769. end
  770. def test_bad_field_names
  771. m = proto_module::BadFieldNames.new(:dup => 1, :class => 2)
  772. m2 = m.dup
  773. assert m == m2
  774. assert m['dup'] == 1
  775. assert m['class'] == 2
  776. m['dup'] = 3
  777. assert m['dup'] == 3
  778. m['a.b'] = 4
  779. assert m['a.b'] == 4
  780. end
  781. def test_int_ranges
  782. m = proto_module::TestMessage.new
  783. m.optional_int32 = 0
  784. m.optional_int32 = -0x8000_0000
  785. m.optional_int32 = +0x7fff_ffff
  786. m.optional_int32 = 1.0
  787. m.optional_int32 = -1.0
  788. m.optional_int32 = 2e9
  789. assert_raise RangeError do
  790. m.optional_int32 = -0x8000_0001
  791. end
  792. assert_raise RangeError do
  793. m.optional_int32 = +0x8000_0000
  794. end
  795. assert_raise RangeError do
  796. m.optional_int32 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  797. end
  798. assert_raise RangeError do
  799. m.optional_int32 = 1e12
  800. end
  801. assert_raise RangeError do
  802. m.optional_int32 = 1.5
  803. end
  804. m.optional_uint32 = 0
  805. m.optional_uint32 = +0xffff_ffff
  806. m.optional_uint32 = 1.0
  807. m.optional_uint32 = 4e9
  808. assert_raise RangeError do
  809. m.optional_uint32 = -1
  810. end
  811. assert_raise RangeError do
  812. m.optional_uint32 = -1.5
  813. end
  814. assert_raise RangeError do
  815. m.optional_uint32 = -1.5e12
  816. end
  817. assert_raise RangeError do
  818. m.optional_uint32 = -0x1000_0000_0000_0000
  819. end
  820. assert_raise RangeError do
  821. m.optional_uint32 = +0x1_0000_0000
  822. end
  823. assert_raise RangeError do
  824. m.optional_uint32 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  825. end
  826. assert_raise RangeError do
  827. m.optional_uint32 = 1e12
  828. end
  829. assert_raise RangeError do
  830. m.optional_uint32 = 1.5
  831. end
  832. m.optional_int64 = 0
  833. m.optional_int64 = -0x8000_0000_0000_0000
  834. m.optional_int64 = +0x7fff_ffff_ffff_ffff
  835. m.optional_int64 = 1.0
  836. m.optional_int64 = -1.0
  837. m.optional_int64 = 8e18
  838. m.optional_int64 = -8e18
  839. assert_raise RangeError do
  840. m.optional_int64 = -0x8000_0000_0000_0001
  841. end
  842. assert_raise RangeError do
  843. m.optional_int64 = +0x8000_0000_0000_0000
  844. end
  845. assert_raise RangeError do
  846. m.optional_int64 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  847. end
  848. assert_raise RangeError do
  849. m.optional_int64 = 1e50
  850. end
  851. assert_raise RangeError do
  852. m.optional_int64 = 1.5
  853. end
  854. m.optional_uint64 = 0
  855. m.optional_uint64 = +0xffff_ffff_ffff_ffff
  856. m.optional_uint64 = 1.0
  857. m.optional_uint64 = 16e18
  858. assert_raise RangeError do
  859. m.optional_uint64 = -1
  860. end
  861. assert_raise RangeError do
  862. m.optional_uint64 = -1.5
  863. end
  864. assert_raise RangeError do
  865. m.optional_uint64 = -1.5e12
  866. end
  867. assert_raise RangeError do
  868. m.optional_uint64 = -0x1_0000_0000_0000_0000
  869. end
  870. assert_raise RangeError do
  871. m.optional_uint64 = +0x1_0000_0000_0000_0000
  872. end
  873. assert_raise RangeError do
  874. m.optional_uint64 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  875. end
  876. assert_raise RangeError do
  877. m.optional_uint64 = 1e50
  878. end
  879. assert_raise RangeError do
  880. m.optional_uint64 = 1.5
  881. end
  882. end
  883. def test_stress_test
  884. m = proto_module::TestMessage.new
  885. m.optional_int32 = 42
  886. m.optional_int64 = 0x100000000
  887. m.optional_string = "hello world"
  888. 10.times do m.repeated_msg.push proto_module::TestMessage2.new(:foo => 42) end
  889. 10.times do m.repeated_string.push "hello world" end
  890. data = proto_module::TestMessage.encode(m)
  891. 10_000.times do
  892. m = proto_module::TestMessage.decode(data)
  893. data_new = proto_module::TestMessage.encode(m)
  894. assert data_new == data
  895. data = data_new
  896. end
  897. end
  898. def test_reflection
  899. m = proto_module::TestMessage.new(:optional_int32 => 1234)
  900. msgdef = m.class.descriptor
  901. assert msgdef.class == Google::Protobuf::Descriptor
  902. assert msgdef.any? {|field| field.name == "optional_int32"}
  903. optional_int32 = msgdef.lookup "optional_int32"
  904. assert optional_int32.class == Google::Protobuf::FieldDescriptor
  905. assert optional_int32 != nil
  906. assert optional_int32.name == "optional_int32"
  907. assert optional_int32.type == :int32
  908. optional_int32.set(m, 5678)
  909. assert m.optional_int32 == 5678
  910. m.optional_int32 = 1000
  911. assert optional_int32.get(m) == 1000
  912. optional_msg = msgdef.lookup "optional_msg"
  913. assert optional_msg.subtype == proto_module::TestMessage2.descriptor
  914. optional_msg.set(m, optional_msg.subtype.msgclass.new)
  915. assert msgdef.msgclass == proto_module::TestMessage
  916. optional_enum = msgdef.lookup "optional_enum"
  917. assert optional_enum.subtype == proto_module::TestEnum.descriptor
  918. assert optional_enum.subtype.class == Google::Protobuf::EnumDescriptor
  919. optional_enum.subtype.each do |k, v|
  920. # set with integer, check resolution to symbolic name
  921. optional_enum.set(m, v)
  922. assert optional_enum.get(m) == k
  923. end
  924. end
  925. def test_json
  926. # TODO: Fix JSON in JRuby version.
  927. return if RUBY_PLATFORM == "java"
  928. m = proto_module::TestMessage.new(:optional_int32 => 1234,
  929. :optional_int64 => -0x1_0000_0000,
  930. :optional_uint32 => 0x8000_0000,
  931. :optional_uint64 => 0xffff_ffff_ffff_ffff,
  932. :optional_bool => true,
  933. :optional_float => 1.0,
  934. :optional_double => -1e100,
  935. :optional_string => "Test string",
  936. :optional_bytes => ["FFFFFFFF"].pack('H*'),
  937. :optional_msg => proto_module::TestMessage2.new(:foo => 42),
  938. :repeated_int32 => [1, 2, 3, 4],
  939. :repeated_string => ["a", "b", "c"],
  940. :repeated_bool => [true, false, true, false],
  941. :repeated_msg => [proto_module::TestMessage2.new(:foo => 1),
  942. proto_module::TestMessage2.new(:foo => 2)])
  943. json_text = proto_module::TestMessage.encode_json(m)
  944. m2 = proto_module::TestMessage.decode_json(json_text)
  945. puts m.inspect
  946. puts m2.inspect
  947. assert m == m2
  948. # Crash case from GitHub issue 283.
  949. bar = proto_module::Bar.new(msg: "bar")
  950. baz1 = proto_module::Baz.new(msg: "baz")
  951. baz2 = proto_module::Baz.new(msg: "quux")
  952. proto_module::Foo.encode_json(proto_module::Foo.new)
  953. proto_module::Foo.encode_json(proto_module::Foo.new(bar: bar))
  954. proto_module::Foo.encode_json(proto_module::Foo.new(bar: bar, baz: [baz1, baz2]))
  955. end
  956. def test_json_empty
  957. assert proto_module::TestMessage.encode_json(proto_module::TestMessage.new) == '{}'
  958. end
  959. def test_json_emit_defaults
  960. # TODO: Fix JSON in JRuby version.
  961. return if RUBY_PLATFORM == "java"
  962. m = proto_module::TestMessage.new
  963. expected = {
  964. optionalInt32: 0,
  965. optionalInt64: 0,
  966. optionalUint32: 0,
  967. optionalUint64: 0,
  968. optionalBool: false,
  969. optionalFloat: 0,
  970. optionalDouble: 0,
  971. optionalString: "",
  972. optionalBytes: "",
  973. optionalEnum: "Default",
  974. repeatedInt32: [],
  975. repeatedInt64: [],
  976. repeatedUint32: [],
  977. repeatedUint64: [],
  978. repeatedBool: [],
  979. repeatedFloat: [],
  980. repeatedDouble: [],
  981. repeatedString: [],
  982. repeatedBytes: [],
  983. repeatedMsg: [],
  984. repeatedEnum: []
  985. }
  986. actual = proto_module::TestMessage.encode_json(m, :emit_defaults => true)
  987. assert JSON.parse(actual, :symbolize_names => true) == expected
  988. end
  989. def test_json_emit_defaults_submsg
  990. # TODO: Fix JSON in JRuby version.
  991. return if RUBY_PLATFORM == "java"
  992. m = proto_module::TestMessage.new(optional_msg: proto_module::TestMessage2.new)
  993. expected = {
  994. optionalInt32: 0,
  995. optionalInt64: 0,
  996. optionalUint32: 0,
  997. optionalUint64: 0,
  998. optionalBool: false,
  999. optionalFloat: 0,
  1000. optionalDouble: 0,
  1001. optionalString: "",
  1002. optionalBytes: "",
  1003. optionalMsg: {foo: 0},
  1004. optionalEnum: "Default",
  1005. repeatedInt32: [],
  1006. repeatedInt64: [],
  1007. repeatedUint32: [],
  1008. repeatedUint64: [],
  1009. repeatedBool: [],
  1010. repeatedFloat: [],
  1011. repeatedDouble: [],
  1012. repeatedString: [],
  1013. repeatedBytes: [],
  1014. repeatedMsg: [],
  1015. repeatedEnum: []
  1016. }
  1017. actual = proto_module::TestMessage.encode_json(m, :emit_defaults => true)
  1018. assert JSON.parse(actual, :symbolize_names => true) == expected
  1019. end
  1020. def test_json_emit_defaults_repeated_submsg
  1021. # TODO: Fix JSON in JRuby version.
  1022. return if RUBY_PLATFORM == "java"
  1023. m = proto_module::TestMessage.new(repeated_msg: [proto_module::TestMessage2.new])
  1024. expected = {
  1025. optionalInt32: 0,
  1026. optionalInt64: 0,
  1027. optionalUint32: 0,
  1028. optionalUint64: 0,
  1029. optionalBool: false,
  1030. optionalFloat: 0,
  1031. optionalDouble: 0,
  1032. optionalString: "",
  1033. optionalBytes: "",
  1034. optionalEnum: "Default",
  1035. repeatedInt32: [],
  1036. repeatedInt64: [],
  1037. repeatedUint32: [],
  1038. repeatedUint64: [],
  1039. repeatedBool: [],
  1040. repeatedFloat: [],
  1041. repeatedDouble: [],
  1042. repeatedString: [],
  1043. repeatedBytes: [],
  1044. repeatedMsg: [{foo: 0}],
  1045. repeatedEnum: []
  1046. }
  1047. actual = proto_module::TestMessage.encode_json(m, :emit_defaults => true)
  1048. assert JSON.parse(actual, :symbolize_names => true) == expected
  1049. end
  1050. def value_from_ruby(value)
  1051. ret = Google::Protobuf::Value.new
  1052. case value
  1053. when String
  1054. ret.string_value = value
  1055. when Google::Protobuf::Struct
  1056. ret.struct_value = value
  1057. when Hash
  1058. ret.struct_value = struct_from_ruby(value)
  1059. when Google::Protobuf::ListValue
  1060. ret.list_value = value
  1061. when Array
  1062. ret.list_value = list_from_ruby(value)
  1063. else
  1064. @log.error "Unknown type: #{value.class}"
  1065. raise Google::Protobuf::Error, "Unknown type: #{value.class}"
  1066. end
  1067. ret
  1068. end
  1069. def list_from_ruby(arr)
  1070. ret = Google::Protobuf::ListValue.new
  1071. arr.each do |v|
  1072. ret.values << value_from_ruby(v)
  1073. end
  1074. ret
  1075. end
  1076. def struct_from_ruby(hash)
  1077. ret = Google::Protobuf::Struct.new
  1078. hash.each do |k, v|
  1079. ret.fields[k] ||= value_from_ruby(v)
  1080. end
  1081. ret
  1082. end
  1083. def test_deep_json
  1084. # will not overflow
  1085. json = '{"a":{"a":{"a":{"a":{"a":{"a":{"a":{"a":{"a":{"a":{"a":'\
  1086. '{"a":{"a":{"a":{"a":{}}}}}}}}}}}}}}}}'
  1087. struct = struct_from_ruby(JSON.parse(json))
  1088. assert_equal json, struct.to_json
  1089. encoded = proto_module::MyRepeatedStruct.encode(
  1090. proto_module::MyRepeatedStruct.new(structs: [proto_module::MyStruct.new(struct: struct)]))
  1091. assert_equal json, proto_module::MyRepeatedStruct.decode(encoded).structs[0].struct.to_json
  1092. # will overflow
  1093. json = '{"a":{"a":{"a":[{"a":{"a":[{"a":[{"a":{"a":[{"a":[{"a":'\
  1094. '{"a":[{"a":[{"a":{"a":{"a":[{"a":"a"}]}}}]}]}}]}]}}]}]}}]}}}'
  1095. struct = struct_from_ruby(JSON.parse(json))
  1096. assert_equal json, struct.to_json
  1097. assert_raise(RuntimeError, "Maximum recursion depth exceeded during encoding") do
  1098. proto_module::MyRepeatedStruct.encode(
  1099. proto_module::MyRepeatedStruct.new(structs: [proto_module::MyStruct.new(struct: struct)]))
  1100. end
  1101. end
  1102. def test_comparison_with_arbitrary_object
  1103. assert proto_module::TestMessage.new != nil
  1104. end
  1105. def test_wrapper_getters
  1106. m = proto_module::Wrapper.new(
  1107. double: Google::Protobuf::DoubleValue.new(value: 2.0),
  1108. float: Google::Protobuf::FloatValue.new(value: 4.0),
  1109. int32: Google::Protobuf::Int32Value.new(value: 3),
  1110. int64: Google::Protobuf::Int64Value.new(value: 4),
  1111. uint32: Google::Protobuf::UInt32Value.new(value: 5),
  1112. uint64: Google::Protobuf::UInt64Value.new(value: 6),
  1113. bool: Google::Protobuf::BoolValue.new(value: true),
  1114. string: Google::Protobuf::StringValue.new(value: 'str'),
  1115. bytes: Google::Protobuf::BytesValue.new(value: 'fun'),
  1116. real_string: '100'
  1117. )
  1118. assert_equal 2.0, m.double_as_value
  1119. assert_equal 2.0, m.double.value
  1120. assert_equal 4.0, m.float_as_value
  1121. assert_equal 4.0, m.float.value
  1122. assert_equal 3, m.int32_as_value
  1123. assert_equal 3, m.int32.value
  1124. assert_equal 4, m.int64_as_value
  1125. assert_equal 4, m.int64.value
  1126. assert_equal 5, m.uint32_as_value
  1127. assert_equal 5, m.uint32.value
  1128. assert_equal 6, m.uint64_as_value
  1129. assert_equal 6, m.uint64.value
  1130. assert_equal true, m.bool_as_value
  1131. assert_equal true, m.bool.value
  1132. assert_equal 'str', m.string_as_value
  1133. assert_equal 'str', m.string.value
  1134. assert_equal 'fun', m.bytes_as_value
  1135. assert_equal 'fun', m.bytes.value
  1136. end
  1137. def test_wrapper_setters_as_value
  1138. m = proto_module::Wrapper.new
  1139. m.double_as_value = 4.8
  1140. assert_equal 4.8, m.double_as_value
  1141. assert_equal Google::Protobuf::DoubleValue.new(value: 4.8), m.double
  1142. m.float_as_value = 2.4
  1143. assert_in_delta 2.4, m.float_as_value
  1144. assert_in_delta Google::Protobuf::FloatValue.new(value: 2.4).value, m.float.value
  1145. m.int32_as_value = 5
  1146. assert_equal 5, m.int32_as_value
  1147. assert_equal Google::Protobuf::Int32Value.new(value: 5), m.int32
  1148. m.int64_as_value = 15
  1149. assert_equal 15, m.int64_as_value
  1150. assert_equal Google::Protobuf::Int64Value.new(value: 15), m.int64
  1151. m.uint32_as_value = 50
  1152. assert_equal 50, m.uint32_as_value
  1153. assert_equal Google::Protobuf::UInt32Value.new(value: 50), m.uint32
  1154. m.uint64_as_value = 500
  1155. assert_equal 500, m.uint64_as_value
  1156. assert_equal Google::Protobuf::UInt64Value.new(value: 500), m.uint64
  1157. m.bool_as_value = false
  1158. assert_equal false, m.bool_as_value
  1159. assert_equal Google::Protobuf::BoolValue.new(value: false), m.bool
  1160. m.string_as_value = 'xy'
  1161. assert_equal 'xy', m.string_as_value
  1162. assert_equal Google::Protobuf::StringValue.new(value: 'xy'), m.string
  1163. m.bytes_as_value = '123'
  1164. assert_equal '123', m.bytes_as_value
  1165. assert_equal Google::Protobuf::BytesValue.new(value: '123'), m.bytes
  1166. m.double_as_value = nil
  1167. assert_nil m.double
  1168. assert_nil m.double_as_value
  1169. m.float_as_value = nil
  1170. assert_nil m.float
  1171. assert_nil m.float_as_value
  1172. m.int32_as_value = nil
  1173. assert_nil m.int32
  1174. assert_nil m.int32_as_value
  1175. m.int64_as_value = nil
  1176. assert_nil m.int64
  1177. assert_nil m.int64_as_value
  1178. m.uint32_as_value = nil
  1179. assert_nil m.uint32
  1180. assert_nil m.uint32_as_value
  1181. m.uint64_as_value = nil
  1182. assert_nil m.uint64
  1183. assert_nil m.uint64_as_value
  1184. m.bool_as_value = nil
  1185. assert_nil m.bool
  1186. assert_nil m.bool_as_value
  1187. m.string_as_value = nil
  1188. assert_nil m.string
  1189. assert_nil m.string_as_value
  1190. m.bytes_as_value = nil
  1191. assert_nil m.bytes
  1192. assert_nil m.bytes_as_value
  1193. end
  1194. def test_wrapper_setters
  1195. m = proto_module::Wrapper.new
  1196. m.double = Google::Protobuf::DoubleValue.new(value: 4.8)
  1197. assert_equal 4.8, m.double_as_value
  1198. assert_equal Google::Protobuf::DoubleValue.new(value: 4.8), m.double
  1199. m.float = Google::Protobuf::FloatValue.new(value: 2.4)
  1200. assert_in_delta 2.4, m.float_as_value
  1201. assert_in_delta Google::Protobuf::FloatValue.new(value: 2.4).value, m.float.value
  1202. m.int32 = Google::Protobuf::Int32Value.new(value: 5)
  1203. assert_equal 5, m.int32_as_value
  1204. assert_equal Google::Protobuf::Int32Value.new(value: 5), m.int32
  1205. m.int64 = Google::Protobuf::Int64Value.new(value: 15)
  1206. assert_equal 15, m.int64_as_value
  1207. assert_equal Google::Protobuf::Int64Value.new(value: 15), m.int64
  1208. m.uint32 = Google::Protobuf::UInt32Value.new(value: 50)
  1209. assert_equal 50, m.uint32_as_value
  1210. assert_equal Google::Protobuf::UInt32Value.new(value: 50), m.uint32
  1211. m.uint64 = Google::Protobuf::UInt64Value.new(value: 500)
  1212. assert_equal 500, m.uint64_as_value
  1213. assert_equal Google::Protobuf::UInt64Value.new(value: 500), m.uint64
  1214. m.bool = Google::Protobuf::BoolValue.new(value: false)
  1215. assert_equal false, m.bool_as_value
  1216. assert_equal Google::Protobuf::BoolValue.new(value: false), m.bool
  1217. m.string = Google::Protobuf::StringValue.new(value: 'xy')
  1218. assert_equal 'xy', m.string_as_value
  1219. assert_equal Google::Protobuf::StringValue.new(value: 'xy'), m.string
  1220. m.bytes = Google::Protobuf::BytesValue.new(value: '123')
  1221. assert_equal '123', m.bytes_as_value
  1222. assert_equal Google::Protobuf::BytesValue.new(value: '123'), m.bytes
  1223. m.double = nil
  1224. assert_nil m.double
  1225. assert_nil m.double_as_value
  1226. m.float = nil
  1227. assert_nil m.float
  1228. assert_nil m.float_as_value
  1229. m.int32 = nil
  1230. assert_nil m.int32
  1231. assert_nil m.int32_as_value
  1232. m.int64 = nil
  1233. assert_nil m.int64
  1234. assert_nil m.int64_as_value
  1235. m.uint32 = nil
  1236. assert_nil m.uint32
  1237. assert_nil m.uint32_as_value
  1238. m.uint64 = nil
  1239. assert_nil m.uint64
  1240. assert_nil m.uint64_as_value
  1241. m.bool = nil
  1242. assert_nil m.bool
  1243. assert_nil m.bool_as_value
  1244. m.string = nil
  1245. assert_nil m.string
  1246. assert_nil m.string_as_value
  1247. m.bytes = nil
  1248. assert_nil m.bytes
  1249. assert_nil m.bytes_as_value
  1250. end
  1251. def test_wrappers_only
  1252. m = proto_module::Wrapper.new(real_string: 'hi', oneof_string: 'there')
  1253. assert_raise(NoMethodError) { m.real_string_as_value }
  1254. assert_raise(NoMethodError) { m.as_value }
  1255. assert_raise(NoMethodError) { m._as_value }
  1256. assert_raise(NoMethodError) { m.oneof_string_as_value }
  1257. m = proto_module::Wrapper.new
  1258. m.string_as_value = 'you'
  1259. assert_equal 'you', m.string.value
  1260. assert_equal 'you', m.string_as_value
  1261. assert_raise(NoMethodError) { m.string_ }
  1262. assert_raise(NoMethodError) { m.string_X }
  1263. assert_raise(NoMethodError) { m.string_XX }
  1264. assert_raise(NoMethodError) { m.string_XXX }
  1265. assert_raise(NoMethodError) { m.string_XXXX }
  1266. assert_raise(NoMethodError) { m.string_XXXXX }
  1267. assert_raise(NoMethodError) { m.string_XXXXXX }
  1268. assert_raise(NoMethodError) { m.string_XXXXXXX }
  1269. assert_raise(NoMethodError) { m.string_XXXXXXXX }
  1270. assert_raise(NoMethodError) { m.string_XXXXXXXXX }
  1271. assert_raise(NoMethodError) { m.string_XXXXXXXXXX }
  1272. end
  1273. def test_converts_time
  1274. m = proto_module::TimeMessage.new
  1275. m.timestamp = Google::Protobuf::Timestamp.new(seconds: 5, nanos: 6)
  1276. assert_kind_of Google::Protobuf::Timestamp, m.timestamp
  1277. assert_equal 5, m.timestamp.seconds
  1278. assert_equal 6, m.timestamp.nanos
  1279. m.timestamp = Time.at(9466, 123456.789)
  1280. assert_equal Google::Protobuf::Timestamp.new(seconds: 9466, nanos: 123456789), m.timestamp
  1281. m = proto_module::TimeMessage.new(timestamp: Time.at(1))
  1282. assert_equal Google::Protobuf::Timestamp.new(seconds: 1, nanos: 0), m.timestamp
  1283. assert_raise(Google::Protobuf::TypeError) { m.timestamp = 2 }
  1284. assert_raise(Google::Protobuf::TypeError) { m.timestamp = 2.4 }
  1285. assert_raise(Google::Protobuf::TypeError) { m.timestamp = '4' }
  1286. assert_raise(Google::Protobuf::TypeError) { m.timestamp = proto_module::TimeMessage.new }
  1287. end
  1288. def test_converts_duration
  1289. m = proto_module::TimeMessage.new
  1290. m.duration = Google::Protobuf::Duration.new(seconds: 2, nanos: 22)
  1291. assert_kind_of Google::Protobuf::Duration, m.duration
  1292. assert_equal 2, m.duration.seconds
  1293. assert_equal 22, m.duration.nanos
  1294. m.duration = 10.5
  1295. assert_equal Google::Protobuf::Duration.new(seconds: 10, nanos: 500_000_000), m.duration
  1296. m.duration = 200
  1297. assert_equal Google::Protobuf::Duration.new(seconds: 200, nanos: 0), m.duration
  1298. m.duration = Rational(3, 2)
  1299. assert_equal Google::Protobuf::Duration.new(seconds: 1, nanos: 500_000_000), m.duration
  1300. m.duration = BigDecimal.new("5")
  1301. assert_equal Google::Protobuf::Duration.new(seconds: 5, nanos: 0), m.duration
  1302. m = proto_module::TimeMessage.new(duration: 1.1)
  1303. assert_equal Google::Protobuf::Duration.new(seconds: 1, nanos: 100_000_000), m.duration
  1304. assert_raise(Google::Protobuf::TypeError) { m.duration = '2' }
  1305. assert_raise(Google::Protobuf::TypeError) { m.duration = proto_module::TimeMessage.new }
  1306. end
  1307. def test_freeze
  1308. m = proto_module::TestMessage.new
  1309. m.optional_int32 = 10
  1310. m.freeze
  1311. frozen_error = assert_raise(FrozenErrorType) { m.optional_int32 = 20 }
  1312. assert_equal "can't modify frozen #{proto_module}::TestMessage", frozen_error.message
  1313. assert_equal 10, m.optional_int32
  1314. assert_equal true, m.frozen?
  1315. assert_raise(FrozenErrorType) { m.optional_int64 = 2 }
  1316. assert_raise(FrozenErrorType) { m.optional_uint32 = 3 }
  1317. assert_raise(FrozenErrorType) { m.optional_uint64 = 4 }
  1318. assert_raise(FrozenErrorType) { m.optional_bool = true }
  1319. assert_raise(FrozenErrorType) { m.optional_float = 6.0 }
  1320. assert_raise(FrozenErrorType) { m.optional_double = 7.0 }
  1321. assert_raise(FrozenErrorType) { m.optional_string = '8' }
  1322. assert_raise(FrozenErrorType) { m.optional_bytes = nil }
  1323. assert_raise(FrozenErrorType) { m.optional_msg = proto_module::TestMessage2.new }
  1324. assert_raise(FrozenErrorType) { m.optional_enum = :A }
  1325. assert_raise(FrozenErrorType) { m.repeated_int32 = 1 }
  1326. assert_raise(FrozenErrorType) { m.repeated_int64 = 2 }
  1327. assert_raise(FrozenErrorType) { m.repeated_uint32 = 3 }
  1328. assert_raise(FrozenErrorType) { m.repeated_uint64 = 4 }
  1329. assert_raise(FrozenErrorType) { m.repeated_bool = true }
  1330. assert_raise(FrozenErrorType) { m.repeated_float = 6.0 }
  1331. assert_raise(FrozenErrorType) { m.repeated_double = 7.0 }
  1332. assert_raise(FrozenErrorType) { m.repeated_string = '8' }
  1333. assert_raise(FrozenErrorType) { m.repeated_bytes = nil }
  1334. assert_raise(FrozenErrorType) { m.repeated_msg = proto_module::TestMessage2.new }
  1335. assert_raise(FrozenErrorType) { m.repeated_enum = :A }
  1336. end
  1337. def test_eq
  1338. m1 = proto_module::TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  1339. m2 = proto_module::TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  1340. h = {}
  1341. h[m1] = :yes
  1342. assert m1 == m2
  1343. assert m1.eql?(m2)
  1344. assert m1.hash == m2.hash
  1345. assert h[m1] == :yes
  1346. assert h[m2] == :yes
  1347. m1.optional_int32 = 2
  1348. assert m1 != m2
  1349. assert !m1.eql?(m2)
  1350. assert m1.hash != m2.hash
  1351. assert_nil h[m2]
  1352. end
  1353. end