common_tests.rb 65 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988
  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(foo: 0),
  104. :repeated_string => ["hello", "there", "world"])
  105. expected = "<#{proto_module}::TestMessage: optional_int32: -42, 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: b: -42>"
  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. l.push :v0
  287. assert l.count == 4
  288. assert_raise RangeError do
  289. l.push :D
  290. end
  291. assert l[0] == :A
  292. assert l[3] == :v0
  293. l.push 5
  294. assert l[4] == 5
  295. end
  296. def test_rptfield_initialize
  297. assert_raise ArgumentError do
  298. Google::Protobuf::RepeatedField.new
  299. end
  300. assert_raise ArgumentError do
  301. Google::Protobuf::RepeatedField.new(:message)
  302. end
  303. assert_raise ArgumentError do
  304. Google::Protobuf::RepeatedField.new([1, 2, 3])
  305. end
  306. assert_raise ArgumentError do
  307. Google::Protobuf::RepeatedField.new(:message, [proto_module::TestMessage2.new])
  308. end
  309. end
  310. def test_rptfield_array_ducktyping
  311. l = Google::Protobuf::RepeatedField.new(:int32)
  312. length_methods = %w(count length size)
  313. length_methods.each do |lm|
  314. assert l.send(lm) == 0
  315. end
  316. # out of bounds returns a nil
  317. assert l[0] == nil
  318. assert l[1] == nil
  319. assert l[-1] == nil
  320. l.push 4
  321. length_methods.each do |lm|
  322. assert l.send(lm) == 1
  323. end
  324. assert l[0] == 4
  325. assert l[1] == nil
  326. assert l[-1] == 4
  327. assert l[-2] == nil
  328. l.push 2
  329. length_methods.each do |lm|
  330. assert l.send(lm) == 2
  331. end
  332. assert l[0] == 4
  333. assert l[1] == 2
  334. assert l[2] == nil
  335. assert l[-1] == 2
  336. assert l[-2] == 4
  337. assert l[-3] == nil
  338. #adding out of scope will backfill with empty objects
  339. end
  340. def test_map_basic
  341. # allowed key types:
  342. # :int32, :int64, :uint32, :uint64, :bool, :string, :bytes.
  343. m = Google::Protobuf::Map.new(:string, :int32)
  344. m["asdf"] = 1
  345. assert m["asdf"] == 1
  346. m["jkl;"] = 42
  347. assert m == { "jkl;" => 42, "asdf" => 1 }
  348. assert m.has_key?("asdf")
  349. assert !m.has_key?("qwerty")
  350. assert m.length == 2
  351. m2 = m.dup
  352. assert_equal m, m2
  353. assert m.hash != 0
  354. assert_equal m.hash, m2.hash
  355. collected = {}
  356. m.each { |k,v| collected[v] = k }
  357. assert collected == { 42 => "jkl;", 1 => "asdf" }
  358. assert m.delete("asdf") == 1
  359. assert !m.has_key?("asdf")
  360. assert m["asdf"] == nil
  361. assert !m.has_key?("asdf")
  362. # We only assert on inspect value when there is one map entry because the
  363. # order in which elements appear is unspecified (depends on the internal
  364. # hash function). We don't want a brittle test.
  365. assert m.inspect == "{\"jkl;\"=>42}"
  366. assert m.keys == ["jkl;"]
  367. assert m.values == [42]
  368. m.clear
  369. assert m.length == 0
  370. assert m == {}
  371. assert_raise Google::Protobuf::TypeError do
  372. m[1] = 1
  373. end
  374. assert_raise RangeError do
  375. m["asdf"] = 0x1_0000_0000
  376. end
  377. end
  378. def test_b_8385
  379. m1 = Google::Protobuf::Map.new(:string, :string)
  380. m2 = Google::Protobuf::Map.new(:string, :string)
  381. assert_equal m1, m2
  382. m1["counter"] = "a"
  383. m2["counter"] = "aa"
  384. assert_not_equal m1, m2
  385. end
  386. def test_map_ctor
  387. m = Google::Protobuf::Map.new(:string, :int32,
  388. {"a" => 1, "b" => 2, "c" => 3})
  389. assert m == {"a" => 1, "c" => 3, "b" => 2}
  390. end
  391. def test_map_keytypes
  392. m = Google::Protobuf::Map.new(:int32, :int32)
  393. m[1] = 42
  394. m[-1] = 42
  395. assert_raise RangeError do
  396. m[0x8000_0000] = 1
  397. end
  398. assert_raise Google::Protobuf::TypeError do
  399. m["asdf"] = 1
  400. end
  401. m = Google::Protobuf::Map.new(:int64, :int32)
  402. m[0x1000_0000_0000_0000] = 1
  403. assert_raise RangeError do
  404. m[0x1_0000_0000_0000_0000] = 1
  405. end
  406. assert_raise Google::Protobuf::TypeError do
  407. m["asdf"] = 1
  408. end
  409. m = Google::Protobuf::Map.new(:uint32, :int32)
  410. m[0x8000_0000] = 1
  411. assert_raise RangeError do
  412. m[0x1_0000_0000] = 1
  413. end
  414. assert_raise RangeError do
  415. m[-1] = 1
  416. end
  417. m = Google::Protobuf::Map.new(:uint64, :int32)
  418. m[0x8000_0000_0000_0000] = 1
  419. assert_raise RangeError do
  420. m[0x1_0000_0000_0000_0000] = 1
  421. end
  422. assert_raise RangeError do
  423. m[-1] = 1
  424. end
  425. m = Google::Protobuf::Map.new(:bool, :int32)
  426. m[true] = 1
  427. m[false] = 2
  428. assert_raise Google::Protobuf::TypeError do
  429. m[1] = 1
  430. end
  431. assert_raise Google::Protobuf::TypeError do
  432. m["asdf"] = 1
  433. end
  434. m = Google::Protobuf::Map.new(:string, :int32)
  435. m["asdf"] = 1
  436. assert_raise Google::Protobuf::TypeError do
  437. m[1] = 1
  438. end
  439. assert_raise Encoding::UndefinedConversionError do
  440. bytestring = ["FFFF"].pack("H*")
  441. m[bytestring] = 1
  442. end
  443. m = Google::Protobuf::Map.new(:bytes, :int32)
  444. bytestring = ["FFFF"].pack("H*")
  445. m[bytestring] = 1
  446. # Allowed -- we will automatically convert to ASCII-8BIT.
  447. m["asdf"] = 1
  448. assert_raise Google::Protobuf::TypeError do
  449. m[1] = 1
  450. end
  451. end
  452. def test_map_msg_enum_valuetypes
  453. m = Google::Protobuf::Map.new(:string, :message, proto_module::TestMessage)
  454. m["asdf"] = proto_module::TestMessage.new
  455. assert_raise Google::Protobuf::TypeError do
  456. m["jkl;"] = proto_module::TestMessage2.new
  457. end
  458. m = Google::Protobuf::Map.new(
  459. :string, :message, proto_module::TestMessage,
  460. { "a" => proto_module::TestMessage.new(:optional_int32 => 42),
  461. "b" => proto_module::TestMessage.new(:optional_int32 => 84) })
  462. assert m.length == 2
  463. assert m.values.map{|msg| msg.optional_int32}.sort == [42, 84]
  464. m = Google::Protobuf::Map.new(:string, :enum, proto_module::TestEnum,
  465. { "x" => :A, "y" => :B, "z" => :C })
  466. assert m.length == 3
  467. assert m["z"] == :C
  468. m["z"] = 2
  469. assert m["z"] == :B
  470. m["z"] = 5
  471. assert m["z"] == 5
  472. assert_raise RangeError do
  473. m["z"] = :Z
  474. end
  475. assert_raise RangeError do
  476. m["z"] = "z"
  477. end
  478. end
  479. def test_map_dup_deep_copy
  480. m = Google::Protobuf::Map.new(
  481. :string, :message, proto_module::TestMessage,
  482. { "a" => proto_module::TestMessage.new(:optional_int32 => 42),
  483. "b" => proto_module::TestMessage.new(:optional_int32 => 84) })
  484. m2 = m.dup
  485. assert m.to_h == m2.to_h
  486. assert m == m2
  487. assert m.object_id != m2.object_id
  488. assert m["a"].object_id == m2["a"].object_id
  489. assert m["b"].object_id == m2["b"].object_id
  490. m2 = Google::Protobuf.deep_copy(m)
  491. assert m == m2
  492. assert m.object_id != m2.object_id
  493. assert m["a"].object_id != m2["a"].object_id
  494. assert m["b"].object_id != m2["b"].object_id
  495. end
  496. def test_oneof_descriptors
  497. d = proto_module::OneofMessage.descriptor
  498. o = d.lookup_oneof("my_oneof")
  499. assert o != nil
  500. assert o.class == Google::Protobuf::OneofDescriptor
  501. assert o.name == "my_oneof"
  502. oneof_count = 0
  503. d.each_oneof{ |oneof|
  504. oneof_count += 1
  505. assert oneof == o
  506. }
  507. assert oneof_count == 1
  508. assert o.count == 4
  509. field_names = o.map{|f| f.name}.sort
  510. assert field_names == ["a", "b", "c", "d"]
  511. end
  512. def test_oneof
  513. d = proto_module::OneofMessage.new
  514. assert d.a == ""
  515. assert d.b == 0
  516. assert d.c == nil
  517. assert d.d == :Default
  518. assert d.my_oneof == nil
  519. d.a = "hi"
  520. assert d.a == "hi"
  521. assert d.b == 0
  522. assert d.c == nil
  523. assert d.d == :Default
  524. assert d.my_oneof == :a
  525. d.b = 42
  526. assert d.a == ""
  527. assert d.b == 42
  528. assert d.c == nil
  529. assert d.d == :Default
  530. assert d.my_oneof == :b
  531. d.c = proto_module::TestMessage2.new(:foo => 100)
  532. assert d.a == ""
  533. assert d.b == 0
  534. assert d.c.foo == 100
  535. assert d.d == :Default
  536. assert d.my_oneof == :c
  537. d.d = :C
  538. assert d.a == ""
  539. assert d.b == 0
  540. assert d.c == nil
  541. assert d.d == :C
  542. assert d.my_oneof == :d
  543. d2 = proto_module::OneofMessage.decode(proto_module::OneofMessage.encode(d))
  544. assert d2 == d
  545. encoded_field_a = proto_module::OneofMessage.encode(proto_module::OneofMessage.new(:a => "string"))
  546. encoded_field_b = proto_module::OneofMessage.encode(proto_module::OneofMessage.new(:b => 1000))
  547. encoded_field_c = proto_module::OneofMessage.encode(
  548. proto_module::OneofMessage.new(:c => proto_module::TestMessage2.new(:foo => 1)))
  549. encoded_field_d = proto_module::OneofMessage.encode(proto_module::OneofMessage.new(:d => :B))
  550. d3 = proto_module::OneofMessage.decode(
  551. encoded_field_c + encoded_field_a + encoded_field_b + encoded_field_d)
  552. assert d3.a == ""
  553. assert d3.b == 0
  554. assert d3.c == nil
  555. assert d3.d == :B
  556. d4 = proto_module::OneofMessage.decode(
  557. encoded_field_c + encoded_field_a + encoded_field_b + encoded_field_d +
  558. encoded_field_c)
  559. assert d4.a == ""
  560. assert d4.b == 0
  561. assert d4.c.foo == 1
  562. assert d4.d == :Default
  563. d5 = proto_module::OneofMessage.new(:a => "hello")
  564. assert d5.a == "hello"
  565. d5.a = nil
  566. assert d5.a == ""
  567. assert proto_module::OneofMessage.encode(d5) == ''
  568. assert d5.my_oneof == nil
  569. end
  570. def test_enum_field
  571. m = proto_module::TestMessage.new
  572. assert m.optional_enum == :Default
  573. m.optional_enum = :A
  574. assert m.optional_enum == :A
  575. assert_raise RangeError do
  576. m.optional_enum = :ASDF
  577. end
  578. m.optional_enum = 1
  579. assert m.optional_enum == :A
  580. m.optional_enum = 100
  581. assert m.optional_enum == 100
  582. end
  583. def test_dup
  584. m = proto_module::TestMessage.new
  585. m.optional_string = "hello"
  586. m.optional_int32 = 42
  587. tm1 = proto_module::TestMessage2.new(:foo => 100)
  588. tm2 = proto_module::TestMessage2.new(:foo => 200)
  589. m.repeated_msg.push tm1
  590. assert m.repeated_msg[-1] == tm1
  591. m.repeated_msg.push tm2
  592. assert m.repeated_msg[-1] == tm2
  593. m2 = m.dup
  594. assert m == m2
  595. m.optional_int32 += 1
  596. assert m != m2
  597. assert m.repeated_msg[0] == m2.repeated_msg[0]
  598. assert m.repeated_msg[0].object_id == m2.repeated_msg[0].object_id
  599. end
  600. def test_deep_copy
  601. m = proto_module::TestMessage.new(:optional_int32 => 42,
  602. :repeated_msg => [proto_module::TestMessage2.new(:foo => 100)])
  603. m2 = Google::Protobuf.deep_copy(m)
  604. assert m == m2
  605. assert m.repeated_msg == m2.repeated_msg
  606. assert m.repeated_msg.object_id != m2.repeated_msg.object_id
  607. assert m.repeated_msg[0].object_id != m2.repeated_msg[0].object_id
  608. end
  609. def test_message_eq
  610. m = proto_module::TestMessage.new(:optional_int32 => 42,
  611. :repeated_int32 => [1, 2, 3])
  612. m2 = proto_module::TestMessage.new(:optional_int32 => 43,
  613. :repeated_int32 => [1, 2, 3])
  614. assert m != m2
  615. assert_not_equal proto_module::TestMessage.new, proto_module::TestMessage2.new
  616. end
  617. def test_enum_lookup
  618. assert proto_module::TestEnum::A == 1
  619. assert proto_module::TestEnum::B == 2
  620. assert proto_module::TestEnum::C == 3
  621. assert proto_module::TestEnum::V0 == 4
  622. assert proto_module::TestEnum::lookup(1) == :A
  623. assert proto_module::TestEnum::lookup(2) == :B
  624. assert proto_module::TestEnum::lookup(3) == :C
  625. assert proto_module::TestEnum::lookup(4) == :v0
  626. assert proto_module::TestEnum::resolve(:A) == 1
  627. assert proto_module::TestEnum::resolve(:B) == 2
  628. assert proto_module::TestEnum::resolve(:C) == 3
  629. assert proto_module::TestEnum::resolve(:v0) == 4
  630. end
  631. def test_enum_const_get_helpers
  632. m = proto_module::TestMessage.new
  633. assert_equal proto_module::TestEnum::Default, m.optional_enum_const
  634. assert_equal proto_module::TestEnum.const_get(:Default), m.optional_enum_const
  635. m = proto_module::TestMessage.new({optional_enum: proto_module::TestEnum::A})
  636. assert_equal proto_module::TestEnum::A, m.optional_enum_const
  637. assert_equal proto_module::TestEnum.const_get(:A), m.optional_enum_const
  638. m = proto_module::TestMessage.new({optional_enum: proto_module::TestEnum::B})
  639. assert_equal proto_module::TestEnum::B, m.optional_enum_const
  640. assert_equal proto_module::TestEnum.const_get(:B), m.optional_enum_const
  641. m = proto_module::TestMessage.new({optional_enum: proto_module::TestEnum::C})
  642. assert_equal proto_module::TestEnum::C, m.optional_enum_const
  643. assert_equal proto_module::TestEnum.const_get(:C), m.optional_enum_const
  644. m = proto_module::TestMessage2.new({foo: 2})
  645. assert_equal 2, m.foo
  646. assert_raise(NoMethodError) { m.foo_ }
  647. assert_raise(NoMethodError) { m.foo_X }
  648. assert_raise(NoMethodError) { m.foo_XX }
  649. assert_raise(NoMethodError) { m.foo_XXX }
  650. assert_raise(NoMethodError) { m.foo_XXXX }
  651. assert_raise(NoMethodError) { m.foo_XXXXX }
  652. assert_raise(NoMethodError) { m.foo_XXXXXX }
  653. m = proto_module::Enumer.new({optional_enum: :B})
  654. assert_equal :B, m.optional_enum
  655. assert_raise(NoMethodError) { m.optional_enum_ }
  656. assert_raise(NoMethodError) { m.optional_enum_X }
  657. assert_raise(NoMethodError) { m.optional_enum_XX }
  658. assert_raise(NoMethodError) { m.optional_enum_XXX }
  659. assert_raise(NoMethodError) { m.optional_enum_XXXX }
  660. assert_raise(NoMethodError) { m.optional_enum_XXXXX }
  661. assert_raise(NoMethodError) { m.optional_enum_XXXXXX }
  662. end
  663. def test_enum_getter
  664. m = proto_module::Enumer.new(:optional_enum => :B, :repeated_enum => [:A, :C])
  665. assert_equal :B, m.optional_enum
  666. assert_equal 2, m.optional_enum_const
  667. assert_equal proto_module::TestEnum::B, m.optional_enum_const
  668. assert_equal [:A, :C], m.repeated_enum
  669. assert_equal [1, 3], m.repeated_enum_const
  670. assert_equal [proto_module::TestEnum::A, proto_module::TestEnum::C], m.repeated_enum_const
  671. end
  672. def test_enum_getter_oneof
  673. m = proto_module::Enumer.new(:const => :C)
  674. assert_equal :C, m.const
  675. assert_equal 3, m.const_const
  676. assert_equal proto_module::TestEnum::C, m.const_const
  677. end
  678. def test_enum_getter_only_enums
  679. m = proto_module::Enumer.new(:optional_enum => :B, :a_const => 'thing')
  680. assert_equal 'thing', m.a_const
  681. assert_equal :B, m.optional_enum
  682. assert_raise(NoMethodError) { m.a }
  683. assert_raise(NoMethodError) { m.a_const_const }
  684. end
  685. def test_repeated_push
  686. m = proto_module::TestMessage.new
  687. m.repeated_string += ['one']
  688. m.repeated_string += %w[two three]
  689. assert_equal %w[one two three], m.repeated_string
  690. m.repeated_string.push( *['four', 'five'] )
  691. assert_equal %w[one two three four five], m.repeated_string
  692. m.repeated_string.push 'six', 'seven'
  693. assert_equal %w[one two three four five six seven], m.repeated_string
  694. m = proto_module::TestMessage.new
  695. m.repeated_msg += [proto_module::TestMessage2.new(:foo => 1), proto_module::TestMessage2.new(:foo => 2)]
  696. m.repeated_msg += [proto_module::TestMessage2.new(:foo => 3)]
  697. m.repeated_msg.push proto_module::TestMessage2.new(:foo => 4), proto_module::TestMessage2.new(:foo => 5)
  698. assert_equal [1, 2, 3, 4, 5], m.repeated_msg.map {|x| x.foo}
  699. end
  700. def test_parse_serialize
  701. m = proto_module::TestMessage.new(:optional_int32 => 42,
  702. :optional_string => "hello world",
  703. :optional_enum => :B,
  704. :repeated_string => ["a", "b", "c"],
  705. :repeated_int32 => [42, 43, 44],
  706. :repeated_enum => [:A, :B, :C],
  707. :repeated_msg => [proto_module::TestMessage2.new(:foo => 1),
  708. proto_module::TestMessage2.new(:foo => 2)])
  709. if proto_module == ::BasicTest
  710. # For proto3 we can add an unknown enum value safely.
  711. m.repeated_enum << 100
  712. end
  713. data = proto_module::TestMessage.encode m
  714. m2 = proto_module::TestMessage.decode data
  715. assert_equal m, m2
  716. data = Google::Protobuf.encode m
  717. m2 = Google::Protobuf.decode(proto_module::TestMessage, data)
  718. assert m == m2
  719. end
  720. def test_encode_decode_helpers
  721. m = proto_module::TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  722. assert_equal 'foo', m.optional_string
  723. assert_equal ['bar1', 'bar2'], m.repeated_string
  724. json = m.to_json
  725. m2 = proto_module::TestMessage.decode_json(json)
  726. assert_equal 'foo', m2.optional_string
  727. assert_equal ['bar1', 'bar2'], m2.repeated_string
  728. if RUBY_PLATFORM != "java"
  729. assert m2.optional_string.frozen?
  730. assert m2.repeated_string[0].frozen?
  731. end
  732. proto = m.to_proto
  733. m2 = proto_module::TestMessage.decode(proto)
  734. assert_equal 'foo', m2.optional_string
  735. assert_equal ['bar1', 'bar2'], m2.repeated_string
  736. end
  737. def test_protobuf_encode_decode_helpers
  738. m = proto_module::TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  739. encoded_msg = Google::Protobuf.encode(m)
  740. assert_equal m.to_proto, encoded_msg
  741. decoded_msg = Google::Protobuf.decode(proto_module::TestMessage, encoded_msg)
  742. assert_equal proto_module::TestMessage.decode(m.to_proto), decoded_msg
  743. end
  744. def test_protobuf_encode_decode_json_helpers
  745. m = proto_module::TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  746. encoded_msg = Google::Protobuf.encode_json(m)
  747. assert_equal m.to_json, encoded_msg
  748. decoded_msg = Google::Protobuf.decode_json(proto_module::TestMessage, encoded_msg)
  749. assert_equal proto_module::TestMessage.decode_json(m.to_json), decoded_msg
  750. assert_equal [m].to_json, Google::Protobuf.encode_json([m])
  751. assert_equal proto_module::TestMessage.decode_json([m.to_json].first), decoded_msg
  752. end
  753. def test_def_errors
  754. s = Google::Protobuf::DescriptorPool.new
  755. assert_raise Google::Protobuf::TypeError do
  756. s.build do
  757. # enum with no default (integer value 0)
  758. add_enum "MyEnum" do
  759. value :A, 1
  760. end
  761. end
  762. end
  763. assert_raise Google::Protobuf::TypeError do
  764. s.build do
  765. # message with required field (unsupported in proto3)
  766. add_message "MyMessage" do
  767. required :foo, :int32, 1
  768. end
  769. end
  770. end
  771. end
  772. def test_corecursive
  773. # just be sure that we can instantiate types with corecursive field-type
  774. # references.
  775. m = proto_module::Recursive1.new(:foo => proto_module::Recursive2.new(:foo => proto_module::Recursive1.new))
  776. assert proto_module::Recursive1.descriptor.lookup("foo").subtype ==
  777. proto_module::Recursive2.descriptor
  778. assert proto_module::Recursive2.descriptor.lookup("foo").subtype ==
  779. proto_module::Recursive1.descriptor
  780. serialized = proto_module::Recursive1.encode(m)
  781. m2 = proto_module::Recursive1.decode(serialized)
  782. assert m == m2
  783. end
  784. def test_serialize_cycle
  785. m = proto_module::Recursive1.new(:foo => proto_module::Recursive2.new)
  786. m.foo.foo = m
  787. assert_raise RuntimeError do
  788. proto_module::Recursive1.encode(m)
  789. end
  790. end
  791. def test_bad_field_names
  792. m = proto_module::BadFieldNames.new(:dup => 1, :class => 2)
  793. m2 = m.dup
  794. assert m == m2
  795. assert m['dup'] == 1
  796. assert m['class'] == 2
  797. m['dup'] = 3
  798. assert m['dup'] == 3
  799. end
  800. def test_int_ranges
  801. m = proto_module::TestMessage.new
  802. m.optional_int32 = 0
  803. m.optional_int32 = -0x8000_0000
  804. m.optional_int32 = +0x7fff_ffff
  805. m.optional_int32 = 1.0
  806. m.optional_int32 = -1.0
  807. m.optional_int32 = 2e9
  808. assert_raise RangeError do
  809. m.optional_int32 = -0x8000_0001
  810. end
  811. assert_raise RangeError do
  812. m.optional_int32 = +0x8000_0000
  813. end
  814. assert_raise RangeError do
  815. m.optional_int32 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  816. end
  817. assert_raise RangeError do
  818. m.optional_int32 = 1e12
  819. end
  820. assert_raise RangeError do
  821. m.optional_int32 = 1.5
  822. end
  823. m.optional_uint32 = 0
  824. m.optional_uint32 = +0xffff_ffff
  825. m.optional_uint32 = 1.0
  826. m.optional_uint32 = 4e9
  827. assert_raise RangeError do
  828. m.optional_uint32 = -1
  829. end
  830. assert_raise RangeError do
  831. m.optional_uint32 = -1.5
  832. end
  833. assert_raise RangeError do
  834. m.optional_uint32 = -1.5e12
  835. end
  836. assert_raise RangeError do
  837. m.optional_uint32 = -0x1000_0000_0000_0000
  838. end
  839. assert_raise RangeError do
  840. m.optional_uint32 = +0x1_0000_0000
  841. end
  842. assert_raise RangeError do
  843. m.optional_uint32 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  844. end
  845. assert_raise RangeError do
  846. m.optional_uint32 = 1e12
  847. end
  848. assert_raise RangeError do
  849. m.optional_uint32 = 1.5
  850. end
  851. m.optional_int64 = 0
  852. m.optional_int64 = -0x8000_0000_0000_0000
  853. m.optional_int64 = +0x7fff_ffff_ffff_ffff
  854. m.optional_int64 = 1.0
  855. m.optional_int64 = -1.0
  856. m.optional_int64 = 8e18
  857. m.optional_int64 = -8e18
  858. assert_raise RangeError do
  859. m.optional_int64 = -0x8000_0000_0000_0001
  860. end
  861. assert_raise RangeError do
  862. m.optional_int64 = +0x8000_0000_0000_0000
  863. end
  864. assert_raise RangeError do
  865. m.optional_int64 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  866. end
  867. assert_raise RangeError do
  868. m.optional_int64 = 1e50
  869. end
  870. assert_raise RangeError do
  871. m.optional_int64 = 1.5
  872. end
  873. m.optional_uint64 = 0
  874. m.optional_uint64 = +0xffff_ffff_ffff_ffff
  875. m.optional_uint64 = 1.0
  876. m.optional_uint64 = 16e18
  877. assert_raise RangeError do
  878. m.optional_uint64 = -1
  879. end
  880. assert_raise RangeError do
  881. m.optional_uint64 = -1.5
  882. end
  883. assert_raise RangeError do
  884. m.optional_uint64 = -1.5e12
  885. end
  886. assert_raise RangeError do
  887. m.optional_uint64 = -0x1_0000_0000_0000_0000
  888. end
  889. assert_raise RangeError do
  890. m.optional_uint64 = +0x1_0000_0000_0000_0000
  891. end
  892. assert_raise RangeError do
  893. m.optional_uint64 = +0x1000_0000_0000_0000_0000_0000 # force Bignum
  894. end
  895. assert_raise RangeError do
  896. m.optional_uint64 = 1e50
  897. end
  898. assert_raise RangeError do
  899. m.optional_uint64 = 1.5
  900. end
  901. end
  902. def test_stress_test
  903. m = proto_module::TestMessage.new
  904. m.optional_int32 = 42
  905. m.optional_int64 = 0x100000000
  906. m.optional_string = "hello world"
  907. 10.times do m.repeated_msg.push proto_module::TestMessage2.new(:foo => 42) end
  908. 10.times do m.repeated_string.push "hello world" end
  909. data = proto_module::TestMessage.encode(m)
  910. 10_000.times do
  911. m = proto_module::TestMessage.decode(data)
  912. data_new = proto_module::TestMessage.encode(m)
  913. assert data_new == data
  914. data = data_new
  915. end
  916. end
  917. def test_reflection
  918. m = proto_module::TestMessage.new(:optional_int32 => 1234)
  919. msgdef = m.class.descriptor
  920. assert msgdef.class == Google::Protobuf::Descriptor
  921. assert msgdef.any? {|field| field.name == "optional_int32"}
  922. optional_int32 = msgdef.lookup "optional_int32"
  923. assert optional_int32.class == Google::Protobuf::FieldDescriptor
  924. assert optional_int32 != nil
  925. assert optional_int32.name == "optional_int32"
  926. assert optional_int32.type == :int32
  927. optional_int32.set(m, 5678)
  928. assert m.optional_int32 == 5678
  929. m.optional_int32 = 1000
  930. assert optional_int32.get(m) == 1000
  931. optional_msg = msgdef.lookup "optional_msg"
  932. assert optional_msg.subtype == proto_module::TestMessage2.descriptor
  933. optional_msg.set(m, optional_msg.subtype.msgclass.new)
  934. assert msgdef.msgclass == proto_module::TestMessage
  935. optional_enum = msgdef.lookup "optional_enum"
  936. assert optional_enum.subtype == proto_module::TestEnum.descriptor
  937. assert optional_enum.subtype.class == Google::Protobuf::EnumDescriptor
  938. optional_enum.subtype.each do |k, v|
  939. # set with integer, check resolution to symbolic name
  940. optional_enum.set(m, v)
  941. assert optional_enum.get(m) == k
  942. end
  943. end
  944. def test_json
  945. m = proto_module::TestMessage.new(:optional_int32 => 1234,
  946. :optional_int64 => -0x1_0000_0000,
  947. :optional_uint32 => 0x8000_0000,
  948. :optional_uint64 => 0xffff_ffff_ffff_ffff,
  949. :optional_bool => true,
  950. :optional_float => 1.0,
  951. :optional_double => -1e100,
  952. :optional_string => "Test string",
  953. :optional_bytes => ["FFFFFFFF"].pack('H*'),
  954. :optional_msg => proto_module::TestMessage2.new(:foo => 42),
  955. :repeated_int32 => [1, 2, 3, 4],
  956. :repeated_string => ["a", "b", "c"],
  957. :repeated_bool => [true, false, true, false],
  958. :repeated_msg => [proto_module::TestMessage2.new(:foo => 1),
  959. proto_module::TestMessage2.new(:foo => 2)])
  960. json_text = proto_module::TestMessage.encode_json(m)
  961. m2 = proto_module::TestMessage.decode_json(json_text)
  962. assert_equal m, m2
  963. # Crash case from GitHub issue 283.
  964. bar = proto_module::Bar.new(msg: "bar")
  965. baz1 = proto_module::Baz.new(msg: "baz")
  966. baz2 = proto_module::Baz.new(msg: "quux")
  967. proto_module::Foo.encode_json(proto_module::Foo.new)
  968. proto_module::Foo.encode_json(proto_module::Foo.new(bar: bar))
  969. proto_module::Foo.encode_json(proto_module::Foo.new(bar: bar, baz: [baz1, baz2]))
  970. end
  971. def test_json_empty
  972. assert proto_module::TestMessage.encode_json(proto_module::TestMessage.new) == '{}'
  973. end
  974. def test_json_emit_defaults
  975. # TODO: Fix JSON in JRuby version.
  976. return if RUBY_PLATFORM == "java"
  977. m = proto_module::TestMessage.new
  978. expected = {
  979. repeatedInt32: [],
  980. repeatedInt64: [],
  981. repeatedUint32: [],
  982. repeatedUint64: [],
  983. repeatedBool: [],
  984. repeatedFloat: [],
  985. repeatedDouble: [],
  986. repeatedString: [],
  987. repeatedBytes: [],
  988. repeatedMsg: [],
  989. repeatedEnum: []
  990. }
  991. actual = proto_module::TestMessage.encode_json(m, :emit_defaults => true)
  992. assert_equal expected, JSON.parse(actual, :symbolize_names => true)
  993. end
  994. def test_json_emit_defaults_submsg
  995. # TODO: Fix JSON in JRuby version.
  996. return if RUBY_PLATFORM == "java"
  997. m = proto_module::TestMessage.new(optional_msg: proto_module::TestMessage2.new)
  998. expected = {
  999. optionalMsg: {},
  1000. repeatedInt32: [],
  1001. repeatedInt64: [],
  1002. repeatedUint32: [],
  1003. repeatedUint64: [],
  1004. repeatedBool: [],
  1005. repeatedFloat: [],
  1006. repeatedDouble: [],
  1007. repeatedString: [],
  1008. repeatedBytes: [],
  1009. repeatedMsg: [],
  1010. repeatedEnum: []
  1011. }
  1012. actual = proto_module::TestMessage.encode_json(m, :emit_defaults => true)
  1013. assert_equal expected, JSON.parse(actual, :symbolize_names => true)
  1014. end
  1015. def test_json_emit_defaults_repeated_submsg
  1016. # TODO: Fix JSON in JRuby version.
  1017. return if RUBY_PLATFORM == "java"
  1018. m = proto_module::TestMessage.new(repeated_msg: [proto_module::TestMessage2.new])
  1019. expected = {
  1020. repeatedInt32: [],
  1021. repeatedInt64: [],
  1022. repeatedUint32: [],
  1023. repeatedUint64: [],
  1024. repeatedBool: [],
  1025. repeatedFloat: [],
  1026. repeatedDouble: [],
  1027. repeatedString: [],
  1028. repeatedBytes: [],
  1029. repeatedMsg: [{}],
  1030. repeatedEnum: []
  1031. }
  1032. actual = proto_module::TestMessage.encode_json(m, :emit_defaults => true)
  1033. assert_equal expected, JSON.parse(actual, :symbolize_names => true)
  1034. end
  1035. def value_from_ruby(value)
  1036. ret = Google::Protobuf::Value.new
  1037. case value
  1038. when String
  1039. ret.string_value = value
  1040. when Google::Protobuf::Struct
  1041. ret.struct_value = value
  1042. when Hash
  1043. ret.struct_value = struct_from_ruby(value)
  1044. when Google::Protobuf::ListValue
  1045. ret.list_value = value
  1046. when Array
  1047. ret.list_value = list_from_ruby(value)
  1048. else
  1049. @log.error "Unknown type: #{value.class}"
  1050. raise Google::Protobuf::Error, "Unknown type: #{value.class}"
  1051. end
  1052. ret
  1053. end
  1054. def list_from_ruby(arr)
  1055. ret = Google::Protobuf::ListValue.new
  1056. arr.each do |v|
  1057. ret.values << value_from_ruby(v)
  1058. end
  1059. ret
  1060. end
  1061. def struct_from_ruby(hash)
  1062. ret = Google::Protobuf::Struct.new
  1063. hash.each do |k, v|
  1064. ret.fields[k] ||= value_from_ruby(v)
  1065. end
  1066. ret
  1067. end
  1068. def test_deep_json
  1069. # will not overflow
  1070. json = '{"a":{"a":{"a":{"a":{"a":{"a":{"a":{"a":{"a":{"a":{"a":'\
  1071. '{"a":{"a":{"a":{"a":{}}}}}}}}}}}}}}}}'
  1072. struct = struct_from_ruby(JSON.parse(json))
  1073. assert_equal json, struct.to_json
  1074. encoded = proto_module::MyRepeatedStruct.encode(
  1075. proto_module::MyRepeatedStruct.new(structs: [proto_module::MyStruct.new(struct: struct)]))
  1076. assert_equal json, proto_module::MyRepeatedStruct.decode(encoded).structs[0].struct.to_json
  1077. # will overflow
  1078. json = '{"a":{"a":{"a":[{"a":{"a":[{"a":[{"a":{"a":[{"a":[{"a":'\
  1079. '{"a":[{"a":[{"a":{"a":{"a":[{"a":"a"}]}}}]}]}}]}]}}]}]}}]}}}'
  1080. struct = struct_from_ruby(JSON.parse(json))
  1081. assert_equal json, struct.to_json
  1082. assert_raise(RuntimeError, "Recursion limit exceeded during encoding") do
  1083. struct = Google::Protobuf::Struct.new
  1084. struct.fields["foobar"] = Google::Protobuf::Value.new(struct_value: struct)
  1085. Google::Protobuf::Struct.encode(struct)
  1086. end
  1087. end
  1088. def test_comparison_with_arbitrary_object
  1089. assert proto_module::TestMessage.new != nil
  1090. end
  1091. def test_wrappers_set_to_default
  1092. run_asserts = ->(m) {
  1093. assert_equal 0.0, m.double.value
  1094. assert_equal 0.0, m.float.value
  1095. assert_equal 0, m.int32.value
  1096. assert_equal 0, m.int64.value
  1097. assert_equal 0, m.uint32.value
  1098. assert_equal 0, m.uint64.value
  1099. assert_equal false, m.bool.value
  1100. assert_equal '', m.string.value
  1101. assert_equal '', m.bytes.value
  1102. }
  1103. m = proto_module::Wrapper.new(
  1104. double: Google::Protobuf::DoubleValue.new(value: 0.0),
  1105. float: Google::Protobuf::FloatValue.new(value: 0.0),
  1106. int32: Google::Protobuf::Int32Value.new(value: 0),
  1107. int64: Google::Protobuf::Int64Value.new(value: 0),
  1108. uint32: Google::Protobuf::UInt32Value.new(value: 0),
  1109. uint64: Google::Protobuf::UInt64Value.new(value: 0),
  1110. bool: Google::Protobuf::BoolValue.new(value: false),
  1111. string: Google::Protobuf::StringValue.new(value: ""),
  1112. bytes: Google::Protobuf::BytesValue.new(value: ''),
  1113. )
  1114. run_asserts.call(m)
  1115. m2 = proto_module::Wrapper.decode(m.to_proto)
  1116. run_asserts.call(m2)
  1117. m3 = proto_module::Wrapper.decode_json(m.to_json)
  1118. run_asserts.call(m3)
  1119. end
  1120. def test_wrapper_getters
  1121. run_asserts = ->(m) {
  1122. assert_equal 2.0, m.double_as_value
  1123. assert_equal 2.0, m.double.value
  1124. assert_equal 2.0, m.double_as_value
  1125. assert_equal 4.0, m.float_as_value
  1126. assert_equal 4.0, m.float.value
  1127. assert_equal 4.0, m.float_as_value
  1128. assert_equal 3, m.int32_as_value
  1129. assert_equal 3, m.int32.value
  1130. assert_equal 3, m.int32_as_value
  1131. assert_equal 4, m.int64_as_value
  1132. assert_equal 4, m.int64.value
  1133. assert_equal 4, m.int64_as_value
  1134. assert_equal 5, m.uint32_as_value
  1135. assert_equal 5, m.uint32.value
  1136. assert_equal 5, m.uint32_as_value
  1137. assert_equal 6, m.uint64_as_value
  1138. assert_equal 6, m.uint64.value
  1139. assert_equal 6, m.uint64_as_value
  1140. assert_equal true, m.bool_as_value
  1141. assert_equal true, m.bool.value
  1142. assert_equal true, m.bool_as_value
  1143. assert_equal "st\nr", m.string_as_value
  1144. assert_equal "st\nr", m.string.value
  1145. assert_equal "st\nr", m.string_as_value
  1146. assert_equal 'fun', m.bytes_as_value
  1147. assert_equal 'fun', m.bytes.value
  1148. assert_equal 'fun', m.bytes_as_value
  1149. }
  1150. m = proto_module::Wrapper.new(
  1151. double: Google::Protobuf::DoubleValue.new(value: 2.0),
  1152. float: Google::Protobuf::FloatValue.new(value: 4.0),
  1153. int32: Google::Protobuf::Int32Value.new(value: 3),
  1154. int64: Google::Protobuf::Int64Value.new(value: 4),
  1155. uint32: Google::Protobuf::UInt32Value.new(value: 5),
  1156. uint64: Google::Protobuf::UInt64Value.new(value: 6),
  1157. bool: Google::Protobuf::BoolValue.new(value: true),
  1158. string: Google::Protobuf::StringValue.new(value: "st\nr"),
  1159. bytes: Google::Protobuf::BytesValue.new(value: 'fun'),
  1160. real_string: '100'
  1161. )
  1162. run_asserts.call(m)
  1163. serialized = proto_module::Wrapper::encode(m)
  1164. m2 = proto_module::Wrapper::decode(serialized)
  1165. run_asserts.call(m2)
  1166. # Test the case where we are serializing directly from the parsed form
  1167. # (before anything lazy is materialized).
  1168. m3 = proto_module::Wrapper::decode(serialized)
  1169. serialized2 = proto_module::Wrapper::encode(m3)
  1170. m4 = proto_module::Wrapper::decode(serialized2)
  1171. run_asserts.call(m4)
  1172. # Test that the lazy form compares equal to the expanded form.
  1173. m5 = proto_module::Wrapper::decode(serialized2)
  1174. assert_equal m5, m
  1175. serialized_json = proto_module::Wrapper::encode_json(m)
  1176. m6 = proto_module::Wrapper::decode_json(serialized_json)
  1177. assert_equal m6, m
  1178. end
  1179. def test_repeated_wrappers
  1180. run_asserts = ->(m) {
  1181. assert_equal 2.0, m.repeated_double[0].value
  1182. assert_equal 4.0, m.repeated_float[0].value
  1183. assert_equal 3, m.repeated_int32[0].value
  1184. assert_equal 4, m.repeated_int64[0].value
  1185. assert_equal 5, m.repeated_uint32[0].value
  1186. assert_equal 6, m.repeated_uint64[0].value
  1187. assert_equal true, m.repeated_bool[0].value
  1188. assert_equal 'str', m.repeated_string[0].value
  1189. assert_equal 'fun', m.repeated_bytes[0].value
  1190. }
  1191. m = proto_module::Wrapper.new(
  1192. repeated_double: [Google::Protobuf::DoubleValue.new(value: 2.0)],
  1193. repeated_float: [Google::Protobuf::FloatValue.new(value: 4.0)],
  1194. repeated_int32: [Google::Protobuf::Int32Value.new(value: 3)],
  1195. repeated_int64: [Google::Protobuf::Int64Value.new(value: 4)],
  1196. repeated_uint32: [Google::Protobuf::UInt32Value.new(value: 5)],
  1197. repeated_uint64: [Google::Protobuf::UInt64Value.new(value: 6)],
  1198. repeated_bool: [Google::Protobuf::BoolValue.new(value: true)],
  1199. repeated_string: [Google::Protobuf::StringValue.new(value: 'str')],
  1200. repeated_bytes: [Google::Protobuf::BytesValue.new(value: 'fun')],
  1201. )
  1202. run_asserts.call(m)
  1203. serialized = proto_module::Wrapper::encode(m)
  1204. m2 = proto_module::Wrapper::decode(serialized)
  1205. run_asserts.call(m2)
  1206. # Test the case where we are serializing directly from the parsed form
  1207. # (before anything lazy is materialized).
  1208. m3 = proto_module::Wrapper::decode(serialized)
  1209. serialized2 = proto_module::Wrapper::encode(m3)
  1210. m4 = proto_module::Wrapper::decode(serialized2)
  1211. run_asserts.call(m4)
  1212. # Test that the lazy form compares equal to the expanded form.
  1213. m5 = proto_module::Wrapper::decode(serialized2)
  1214. assert_equal m5, m
  1215. # Test JSON.
  1216. serialized_json = proto_module::Wrapper::encode_json(m5)
  1217. m6 = proto_module::Wrapper::decode_json(serialized_json)
  1218. run_asserts.call(m6)
  1219. assert_equal m6, m
  1220. end
  1221. def test_oneof_wrappers
  1222. run_test = ->(m) {
  1223. serialized = proto_module::Wrapper::encode(m)
  1224. m2 = proto_module::Wrapper::decode(serialized)
  1225. # Encode directly from lazy form.
  1226. serialized2 = proto_module::Wrapper::encode(m2)
  1227. assert_equal m, m2
  1228. assert_equal serialized, serialized2
  1229. serialized_json = proto_module::Wrapper::encode_json(m)
  1230. m3 = proto_module::Wrapper::decode_json(serialized_json)
  1231. assert_equal m, m3
  1232. }
  1233. m = proto_module::Wrapper.new()
  1234. run_test.call(m)
  1235. m.oneof_double_as_value = 2.0
  1236. run_test.call(m)
  1237. m.oneof_float_as_value = 4.0
  1238. run_test.call(m)
  1239. m.oneof_int32_as_value = 3
  1240. run_test.call(m)
  1241. m.oneof_int64_as_value = 5
  1242. run_test.call(m)
  1243. m.oneof_uint32_as_value = 6
  1244. run_test.call(m)
  1245. m.oneof_uint64_as_value = 7
  1246. run_test.call(m)
  1247. m.oneof_string_as_value = 'str'
  1248. run_test.call(m)
  1249. m.oneof_bytes_as_value = 'fun'
  1250. run_test.call(m)
  1251. end
  1252. def test_top_level_wrappers
  1253. # We don't expect anyone to do this, but we should also make sure it does
  1254. # the right thing.
  1255. run_test = ->(klass, val) {
  1256. m = klass.new(value: val)
  1257. serialized = klass::encode(m)
  1258. m2 = klass::decode(serialized)
  1259. # Encode directly from lazy form.
  1260. serialized2 = klass::encode(m2)
  1261. assert_equal m, m2
  1262. assert_equal serialized, serialized2
  1263. serialized_json = klass::encode_json(m)
  1264. # This is nonsensical to do and does not work. There is no good reason
  1265. # to parse a wrapper type directly.
  1266. assert_raise(RuntimeError) { klass::decode_json(serialized_json) }
  1267. }
  1268. run_test.call(Google::Protobuf::DoubleValue, 2.0)
  1269. run_test.call(Google::Protobuf::FloatValue, 4.0)
  1270. run_test.call(Google::Protobuf::Int32Value, 3)
  1271. run_test.call(Google::Protobuf::Int64Value, 4)
  1272. run_test.call(Google::Protobuf::UInt32Value, 5)
  1273. run_test.call(Google::Protobuf::UInt64Value, 6)
  1274. run_test.call(Google::Protobuf::BoolValue, true)
  1275. run_test.call(Google::Protobuf::StringValue, 'str')
  1276. run_test.call(Google::Protobuf::BytesValue, 'fun')
  1277. end
  1278. def test_wrapper_setters_as_value
  1279. run_asserts = ->(m) {
  1280. m.double_as_value = 4.8
  1281. assert_equal 4.8, m.double_as_value
  1282. assert_equal Google::Protobuf::DoubleValue.new(value: 4.8), m.double
  1283. m.float_as_value = 2.4
  1284. assert_in_delta 2.4, m.float_as_value
  1285. assert_in_delta Google::Protobuf::FloatValue.new(value: 2.4).value, m.float.value
  1286. m.int32_as_value = 5
  1287. assert_equal 5, m.int32_as_value
  1288. assert_equal Google::Protobuf::Int32Value.new(value: 5), m.int32
  1289. m.int64_as_value = 15
  1290. assert_equal 15, m.int64_as_value
  1291. assert_equal Google::Protobuf::Int64Value.new(value: 15), m.int64
  1292. m.uint32_as_value = 50
  1293. assert_equal 50, m.uint32_as_value
  1294. assert_equal Google::Protobuf::UInt32Value.new(value: 50), m.uint32
  1295. m.uint64_as_value = 500
  1296. assert_equal 500, m.uint64_as_value
  1297. assert_equal Google::Protobuf::UInt64Value.new(value: 500), m.uint64
  1298. m.bool_as_value = false
  1299. assert_equal false, m.bool_as_value
  1300. assert_equal Google::Protobuf::BoolValue.new(value: false), m.bool
  1301. m.string_as_value = 'xy'
  1302. assert_equal 'xy', m.string_as_value
  1303. assert_equal Google::Protobuf::StringValue.new(value: 'xy'), m.string
  1304. m.bytes_as_value = '123'
  1305. assert_equal '123', m.bytes_as_value
  1306. assert_equal Google::Protobuf::BytesValue.new(value: '123'), m.bytes
  1307. m.double_as_value = nil
  1308. assert_nil m.double
  1309. assert_nil m.double_as_value
  1310. m.float_as_value = nil
  1311. assert_nil m.float
  1312. assert_nil m.float_as_value
  1313. m.int32_as_value = nil
  1314. assert_nil m.int32
  1315. assert_nil m.int32_as_value
  1316. m.int64_as_value = nil
  1317. assert_nil m.int64
  1318. assert_nil m.int64_as_value
  1319. m.uint32_as_value = nil
  1320. assert_nil m.uint32
  1321. assert_nil m.uint32_as_value
  1322. m.uint64_as_value = nil
  1323. assert_nil m.uint64
  1324. assert_nil m.uint64_as_value
  1325. m.bool_as_value = nil
  1326. assert_nil m.bool
  1327. assert_nil m.bool_as_value
  1328. m.string_as_value = nil
  1329. assert_nil m.string
  1330. assert_nil m.string_as_value
  1331. m.bytes_as_value = nil
  1332. assert_nil m.bytes
  1333. assert_nil m.bytes_as_value
  1334. }
  1335. m2 = proto_module::Wrapper.new(
  1336. double: Google::Protobuf::DoubleValue.new(value: 2.0),
  1337. float: Google::Protobuf::FloatValue.new(value: 4.0),
  1338. int32: Google::Protobuf::Int32Value.new(value: 3),
  1339. int64: Google::Protobuf::Int64Value.new(value: 4),
  1340. uint32: Google::Protobuf::UInt32Value.new(value: 5),
  1341. uint64: Google::Protobuf::UInt64Value.new(value: 6),
  1342. bool: Google::Protobuf::BoolValue.new(value: true),
  1343. string: Google::Protobuf::StringValue.new(value: 'str'),
  1344. bytes: Google::Protobuf::BytesValue.new(value: 'fun'),
  1345. real_string: '100'
  1346. )
  1347. run_asserts.call(m2)
  1348. serialized = proto_module::Wrapper::encode(m2)
  1349. m3 = proto_module::Wrapper::decode(serialized)
  1350. run_asserts.call(m3)
  1351. end
  1352. def test_wrapper_setters
  1353. run_asserts = ->(m) {
  1354. m.double = Google::Protobuf::DoubleValue.new(value: 4.8)
  1355. assert_equal 4.8, m.double_as_value
  1356. assert_equal Google::Protobuf::DoubleValue.new(value: 4.8), m.double
  1357. m.float = Google::Protobuf::FloatValue.new(value: 2.4)
  1358. assert_in_delta 2.4, m.float_as_value
  1359. assert_in_delta Google::Protobuf::FloatValue.new(value: 2.4).value, m.float.value
  1360. m.int32 = Google::Protobuf::Int32Value.new(value: 5)
  1361. assert_equal 5, m.int32_as_value
  1362. assert_equal Google::Protobuf::Int32Value.new(value: 5), m.int32
  1363. m.int64 = Google::Protobuf::Int64Value.new(value: 15)
  1364. assert_equal 15, m.int64_as_value
  1365. assert_equal Google::Protobuf::Int64Value.new(value: 15), m.int64
  1366. m.uint32 = Google::Protobuf::UInt32Value.new(value: 50)
  1367. assert_equal 50, m.uint32_as_value
  1368. assert_equal Google::Protobuf::UInt32Value.new(value: 50), m.uint32
  1369. m.uint64 = Google::Protobuf::UInt64Value.new(value: 500)
  1370. assert_equal 500, m.uint64_as_value
  1371. assert_equal Google::Protobuf::UInt64Value.new(value: 500), m.uint64
  1372. m.bool = Google::Protobuf::BoolValue.new(value: false)
  1373. assert_equal false, m.bool_as_value
  1374. assert_equal Google::Protobuf::BoolValue.new(value: false), m.bool
  1375. m.string = Google::Protobuf::StringValue.new(value: 'xy')
  1376. assert_equal 'xy', m.string_as_value
  1377. assert_equal Google::Protobuf::StringValue.new(value: 'xy'), m.string
  1378. m.bytes = Google::Protobuf::BytesValue.new(value: '123')
  1379. assert_equal '123', m.bytes_as_value
  1380. assert_equal Google::Protobuf::BytesValue.new(value: '123'), m.bytes
  1381. m.double = nil
  1382. assert_nil m.double
  1383. assert_nil m.double_as_value
  1384. m.float = nil
  1385. assert_nil m.float
  1386. assert_nil m.float_as_value
  1387. m.int32 = nil
  1388. assert_nil m.int32
  1389. assert_nil m.int32_as_value
  1390. m.int64 = nil
  1391. assert_nil m.int64
  1392. assert_nil m.int64_as_value
  1393. m.uint32 = nil
  1394. assert_nil m.uint32
  1395. assert_nil m.uint32_as_value
  1396. m.uint64 = nil
  1397. assert_nil m.uint64
  1398. assert_nil m.uint64_as_value
  1399. m.bool = nil
  1400. assert_nil m.bool
  1401. assert_nil m.bool_as_value
  1402. m.string = nil
  1403. assert_nil m.string
  1404. assert_nil m.string_as_value
  1405. m.bytes = nil
  1406. assert_nil m.bytes
  1407. assert_nil m.bytes_as_value
  1408. }
  1409. m = proto_module::Wrapper.new
  1410. run_asserts.call(m)
  1411. m2 = proto_module::Wrapper.new(
  1412. double: Google::Protobuf::DoubleValue.new(value: 2.0),
  1413. float: Google::Protobuf::FloatValue.new(value: 4.0),
  1414. int32: Google::Protobuf::Int32Value.new(value: 3),
  1415. int64: Google::Protobuf::Int64Value.new(value: 4),
  1416. uint32: Google::Protobuf::UInt32Value.new(value: 5),
  1417. uint64: Google::Protobuf::UInt64Value.new(value: 6),
  1418. bool: Google::Protobuf::BoolValue.new(value: true),
  1419. string: Google::Protobuf::StringValue.new(value: 'str'),
  1420. bytes: Google::Protobuf::BytesValue.new(value: 'fun'),
  1421. real_string: '100'
  1422. )
  1423. run_asserts.call(m2)
  1424. serialized = proto_module::Wrapper::encode(m2)
  1425. m3 = proto_module::Wrapper::decode(serialized)
  1426. run_asserts.call(m3)
  1427. end
  1428. def test_wrappers_only
  1429. m = proto_module::Wrapper.new(real_string: 'hi', string_in_oneof: 'there')
  1430. assert_raise(NoMethodError) { m.real_string_as_value }
  1431. assert_raise(NoMethodError) { m.as_value }
  1432. assert_raise(NoMethodError) { m._as_value }
  1433. assert_raise(NoMethodError) { m.string_in_oneof_as_value }
  1434. m = proto_module::Wrapper.new
  1435. m.string_as_value = 'you'
  1436. assert_equal 'you', m.string.value
  1437. assert_equal 'you', m.string_as_value
  1438. assert_raise(NoMethodError) { m.string_ }
  1439. assert_raise(NoMethodError) { m.string_X }
  1440. assert_raise(NoMethodError) { m.string_XX }
  1441. assert_raise(NoMethodError) { m.string_XXX }
  1442. assert_raise(NoMethodError) { m.string_XXXX }
  1443. assert_raise(NoMethodError) { m.string_XXXXX }
  1444. assert_raise(NoMethodError) { m.string_XXXXXX }
  1445. assert_raise(NoMethodError) { m.string_XXXXXXX }
  1446. assert_raise(NoMethodError) { m.string_XXXXXXXX }
  1447. assert_raise(NoMethodError) { m.string_XXXXXXXXX }
  1448. assert_raise(NoMethodError) { m.string_XXXXXXXXXX }
  1449. end
  1450. def test_converts_time
  1451. m = proto_module::TimeMessage.new
  1452. m.timestamp = Google::Protobuf::Timestamp.new(seconds: 5, nanos: 6)
  1453. assert_kind_of Google::Protobuf::Timestamp, m.timestamp
  1454. assert_equal 5, m.timestamp.seconds
  1455. assert_equal 6, m.timestamp.nanos
  1456. m.timestamp = Time.at(9466, 123456.789)
  1457. assert_equal Google::Protobuf::Timestamp.new(seconds: 9466, nanos: 123456789), m.timestamp
  1458. m = proto_module::TimeMessage.new(timestamp: Time.at(1))
  1459. assert_equal Google::Protobuf::Timestamp.new(seconds: 1, nanos: 0), m.timestamp
  1460. assert_raise(Google::Protobuf::TypeError) { m.timestamp = 2 }
  1461. assert_raise(Google::Protobuf::TypeError) { m.timestamp = 2.4 }
  1462. assert_raise(Google::Protobuf::TypeError) { m.timestamp = '4' }
  1463. assert_raise(Google::Protobuf::TypeError) { m.timestamp = proto_module::TimeMessage.new }
  1464. def test_time(year, month, day)
  1465. str = ("\"%04d-%02d-%02dT00:00:00.000+00:00\"" % [year, month, day])
  1466. t = Google::Protobuf::Timestamp.decode_json(str)
  1467. time = Time.new(year, month, day, 0, 0, 0, "+00:00")
  1468. assert_equal t.seconds, time.to_i
  1469. end
  1470. (1970..2010).each do |year|
  1471. test_time(year, 2, 28)
  1472. test_time(year, 3, 01)
  1473. end
  1474. end
  1475. def test_converts_duration
  1476. m = proto_module::TimeMessage.new
  1477. m.duration = Google::Protobuf::Duration.new(seconds: 2, nanos: 22)
  1478. assert_kind_of Google::Protobuf::Duration, m.duration
  1479. assert_equal 2, m.duration.seconds
  1480. assert_equal 22, m.duration.nanos
  1481. m.duration = 10.5
  1482. assert_equal Google::Protobuf::Duration.new(seconds: 10, nanos: 500_000_000), m.duration
  1483. m.duration = 200
  1484. assert_equal Google::Protobuf::Duration.new(seconds: 200, nanos: 0), m.duration
  1485. m.duration = Rational(3, 2)
  1486. assert_equal Google::Protobuf::Duration.new(seconds: 1, nanos: 500_000_000), m.duration
  1487. m.duration = BigDecimal("5")
  1488. assert_equal Google::Protobuf::Duration.new(seconds: 5, nanos: 0), m.duration
  1489. m = proto_module::TimeMessage.new(duration: 1.1)
  1490. assert_equal Google::Protobuf::Duration.new(seconds: 1, nanos: 100_000_000), m.duration
  1491. m = proto_module::TimeMessage.new(duration: 123.321)
  1492. assert_equal Google::Protobuf::Duration.new(seconds: 123, nanos: 321_000_000), m.duration
  1493. m = proto_module::TimeMessage.new(duration: -123.321)
  1494. assert_equal Google::Protobuf::Duration.new(seconds: -123, nanos: -321_000_000), m.duration
  1495. assert_raise(Google::Protobuf::TypeError) { m.duration = '2' }
  1496. assert_raise(Google::Protobuf::TypeError) { m.duration = proto_module::TimeMessage.new }
  1497. end
  1498. def test_freeze
  1499. m = proto_module::TestMessage.new
  1500. m.optional_int32 = 10
  1501. m.freeze
  1502. frozen_error = assert_raise(FrozenErrorType) { m.optional_int32 = 20 }
  1503. assert_match "can't modify frozen #{proto_module}::TestMessage", frozen_error.message
  1504. assert_equal 10, m.optional_int32
  1505. assert_equal true, m.frozen?
  1506. assert_raise(FrozenErrorType) { m.optional_int64 = 2 }
  1507. assert_raise(FrozenErrorType) { m.optional_uint32 = 3 }
  1508. assert_raise(FrozenErrorType) { m.optional_uint64 = 4 }
  1509. assert_raise(FrozenErrorType) { m.optional_bool = true }
  1510. assert_raise(FrozenErrorType) { m.optional_float = 6.0 }
  1511. assert_raise(FrozenErrorType) { m.optional_double = 7.0 }
  1512. assert_raise(FrozenErrorType) { m.optional_string = '8' }
  1513. assert_raise(FrozenErrorType) { m.optional_bytes = nil }
  1514. assert_raise(FrozenErrorType) { m.optional_msg = proto_module::TestMessage2.new }
  1515. assert_raise(FrozenErrorType) { m.optional_enum = :A }
  1516. assert_raise(FrozenErrorType) { m.repeated_int32 = 1 }
  1517. assert_raise(FrozenErrorType) { m.repeated_int64 = 2 }
  1518. assert_raise(FrozenErrorType) { m.repeated_uint32 = 3 }
  1519. assert_raise(FrozenErrorType) { m.repeated_uint64 = 4 }
  1520. assert_raise(FrozenErrorType) { m.repeated_bool = true }
  1521. assert_raise(FrozenErrorType) { m.repeated_float = 6.0 }
  1522. assert_raise(FrozenErrorType) { m.repeated_double = 7.0 }
  1523. assert_raise(FrozenErrorType) { m.repeated_string = '8' }
  1524. assert_raise(FrozenErrorType) { m.repeated_bytes = nil }
  1525. assert_raise(FrozenErrorType) { m.repeated_msg = proto_module::TestMessage2.new }
  1526. assert_raise(FrozenErrorType) { m.repeated_enum = :A }
  1527. end
  1528. def test_eq
  1529. m1 = proto_module::TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  1530. m2 = proto_module::TestMessage.new(:optional_string => 'foo', :repeated_string => ['bar1', 'bar2'])
  1531. h = {}
  1532. h[m1] = :yes
  1533. assert m1 == m2
  1534. assert m1.eql?(m2)
  1535. assert m1.hash == m2.hash
  1536. assert h[m1] == :yes
  1537. assert h[m2] == :yes
  1538. m1.optional_int32 = 2
  1539. assert m1 != m2
  1540. assert !m1.eql?(m2)
  1541. assert m1.hash != m2.hash
  1542. assert_nil h[m2]
  1543. end
  1544. def cruby_or_jruby_9_3_or_higher?
  1545. # https://github.com/jruby/jruby/issues/6818 was fixed in JRuby 9.3.0.0
  1546. match = RUBY_PLATFORM == "java" &&
  1547. JRUBY_VERSION.match(/^(\d+)\.(\d+)\.\d+\.\d+$/)
  1548. match && (match[1].to_i > 9 || (match[1].to_i == 9 && match[2].to_i >= 3))
  1549. end
  1550. def test_object_gc
  1551. m = proto_module::TestMessage.new(optional_msg: proto_module::TestMessage2.new)
  1552. m.optional_msg
  1553. # https://github.com/jruby/jruby/issues/6818 was fixed in JRuby 9.3.0.0
  1554. GC.start(full_mark: true, immediate_sweep: true) if cruby_or_jruby_9_3_or_higher?
  1555. m.optional_msg.inspect
  1556. end
  1557. def test_object_gc_freeze
  1558. m = proto_module::TestMessage.new
  1559. m.repeated_float.freeze
  1560. # https://github.com/jruby/jruby/issues/6818 was fixed in JRuby 9.3.0.0
  1561. GC.start(full_mark: true) if cruby_or_jruby_9_3_or_higher?
  1562. # Make sure we remember that the object is frozen.
  1563. # The wrapper object contains this information, so we need to ensure that
  1564. # the previous GC did not collect it.
  1565. assert m.repeated_float.frozen?
  1566. # https://github.com/jruby/jruby/issues/6818 was fixed in JRuby 9.3.0.0
  1567. GC.start(full_mark: true, immediate_sweep: true) if cruby_or_jruby_9_3_or_higher?
  1568. assert m.repeated_float.frozen?
  1569. end
  1570. def test_optional_fields_respond_to? # regression test for issue 9202
  1571. msg = proto_module::TestMessage.new
  1572. assert msg.respond_to? :optional_int32=
  1573. msg.optional_int32 = 42
  1574. assert msg.respond_to? :optional_int32
  1575. assert_equal 42, msg.optional_int32
  1576. assert msg.respond_to? :clear_optional_int32
  1577. msg.clear_optional_int32
  1578. assert_equal 0, msg.optional_int32
  1579. assert msg.respond_to? :has_optional_int32?
  1580. assert !msg.has_optional_int32?
  1581. assert !msg.respond_to?( :optional_int32_as_value= )
  1582. assert_raise NoMethodError do
  1583. msg.optional_int32_as_value = 42
  1584. end
  1585. assert !msg.respond_to?( :optional_int32_as_value )
  1586. assert_raise NoMethodError do
  1587. msg.optional_int32_as_value
  1588. end
  1589. assert msg.respond_to? :optional_enum_const
  1590. assert_equal 0, msg.optional_enum_const
  1591. assert !msg.respond_to?( :foo )
  1592. assert_raise NoMethodError do
  1593. msg.foo
  1594. end
  1595. assert !msg.respond_to?( :foo_const )
  1596. assert_raise NoMethodError do
  1597. msg.foo_const
  1598. end
  1599. assert !msg.respond_to?( :optional_int32_const )
  1600. assert_raise NoMethodError do
  1601. msg.optional_int32_const
  1602. end
  1603. end
  1604. def test_oneof_fields_respond_to? # regression test for issue 9202
  1605. msg = proto_module::OneofMessage.new
  1606. # names of the elements of a oneof and the oneof itself are valid actions.
  1607. assert msg.respond_to? :my_oneof
  1608. assert_nil msg.my_oneof
  1609. assert msg.respond_to? :a
  1610. assert_equal "", msg.a
  1611. assert msg.respond_to? :b
  1612. assert_equal 0, msg.b
  1613. assert msg.respond_to? :c
  1614. assert_nil msg.c
  1615. assert msg.respond_to? :d
  1616. assert_equal :Default, msg.d
  1617. # `clear` prefix actions work on elements of a oneof and the oneof itself.
  1618. assert msg.respond_to? :clear_my_oneof
  1619. msg.clear_my_oneof
  1620. # Repeatedly clearing a oneof used to cause a NoMethodError under JRuby
  1621. msg.clear_my_oneof
  1622. assert msg.respond_to? :clear_a
  1623. msg.clear_a
  1624. assert msg.respond_to? :clear_b
  1625. msg.clear_b
  1626. assert msg.respond_to? :clear_c
  1627. msg.clear_c
  1628. assert msg.respond_to? :clear_d
  1629. msg.clear_d
  1630. # `=` suffix actions should work on elements of a oneof but not the oneof itself.
  1631. assert !msg.respond_to?( :my_oneof= )
  1632. error = assert_raise RuntimeError do
  1633. msg.my_oneof = nil
  1634. end
  1635. assert_equal "Oneof accessors are read-only.", error.message
  1636. assert msg.respond_to? :a=
  1637. msg.a = "foo"
  1638. assert msg.respond_to? :b=
  1639. msg.b = 42
  1640. assert msg.respond_to? :c=
  1641. msg.c = proto_module::TestMessage2.new
  1642. assert msg.respond_to? :d=
  1643. msg.d = :Default
  1644. # `has_` prefix + "?" suffix actions work for oneofs fields.
  1645. assert msg.respond_to? :has_my_oneof?
  1646. assert msg.has_my_oneof?
  1647. # `_as_value` suffix actions should only work for wrapped fields.
  1648. assert !msg.respond_to?( :my_oneof_as_value )
  1649. assert_raise NoMethodError do
  1650. msg.my_oneof_as_value
  1651. end
  1652. assert !msg.respond_to?( :a_as_value )
  1653. assert_raise NoMethodError do
  1654. msg.a_as_value
  1655. end
  1656. assert !msg.respond_to?( :b_as_value )
  1657. assert_raise NoMethodError do
  1658. msg.b_as_value
  1659. end
  1660. assert !msg.respond_to?( :c_as_value )
  1661. assert_raise NoMethodError do
  1662. msg.c_as_value
  1663. end
  1664. assert !msg.respond_to?( :d_as_value )
  1665. assert_raise NoMethodError do
  1666. msg.d_as_value
  1667. end
  1668. # `_as_value=` suffix actions should only work for wrapped fields.
  1669. assert !msg.respond_to?( :my_oneof_as_value= )
  1670. assert_raise NoMethodError do
  1671. msg.my_oneof_as_value = :boom
  1672. end
  1673. assert !msg.respond_to?( :a_as_value= )
  1674. assert_raise NoMethodError do
  1675. msg.a_as_value = ""
  1676. end
  1677. assert !msg.respond_to?( :b_as_value= )
  1678. assert_raise NoMethodError do
  1679. msg.b_as_value = 42
  1680. end
  1681. assert !msg.respond_to?( :c_as_value= )
  1682. assert_raise NoMethodError do
  1683. msg.c_as_value = proto_module::TestMessage2.new
  1684. end
  1685. assert !msg.respond_to?( :d_as_value= )
  1686. assert_raise NoMethodError do
  1687. msg.d_as_value = :Default
  1688. end
  1689. # `_const` suffix actions should only work for enum fields.
  1690. assert !msg.respond_to?( :my_oneof_const )
  1691. assert_raise NoMethodError do
  1692. msg.my_oneof_const
  1693. end
  1694. assert !msg.respond_to?( :a_const )
  1695. assert_raise NoMethodError do
  1696. msg.a_const
  1697. end
  1698. assert !msg.respond_to?( :b_const )
  1699. assert_raise NoMethodError do
  1700. msg.b_const
  1701. end
  1702. assert !msg.respond_to?( :c_const )
  1703. assert_raise NoMethodError do
  1704. msg.c_const
  1705. end
  1706. assert msg.respond_to? :d_const
  1707. assert_equal 0, msg.d_const
  1708. end
  1709. def test_wrapped_fields_respond_to? # regression test for issue 9202
  1710. msg = proto_module::Wrapper.new
  1711. assert msg.respond_to?( :double_as_value= )
  1712. msg.double_as_value = 42
  1713. assert msg.respond_to?( :double_as_value )
  1714. assert_equal 42, msg.double_as_value
  1715. assert_equal Google::Protobuf::DoubleValue.new(value: 42), msg.double
  1716. end
  1717. end