common_tests.rb 50 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559
  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_equal 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. end
  779. def test_int_ranges
  780. m = proto_module::TestMessage.new
  781. m.optional_int32 = 0
  782. m.optional_int32 = -0x8000_0000
  783. m.optional_int32 = +0x7fff_ffff
  784. m.optional_int32 = 1.0
  785. m.optional_int32 = -1.0
  786. m.optional_int32 = 2e9
  787. assert_raise RangeError do
  788. m.optional_int32 = -0x8000_0001
  789. end
  790. assert_raise RangeError do
  791. m.optional_int32 = +0x8000_0000
  792. end
  793. assert_raise RangeError do
  794. m.optional_int32 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  795. end
  796. assert_raise RangeError do
  797. m.optional_int32 = 1e12
  798. end
  799. assert_raise RangeError do
  800. m.optional_int32 = 1.5
  801. end
  802. m.optional_uint32 = 0
  803. m.optional_uint32 = +0xffff_ffff
  804. m.optional_uint32 = 1.0
  805. m.optional_uint32 = 4e9
  806. assert_raise RangeError do
  807. m.optional_uint32 = -1
  808. end
  809. assert_raise RangeError do
  810. m.optional_uint32 = -1.5
  811. end
  812. assert_raise RangeError do
  813. m.optional_uint32 = -1.5e12
  814. end
  815. assert_raise RangeError do
  816. m.optional_uint32 = -0x1000_0000_0000_0000
  817. end
  818. assert_raise RangeError do
  819. m.optional_uint32 = +0x1_0000_0000
  820. end
  821. assert_raise RangeError do
  822. m.optional_uint32 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  823. end
  824. assert_raise RangeError do
  825. m.optional_uint32 = 1e12
  826. end
  827. assert_raise RangeError do
  828. m.optional_uint32 = 1.5
  829. end
  830. m.optional_int64 = 0
  831. m.optional_int64 = -0x8000_0000_0000_0000
  832. m.optional_int64 = +0x7fff_ffff_ffff_ffff
  833. m.optional_int64 = 1.0
  834. m.optional_int64 = -1.0
  835. m.optional_int64 = 8e18
  836. m.optional_int64 = -8e18
  837. assert_raise RangeError do
  838. m.optional_int64 = -0x8000_0000_0000_0001
  839. end
  840. assert_raise RangeError do
  841. m.optional_int64 = +0x8000_0000_0000_0000
  842. end
  843. assert_raise RangeError do
  844. m.optional_int64 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  845. end
  846. assert_raise RangeError do
  847. m.optional_int64 = 1e50
  848. end
  849. assert_raise RangeError do
  850. m.optional_int64 = 1.5
  851. end
  852. m.optional_uint64 = 0
  853. m.optional_uint64 = +0xffff_ffff_ffff_ffff
  854. m.optional_uint64 = 1.0
  855. m.optional_uint64 = 16e18
  856. assert_raise RangeError do
  857. m.optional_uint64 = -1
  858. end
  859. assert_raise RangeError do
  860. m.optional_uint64 = -1.5
  861. end
  862. assert_raise RangeError do
  863. m.optional_uint64 = -1.5e12
  864. end
  865. assert_raise RangeError do
  866. m.optional_uint64 = -0x1_0000_0000_0000_0000
  867. end
  868. assert_raise RangeError do
  869. m.optional_uint64 = +0x1_0000_0000_0000_0000
  870. end
  871. assert_raise RangeError do
  872. m.optional_uint64 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  873. end
  874. assert_raise RangeError do
  875. m.optional_uint64 = 1e50
  876. end
  877. assert_raise RangeError do
  878. m.optional_uint64 = 1.5
  879. end
  880. end
  881. def test_stress_test
  882. m = proto_module::TestMessage.new
  883. m.optional_int32 = 42
  884. m.optional_int64 = 0x100000000
  885. m.optional_string = "hello world"
  886. 10.times do m.repeated_msg.push proto_module::TestMessage2.new(:foo => 42) end
  887. 10.times do m.repeated_string.push "hello world" end
  888. data = proto_module::TestMessage.encode(m)
  889. 10_000.times do
  890. m = proto_module::TestMessage.decode(data)
  891. data_new = proto_module::TestMessage.encode(m)
  892. assert data_new == data
  893. data = data_new
  894. end
  895. end
  896. def test_reflection
  897. m = proto_module::TestMessage.new(:optional_int32 => 1234)
  898. msgdef = m.class.descriptor
  899. assert msgdef.class == Google::Protobuf::Descriptor
  900. assert msgdef.any? {|field| field.name == "optional_int32"}
  901. optional_int32 = msgdef.lookup "optional_int32"
  902. assert optional_int32.class == Google::Protobuf::FieldDescriptor
  903. assert optional_int32 != nil
  904. assert optional_int32.name == "optional_int32"
  905. assert optional_int32.type == :int32
  906. optional_int32.set(m, 5678)
  907. assert m.optional_int32 == 5678
  908. m.optional_int32 = 1000
  909. assert optional_int32.get(m) == 1000
  910. optional_msg = msgdef.lookup "optional_msg"
  911. assert optional_msg.subtype == proto_module::TestMessage2.descriptor
  912. optional_msg.set(m, optional_msg.subtype.msgclass.new)
  913. assert msgdef.msgclass == proto_module::TestMessage
  914. optional_enum = msgdef.lookup "optional_enum"
  915. assert optional_enum.subtype == proto_module::TestEnum.descriptor
  916. assert optional_enum.subtype.class == Google::Protobuf::EnumDescriptor
  917. optional_enum.subtype.each do |k, v|
  918. # set with integer, check resolution to symbolic name
  919. optional_enum.set(m, v)
  920. assert optional_enum.get(m) == k
  921. end
  922. end
  923. def test_json
  924. # TODO: Fix JSON in JRuby version.
  925. return if RUBY_PLATFORM == "java"
  926. m = proto_module::TestMessage.new(:optional_int32 => 1234,
  927. :optional_int64 => -0x1_0000_0000,
  928. :optional_uint32 => 0x8000_0000,
  929. :optional_uint64 => 0xffff_ffff_ffff_ffff,
  930. :optional_bool => true,
  931. :optional_float => 1.0,
  932. :optional_double => -1e100,
  933. :optional_string => "Test string",
  934. :optional_bytes => ["FFFFFFFF"].pack('H*'),
  935. :optional_msg => proto_module::TestMessage2.new(:foo => 42),
  936. :repeated_int32 => [1, 2, 3, 4],
  937. :repeated_string => ["a", "b", "c"],
  938. :repeated_bool => [true, false, true, false],
  939. :repeated_msg => [proto_module::TestMessage2.new(:foo => 1),
  940. proto_module::TestMessage2.new(:foo => 2)])
  941. json_text = proto_module::TestMessage.encode_json(m)
  942. m2 = proto_module::TestMessage.decode_json(json_text)
  943. assert_equal m, m2
  944. # Crash case from GitHub issue 283.
  945. bar = proto_module::Bar.new(msg: "bar")
  946. baz1 = proto_module::Baz.new(msg: "baz")
  947. baz2 = proto_module::Baz.new(msg: "quux")
  948. proto_module::Foo.encode_json(proto_module::Foo.new)
  949. proto_module::Foo.encode_json(proto_module::Foo.new(bar: bar))
  950. proto_module::Foo.encode_json(proto_module::Foo.new(bar: bar, baz: [baz1, baz2]))
  951. end
  952. def test_json_empty
  953. assert proto_module::TestMessage.encode_json(proto_module::TestMessage.new) == '{}'
  954. end
  955. def test_json_emit_defaults
  956. # TODO: Fix JSON in JRuby version.
  957. return if RUBY_PLATFORM == "java"
  958. m = proto_module::TestMessage.new
  959. expected = {
  960. optionalInt32: 0,
  961. optionalInt64: "0",
  962. optionalUint32: 0,
  963. optionalUint64: "0",
  964. optionalBool: false,
  965. optionalFloat: 0,
  966. optionalDouble: 0,
  967. optionalString: "",
  968. optionalBytes: "",
  969. optionalEnum: "Default",
  970. repeatedInt32: [],
  971. repeatedInt64: [],
  972. repeatedUint32: [],
  973. repeatedUint64: [],
  974. repeatedBool: [],
  975. repeatedFloat: [],
  976. repeatedDouble: [],
  977. repeatedString: [],
  978. repeatedBytes: [],
  979. repeatedMsg: [],
  980. repeatedEnum: []
  981. }
  982. actual = proto_module::TestMessage.encode_json(m, :emit_defaults => true)
  983. assert_equal expected, JSON.parse(actual, :symbolize_names => true)
  984. end
  985. def test_json_emit_defaults_submsg
  986. # TODO: Fix JSON in JRuby version.
  987. return if RUBY_PLATFORM == "java"
  988. m = proto_module::TestMessage.new(optional_msg: proto_module::TestMessage2.new)
  989. expected = {
  990. optionalInt32: 0,
  991. optionalInt64: "0",
  992. optionalUint32: 0,
  993. optionalUint64: "0",
  994. optionalBool: false,
  995. optionalFloat: 0,
  996. optionalDouble: 0,
  997. optionalString: "",
  998. optionalBytes: "",
  999. optionalMsg: {foo: 0},
  1000. optionalEnum: "Default",
  1001. repeatedInt32: [],
  1002. repeatedInt64: [],
  1003. repeatedUint32: [],
  1004. repeatedUint64: [],
  1005. repeatedBool: [],
  1006. repeatedFloat: [],
  1007. repeatedDouble: [],
  1008. repeatedString: [],
  1009. repeatedBytes: [],
  1010. repeatedMsg: [],
  1011. repeatedEnum: []
  1012. }
  1013. actual = proto_module::TestMessage.encode_json(m, :emit_defaults => true)
  1014. assert_equal expected, JSON.parse(actual, :symbolize_names => true)
  1015. end
  1016. def test_json_emit_defaults_repeated_submsg
  1017. # TODO: Fix JSON in JRuby version.
  1018. return if RUBY_PLATFORM == "java"
  1019. m = proto_module::TestMessage.new(repeated_msg: [proto_module::TestMessage2.new])
  1020. expected = {
  1021. optionalInt32: 0,
  1022. optionalInt64: "0",
  1023. optionalUint32: 0,
  1024. optionalUint64: "0",
  1025. optionalBool: false,
  1026. optionalFloat: 0,
  1027. optionalDouble: 0,
  1028. optionalString: "",
  1029. optionalBytes: "",
  1030. optionalEnum: "Default",
  1031. repeatedInt32: [],
  1032. repeatedInt64: [],
  1033. repeatedUint32: [],
  1034. repeatedUint64: [],
  1035. repeatedBool: [],
  1036. repeatedFloat: [],
  1037. repeatedDouble: [],
  1038. repeatedString: [],
  1039. repeatedBytes: [],
  1040. repeatedMsg: [{foo: 0}],
  1041. repeatedEnum: []
  1042. }
  1043. actual = proto_module::TestMessage.encode_json(m, :emit_defaults => true)
  1044. assert_equal expected, JSON.parse(actual, :symbolize_names => true)
  1045. end
  1046. def value_from_ruby(value)
  1047. ret = Google::Protobuf::Value.new
  1048. case value
  1049. when String
  1050. ret.string_value = value
  1051. when Google::Protobuf::Struct
  1052. ret.struct_value = value
  1053. when Hash
  1054. ret.struct_value = struct_from_ruby(value)
  1055. when Google::Protobuf::ListValue
  1056. ret.list_value = value
  1057. when Array
  1058. ret.list_value = list_from_ruby(value)
  1059. else
  1060. @log.error "Unknown type: #{value.class}"
  1061. raise Google::Protobuf::Error, "Unknown type: #{value.class}"
  1062. end
  1063. ret
  1064. end
  1065. def list_from_ruby(arr)
  1066. ret = Google::Protobuf::ListValue.new
  1067. arr.each do |v|
  1068. ret.values << value_from_ruby(v)
  1069. end
  1070. ret
  1071. end
  1072. def struct_from_ruby(hash)
  1073. ret = Google::Protobuf::Struct.new
  1074. hash.each do |k, v|
  1075. ret.fields[k] ||= value_from_ruby(v)
  1076. end
  1077. ret
  1078. end
  1079. def test_deep_json
  1080. # will not overflow
  1081. json = '{"a":{"a":{"a":{"a":{"a":{"a":{"a":{"a":{"a":{"a":{"a":'\
  1082. '{"a":{"a":{"a":{"a":{}}}}}}}}}}}}}}}}'
  1083. struct = struct_from_ruby(JSON.parse(json))
  1084. assert_equal json, struct.to_json
  1085. encoded = proto_module::MyRepeatedStruct.encode(
  1086. proto_module::MyRepeatedStruct.new(structs: [proto_module::MyStruct.new(struct: struct)]))
  1087. assert_equal json, proto_module::MyRepeatedStruct.decode(encoded).structs[0].struct.to_json
  1088. # will overflow
  1089. json = '{"a":{"a":{"a":[{"a":{"a":[{"a":[{"a":{"a":[{"a":[{"a":'\
  1090. '{"a":[{"a":[{"a":{"a":{"a":[{"a":"a"}]}}}]}]}}]}]}}]}]}}]}}}'
  1091. struct = struct_from_ruby(JSON.parse(json))
  1092. assert_equal json, struct.to_json
  1093. assert_raise(RuntimeError, "Maximum recursion depth exceeded during encoding") do
  1094. proto_module::MyRepeatedStruct.encode(
  1095. proto_module::MyRepeatedStruct.new(structs: [proto_module::MyStruct.new(struct: struct)]))
  1096. end
  1097. end
  1098. def test_comparison_with_arbitrary_object
  1099. assert proto_module::TestMessage.new != nil
  1100. end
  1101. def test_wrapper_getters
  1102. m = proto_module::Wrapper.new(
  1103. double: Google::Protobuf::DoubleValue.new(value: 2.0),
  1104. float: Google::Protobuf::FloatValue.new(value: 4.0),
  1105. int32: Google::Protobuf::Int32Value.new(value: 3),
  1106. int64: Google::Protobuf::Int64Value.new(value: 4),
  1107. uint32: Google::Protobuf::UInt32Value.new(value: 5),
  1108. uint64: Google::Protobuf::UInt64Value.new(value: 6),
  1109. bool: Google::Protobuf::BoolValue.new(value: true),
  1110. string: Google::Protobuf::StringValue.new(value: 'str'),
  1111. bytes: Google::Protobuf::BytesValue.new(value: 'fun'),
  1112. real_string: '100'
  1113. )
  1114. assert_equal 2.0, m.double_as_value
  1115. assert_equal 2.0, m.double.value
  1116. assert_equal 4.0, m.float_as_value
  1117. assert_equal 4.0, m.float.value
  1118. assert_equal 3, m.int32_as_value
  1119. assert_equal 3, m.int32.value
  1120. assert_equal 4, m.int64_as_value
  1121. assert_equal 4, m.int64.value
  1122. assert_equal 5, m.uint32_as_value
  1123. assert_equal 5, m.uint32.value
  1124. assert_equal 6, m.uint64_as_value
  1125. assert_equal 6, m.uint64.value
  1126. assert_equal true, m.bool_as_value
  1127. assert_equal true, m.bool.value
  1128. assert_equal 'str', m.string_as_value
  1129. assert_equal 'str', m.string.value
  1130. assert_equal 'fun', m.bytes_as_value
  1131. assert_equal 'fun', m.bytes.value
  1132. end
  1133. def test_wrapper_setters_as_value
  1134. m = proto_module::Wrapper.new
  1135. m.double_as_value = 4.8
  1136. assert_equal 4.8, m.double_as_value
  1137. assert_equal Google::Protobuf::DoubleValue.new(value: 4.8), m.double
  1138. m.float_as_value = 2.4
  1139. assert_in_delta 2.4, m.float_as_value
  1140. assert_in_delta Google::Protobuf::FloatValue.new(value: 2.4).value, m.float.value
  1141. m.int32_as_value = 5
  1142. assert_equal 5, m.int32_as_value
  1143. assert_equal Google::Protobuf::Int32Value.new(value: 5), m.int32
  1144. m.int64_as_value = 15
  1145. assert_equal 15, m.int64_as_value
  1146. assert_equal Google::Protobuf::Int64Value.new(value: 15), m.int64
  1147. m.uint32_as_value = 50
  1148. assert_equal 50, m.uint32_as_value
  1149. assert_equal Google::Protobuf::UInt32Value.new(value: 50), m.uint32
  1150. m.uint64_as_value = 500
  1151. assert_equal 500, m.uint64_as_value
  1152. assert_equal Google::Protobuf::UInt64Value.new(value: 500), m.uint64
  1153. m.bool_as_value = false
  1154. assert_equal false, m.bool_as_value
  1155. assert_equal Google::Protobuf::BoolValue.new(value: false), m.bool
  1156. m.string_as_value = 'xy'
  1157. assert_equal 'xy', m.string_as_value
  1158. assert_equal Google::Protobuf::StringValue.new(value: 'xy'), m.string
  1159. m.bytes_as_value = '123'
  1160. assert_equal '123', m.bytes_as_value
  1161. assert_equal Google::Protobuf::BytesValue.new(value: '123'), m.bytes
  1162. m.double_as_value = nil
  1163. assert_nil m.double
  1164. assert_nil m.double_as_value
  1165. m.float_as_value = nil
  1166. assert_nil m.float
  1167. assert_nil m.float_as_value
  1168. m.int32_as_value = nil
  1169. assert_nil m.int32
  1170. assert_nil m.int32_as_value
  1171. m.int64_as_value = nil
  1172. assert_nil m.int64
  1173. assert_nil m.int64_as_value
  1174. m.uint32_as_value = nil
  1175. assert_nil m.uint32
  1176. assert_nil m.uint32_as_value
  1177. m.uint64_as_value = nil
  1178. assert_nil m.uint64
  1179. assert_nil m.uint64_as_value
  1180. m.bool_as_value = nil
  1181. assert_nil m.bool
  1182. assert_nil m.bool_as_value
  1183. m.string_as_value = nil
  1184. assert_nil m.string
  1185. assert_nil m.string_as_value
  1186. m.bytes_as_value = nil
  1187. assert_nil m.bytes
  1188. assert_nil m.bytes_as_value
  1189. end
  1190. def test_wrapper_setters
  1191. m = proto_module::Wrapper.new
  1192. m.double = Google::Protobuf::DoubleValue.new(value: 4.8)
  1193. assert_equal 4.8, m.double_as_value
  1194. assert_equal Google::Protobuf::DoubleValue.new(value: 4.8), m.double
  1195. m.float = Google::Protobuf::FloatValue.new(value: 2.4)
  1196. assert_in_delta 2.4, m.float_as_value
  1197. assert_in_delta Google::Protobuf::FloatValue.new(value: 2.4).value, m.float.value
  1198. m.int32 = Google::Protobuf::Int32Value.new(value: 5)
  1199. assert_equal 5, m.int32_as_value
  1200. assert_equal Google::Protobuf::Int32Value.new(value: 5), m.int32
  1201. m.int64 = Google::Protobuf::Int64Value.new(value: 15)
  1202. assert_equal 15, m.int64_as_value
  1203. assert_equal Google::Protobuf::Int64Value.new(value: 15), m.int64
  1204. m.uint32 = Google::Protobuf::UInt32Value.new(value: 50)
  1205. assert_equal 50, m.uint32_as_value
  1206. assert_equal Google::Protobuf::UInt32Value.new(value: 50), m.uint32
  1207. m.uint64 = Google::Protobuf::UInt64Value.new(value: 500)
  1208. assert_equal 500, m.uint64_as_value
  1209. assert_equal Google::Protobuf::UInt64Value.new(value: 500), m.uint64
  1210. m.bool = Google::Protobuf::BoolValue.new(value: false)
  1211. assert_equal false, m.bool_as_value
  1212. assert_equal Google::Protobuf::BoolValue.new(value: false), m.bool
  1213. m.string = Google::Protobuf::StringValue.new(value: 'xy')
  1214. assert_equal 'xy', m.string_as_value
  1215. assert_equal Google::Protobuf::StringValue.new(value: 'xy'), m.string
  1216. m.bytes = Google::Protobuf::BytesValue.new(value: '123')
  1217. assert_equal '123', m.bytes_as_value
  1218. assert_equal Google::Protobuf::BytesValue.new(value: '123'), m.bytes
  1219. m.double = nil
  1220. assert_nil m.double
  1221. assert_nil m.double_as_value
  1222. m.float = nil
  1223. assert_nil m.float
  1224. assert_nil m.float_as_value
  1225. m.int32 = nil
  1226. assert_nil m.int32
  1227. assert_nil m.int32_as_value
  1228. m.int64 = nil
  1229. assert_nil m.int64
  1230. assert_nil m.int64_as_value
  1231. m.uint32 = nil
  1232. assert_nil m.uint32
  1233. assert_nil m.uint32_as_value
  1234. m.uint64 = nil
  1235. assert_nil m.uint64
  1236. assert_nil m.uint64_as_value
  1237. m.bool = nil
  1238. assert_nil m.bool
  1239. assert_nil m.bool_as_value
  1240. m.string = nil
  1241. assert_nil m.string
  1242. assert_nil m.string_as_value
  1243. m.bytes = nil
  1244. assert_nil m.bytes
  1245. assert_nil m.bytes_as_value
  1246. end
  1247. def test_wrappers_only
  1248. m = proto_module::Wrapper.new(real_string: 'hi', oneof_string: 'there')
  1249. assert_raise(NoMethodError) { m.real_string_as_value }
  1250. assert_raise(NoMethodError) { m.as_value }
  1251. assert_raise(NoMethodError) { m._as_value }
  1252. assert_raise(NoMethodError) { m.oneof_string_as_value }
  1253. m = proto_module::Wrapper.new
  1254. m.string_as_value = 'you'
  1255. assert_equal 'you', m.string.value
  1256. assert_equal 'you', m.string_as_value
  1257. assert_raise(NoMethodError) { m.string_ }
  1258. assert_raise(NoMethodError) { m.string_X }
  1259. assert_raise(NoMethodError) { m.string_XX }
  1260. assert_raise(NoMethodError) { m.string_XXX }
  1261. assert_raise(NoMethodError) { m.string_XXXX }
  1262. assert_raise(NoMethodError) { m.string_XXXXX }
  1263. assert_raise(NoMethodError) { m.string_XXXXXX }
  1264. assert_raise(NoMethodError) { m.string_XXXXXXX }
  1265. assert_raise(NoMethodError) { m.string_XXXXXXXX }
  1266. assert_raise(NoMethodError) { m.string_XXXXXXXXX }
  1267. assert_raise(NoMethodError) { m.string_XXXXXXXXXX }
  1268. end
  1269. def test_converts_time
  1270. m = proto_module::TimeMessage.new
  1271. m.timestamp = Google::Protobuf::Timestamp.new(seconds: 5, nanos: 6)
  1272. assert_kind_of Google::Protobuf::Timestamp, m.timestamp
  1273. assert_equal 5, m.timestamp.seconds
  1274. assert_equal 6, m.timestamp.nanos
  1275. m.timestamp = Time.at(9466, 123456.789)
  1276. assert_equal Google::Protobuf::Timestamp.new(seconds: 9466, nanos: 123456789), m.timestamp
  1277. m = proto_module::TimeMessage.new(timestamp: Time.at(1))
  1278. assert_equal Google::Protobuf::Timestamp.new(seconds: 1, nanos: 0), m.timestamp
  1279. assert_raise(Google::Protobuf::TypeError) { m.timestamp = 2 }
  1280. assert_raise(Google::Protobuf::TypeError) { m.timestamp = 2.4 }
  1281. assert_raise(Google::Protobuf::TypeError) { m.timestamp = '4' }
  1282. assert_raise(Google::Protobuf::TypeError) { m.timestamp = proto_module::TimeMessage.new }
  1283. def test_time(year, month, day)
  1284. str = ("\"%04d-%02d-%02dT00:00:00.000+00:00\"" % [year, month, day])
  1285. t = Google::Protobuf::Timestamp.decode_json(str)
  1286. time = Time.new(year, month, day, 0, 0, 0, "+00:00")
  1287. assert_equal t.seconds, time.to_i
  1288. end
  1289. (1970..2010).each do |year|
  1290. test_time(year, 2, 28)
  1291. test_time(year, 3, 01)
  1292. end
  1293. end
  1294. def test_converts_duration
  1295. m = proto_module::TimeMessage.new
  1296. m.duration = Google::Protobuf::Duration.new(seconds: 2, nanos: 22)
  1297. assert_kind_of Google::Protobuf::Duration, m.duration
  1298. assert_equal 2, m.duration.seconds
  1299. assert_equal 22, m.duration.nanos
  1300. m.duration = 10.5
  1301. assert_equal Google::Protobuf::Duration.new(seconds: 10, nanos: 500_000_000), m.duration
  1302. m.duration = 200
  1303. assert_equal Google::Protobuf::Duration.new(seconds: 200, nanos: 0), m.duration
  1304. m.duration = Rational(3, 2)
  1305. assert_equal Google::Protobuf::Duration.new(seconds: 1, nanos: 500_000_000), m.duration
  1306. m.duration = BigDecimal.new("5")
  1307. assert_equal Google::Protobuf::Duration.new(seconds: 5, nanos: 0), m.duration
  1308. m = proto_module::TimeMessage.new(duration: 1.1)
  1309. assert_equal Google::Protobuf::Duration.new(seconds: 1, nanos: 100_000_000), m.duration
  1310. assert_raise(Google::Protobuf::TypeError) { m.duration = '2' }
  1311. assert_raise(Google::Protobuf::TypeError) { m.duration = proto_module::TimeMessage.new }
  1312. end
  1313. def test_freeze
  1314. m = proto_module::TestMessage.new
  1315. m.optional_int32 = 10
  1316. m.freeze
  1317. frozen_error = assert_raise(FrozenErrorType) { m.optional_int32 = 20 }
  1318. assert_equal "can't modify frozen #{proto_module}::TestMessage", frozen_error.message
  1319. assert_equal 10, m.optional_int32
  1320. assert_equal true, m.frozen?
  1321. assert_raise(FrozenErrorType) { m.optional_int64 = 2 }
  1322. assert_raise(FrozenErrorType) { m.optional_uint32 = 3 }
  1323. assert_raise(FrozenErrorType) { m.optional_uint64 = 4 }
  1324. assert_raise(FrozenErrorType) { m.optional_bool = true }
  1325. assert_raise(FrozenErrorType) { m.optional_float = 6.0 }
  1326. assert_raise(FrozenErrorType) { m.optional_double = 7.0 }
  1327. assert_raise(FrozenErrorType) { m.optional_string = '8' }
  1328. assert_raise(FrozenErrorType) { m.optional_bytes = nil }
  1329. assert_raise(FrozenErrorType) { m.optional_msg = proto_module::TestMessage2.new }
  1330. assert_raise(FrozenErrorType) { m.optional_enum = :A }
  1331. assert_raise(FrozenErrorType) { m.repeated_int32 = 1 }
  1332. assert_raise(FrozenErrorType) { m.repeated_int64 = 2 }
  1333. assert_raise(FrozenErrorType) { m.repeated_uint32 = 3 }
  1334. assert_raise(FrozenErrorType) { m.repeated_uint64 = 4 }
  1335. assert_raise(FrozenErrorType) { m.repeated_bool = true }
  1336. assert_raise(FrozenErrorType) { m.repeated_float = 6.0 }
  1337. assert_raise(FrozenErrorType) { m.repeated_double = 7.0 }
  1338. assert_raise(FrozenErrorType) { m.repeated_string = '8' }
  1339. assert_raise(FrozenErrorType) { m.repeated_bytes = nil }
  1340. assert_raise(FrozenErrorType) { m.repeated_msg = proto_module::TestMessage2.new }
  1341. assert_raise(FrozenErrorType) { m.repeated_enum = :A }
  1342. end
  1343. def test_eq
  1344. m1 = proto_module::TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  1345. m2 = proto_module::TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  1346. h = {}
  1347. h[m1] = :yes
  1348. assert m1 == m2
  1349. assert m1.eql?(m2)
  1350. assert m1.hash == m2.hash
  1351. assert h[m1] == :yes
  1352. assert h[m2] == :yes
  1353. m1.optional_int32 = 2
  1354. assert m1 != m2
  1355. assert !m1.eql?(m2)
  1356. assert m1.hash != m2.hash
  1357. assert_nil h[m2]
  1358. end
  1359. end