| OLD | NEW |
| (Empty) |
| 1 #!/usr/bin/ruby | |
| 2 | |
| 3 require 'google/protobuf' | |
| 4 require 'test/unit' | |
| 5 | |
| 6 class RepeatedFieldTest < Test::Unit::TestCase | |
| 7 | |
| 8 def test_acts_like_enumerator | |
| 9 m = TestMessage.new | |
| 10 (Enumerable.instance_methods - TestMessage.new.repeated_string.methods).each
do |method_name| | |
| 11 assert m.repeated_string.respond_to?(method_name) == true, "does not respo
nd to #{method_name}" | |
| 12 end | |
| 13 end | |
| 14 | |
| 15 def test_acts_like_an_array | |
| 16 m = TestMessage.new | |
| 17 arr_methods = ([].methods - TestMessage.new.repeated_string.methods) | |
| 18 # jRuby additions to the Array class that we can ignore | |
| 19 arr_methods -= [ :indices, :iter_for_each, :iter_for_each_index, | |
| 20 :iter_for_each_with_index, :dimensions, :copy_data, :copy_data_simple, | |
| 21 :nitems, :iter_for_reverse_each, :indexes] | |
| 22 arr_methods.each do |method_name| | |
| 23 assert m.repeated_string.respond_to?(method_name) == true, "does not respo
nd to #{method_name}" | |
| 24 end | |
| 25 end | |
| 26 | |
| 27 def test_first | |
| 28 m = TestMessage.new | |
| 29 repeated_field_names(TestMessage).each do |field_name| | |
| 30 assert_nil m.send(field_name).first | |
| 31 end | |
| 32 fill_test_msg(m) | |
| 33 assert_equal -10, m.repeated_int32.first | |
| 34 assert_equal -1_000_000, m.repeated_int64.first | |
| 35 assert_equal 10, m.repeated_uint32.first | |
| 36 assert_equal 1_000_000, m.repeated_uint64.first | |
| 37 assert_equal true, m.repeated_bool.first | |
| 38 assert_equal -1.01, m.repeated_float.first.round(2) | |
| 39 assert_equal -1.0000000000001, m.repeated_double.first | |
| 40 assert_equal 'foo', m.repeated_string.first | |
| 41 assert_equal "bar".encode!('ASCII-8BIT'), m.repeated_bytes.first | |
| 42 assert_equal TestMessage2.new(:foo => 1), m.repeated_msg.first | |
| 43 assert_equal :A, m.repeated_enum.first | |
| 44 end | |
| 45 | |
| 46 | |
| 47 def test_last | |
| 48 m = TestMessage.new | |
| 49 repeated_field_names(TestMessage).each do |field_name| | |
| 50 assert_nil m.send(field_name).first | |
| 51 end | |
| 52 fill_test_msg(m) | |
| 53 assert_equal -11, m.repeated_int32.last | |
| 54 assert_equal -1_000_001, m.repeated_int64.last | |
| 55 assert_equal 11, m.repeated_uint32.last | |
| 56 assert_equal 1_000_001, m.repeated_uint64.last | |
| 57 assert_equal false, m.repeated_bool.last | |
| 58 assert_equal -1.02, m.repeated_float.last.round(2) | |
| 59 assert_equal -1.0000000000002, m.repeated_double.last | |
| 60 assert_equal 'bar', m.repeated_string.last | |
| 61 assert_equal "foo".encode!('ASCII-8BIT'), m.repeated_bytes.last | |
| 62 assert_equal TestMessage2.new(:foo => 2), m.repeated_msg.last | |
| 63 assert_equal :B, m.repeated_enum.last | |
| 64 end | |
| 65 | |
| 66 | |
| 67 def test_pop | |
| 68 m = TestMessage.new | |
| 69 repeated_field_names(TestMessage).each do |field_name| | |
| 70 assert_nil m.send(field_name).pop | |
| 71 end | |
| 72 fill_test_msg(m) | |
| 73 | |
| 74 assert_equal -11, m.repeated_int32.pop | |
| 75 assert_equal -10, m.repeated_int32.pop | |
| 76 assert_equal -1_000_001, m.repeated_int64.pop | |
| 77 assert_equal -1_000_000, m.repeated_int64.pop | |
| 78 assert_equal 11, m.repeated_uint32.pop | |
| 79 assert_equal 10, m.repeated_uint32.pop | |
| 80 assert_equal 1_000_001, m.repeated_uint64.pop | |
| 81 assert_equal 1_000_000, m.repeated_uint64.pop | |
| 82 assert_equal false, m.repeated_bool.pop | |
| 83 assert_equal true, m.repeated_bool.pop | |
| 84 assert_equal -1.02, m.repeated_float.pop.round(2) | |
| 85 assert_equal -1.01, m.repeated_float.pop.round(2) | |
| 86 assert_equal -1.0000000000002, m.repeated_double.pop | |
| 87 assert_equal -1.0000000000001, m.repeated_double.pop | |
| 88 assert_equal 'bar', m.repeated_string.pop | |
| 89 assert_equal 'foo', m.repeated_string.pop | |
| 90 assert_equal "foo".encode!('ASCII-8BIT'), m.repeated_bytes.pop | |
| 91 assert_equal "bar".encode!('ASCII-8BIT'), m.repeated_bytes.pop | |
| 92 assert_equal TestMessage2.new(:foo => 2), m.repeated_msg.pop | |
| 93 assert_equal TestMessage2.new(:foo => 1), m.repeated_msg.pop | |
| 94 assert_equal :B, m.repeated_enum.pop | |
| 95 assert_equal :A, m.repeated_enum.pop | |
| 96 repeated_field_names(TestMessage).each do |field_name| | |
| 97 assert_nil m.send(field_name).pop | |
| 98 end | |
| 99 | |
| 100 fill_test_msg(m) | |
| 101 assert_equal ['bar', 'foo'], m.repeated_string.pop(2) | |
| 102 assert_nil m.repeated_string.pop | |
| 103 end | |
| 104 | |
| 105 | |
| 106 def test_each | |
| 107 m = TestMessage.new | |
| 108 5.times{|i| m.repeated_string << 'string' } | |
| 109 count = 0 | |
| 110 m.repeated_string.each do |val| | |
| 111 assert_equal 'string', val | |
| 112 count += 1 | |
| 113 end | |
| 114 assert_equal 5, count | |
| 115 result = m.repeated_string.each{|val| val + '_junk'} | |
| 116 assert_equal ['string'] * 5, result | |
| 117 end | |
| 118 | |
| 119 | |
| 120 def test_empty? | |
| 121 m = TestMessage.new | |
| 122 assert_equal true, m.repeated_string.empty? | |
| 123 m.repeated_string << 'foo' | |
| 124 assert_equal false, m.repeated_string.empty? | |
| 125 m.repeated_string << 'bar' | |
| 126 assert_equal false, m.repeated_string.empty? | |
| 127 end | |
| 128 | |
| 129 def test_array_accessor | |
| 130 m = TestMessage.new | |
| 131 reference_arr = %w(foo bar baz) | |
| 132 m.repeated_string += reference_arr.clone | |
| 133 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 134 arr[1] | |
| 135 end | |
| 136 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 137 arr[-2] | |
| 138 end | |
| 139 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 140 arr[20] | |
| 141 end | |
| 142 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 143 arr[1, 2] | |
| 144 end | |
| 145 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 146 arr[0..2] | |
| 147 end | |
| 148 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 149 arr[-1, 1] | |
| 150 end | |
| 151 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 152 arr[10, 12] | |
| 153 end | |
| 154 end | |
| 155 | |
| 156 def test_array_settor | |
| 157 m = TestMessage.new | |
| 158 reference_arr = %w(foo bar baz) | |
| 159 m.repeated_string += reference_arr.clone | |
| 160 | |
| 161 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 162 arr[1] = 'junk' | |
| 163 end | |
| 164 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 165 arr[-2] = 'snappy' | |
| 166 end | |
| 167 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 168 arr[3] = '' | |
| 169 end | |
| 170 # slight deviation; we are strongly typed, and nil is not allowed | |
| 171 # for string types; | |
| 172 m.repeated_string[5] = 'spacious' | |
| 173 assert_equal ["foo", "snappy", "baz", "", "", "spacious"], m.repeated_string | |
| 174 | |
| 175 #make sure it sests the default types for other fields besides strings | |
| 176 %w(repeated_int32 repeated_int64 repeated_uint32 repeated_uint64).each do |f
ield_name| | |
| 177 m.send(field_name)[3] = 10 | |
| 178 assert_equal [0,0,0,10], m.send(field_name) | |
| 179 end | |
| 180 m.repeated_float[3] = 10.1 | |
| 181 #wonky mri float handling | |
| 182 assert_equal [0,0,0], m.repeated_float.to_a[0..2] | |
| 183 assert_equal 10.1, m.repeated_float[3].round(1) | |
| 184 m.repeated_double[3] = 10.1 | |
| 185 assert_equal [0,0,0,10.1], m.repeated_double | |
| 186 m.repeated_bool[3] = true | |
| 187 assert_equal [false, false, false, true], m.repeated_bool | |
| 188 m.repeated_bytes[3] = "bar".encode!('ASCII-8BIT') | |
| 189 assert_equal ['', '', '', "bar".encode!('ASCII-8BIT')], m.repeated_bytes | |
| 190 m.repeated_msg[3] = TestMessage2.new(:foo => 1) | |
| 191 assert_equal [nil, nil, nil, TestMessage2.new(:foo => 1)], m.repeated_msg | |
| 192 m.repeated_enum[3] = :A | |
| 193 assert_equal [:Default, :Default, :Default, :A], m.repeated_enum | |
| 194 | |
| 195 # check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 196 # arr[20] = 'spacious' | |
| 197 # end | |
| 198 # TODO: accessor doesn't allow other ruby-like methods | |
| 199 # check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 200 # arr[1, 2] = 'fizz' | |
| 201 # end | |
| 202 # check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 203 # arr[0..2] = 'buzz' | |
| 204 # end | |
| 205 end | |
| 206 | |
| 207 def test_push | |
| 208 m = TestMessage.new | |
| 209 reference_arr = %w(foo bar baz) | |
| 210 m.repeated_string += reference_arr.clone | |
| 211 | |
| 212 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 213 arr.push('fizz') | |
| 214 end | |
| 215 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 216 arr << 'fizz' | |
| 217 end | |
| 218 #TODO: push should support multiple | |
| 219 # check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 220 # arr.push('fizz', 'buzz') | |
| 221 # end | |
| 222 end | |
| 223 | |
| 224 def test_clear | |
| 225 m = TestMessage.new | |
| 226 reference_arr = %w(foo bar baz) | |
| 227 m.repeated_string += reference_arr.clone | |
| 228 | |
| 229 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 230 arr.clear | |
| 231 end | |
| 232 end | |
| 233 | |
| 234 def test_concat | |
| 235 m = TestMessage.new | |
| 236 reference_arr = %w(foo bar baz) | |
| 237 m.repeated_string += reference_arr.clone | |
| 238 m.repeated_string.concat(['fizz', 'buzz']) | |
| 239 assert_equal %w(foo bar baz fizz buzz), m.repeated_string | |
| 240 #TODO: concat should return the orig array | |
| 241 # check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 242 # arr.concat(['fizz', 'buzz']) | |
| 243 # end | |
| 244 end | |
| 245 | |
| 246 def test_equal | |
| 247 m = TestMessage.new | |
| 248 reference_arr = %w(foo bar baz) | |
| 249 m.repeated_string += reference_arr.clone | |
| 250 assert_equal reference_arr, m.repeated_string | |
| 251 reference_arr << 'fizz' | |
| 252 assert_not_equal reference_arr, m.repeated_string | |
| 253 m.repeated_string << 'fizz' | |
| 254 assert_equal reference_arr, m.repeated_string | |
| 255 end | |
| 256 | |
| 257 def test_hash | |
| 258 # just a sanity check | |
| 259 m = TestMessage.new | |
| 260 reference_arr = %w(foo bar baz) | |
| 261 m.repeated_string += reference_arr.clone | |
| 262 assert m.repeated_string.hash.is_a?(Integer) | |
| 263 hash = m.repeated_string.hash | |
| 264 assert_equal hash, m.repeated_string.hash | |
| 265 m.repeated_string << 'j' | |
| 266 assert_not_equal hash, m.repeated_string.hash | |
| 267 end | |
| 268 | |
| 269 def test_plus | |
| 270 m = TestMessage.new | |
| 271 reference_arr = %w(foo bar baz) | |
| 272 m.repeated_string += reference_arr.clone | |
| 273 | |
| 274 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 275 arr + ['fizz', 'buzz'] | |
| 276 end | |
| 277 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 278 arr += ['fizz', 'buzz'] | |
| 279 end | |
| 280 end | |
| 281 | |
| 282 def test_replace | |
| 283 m = TestMessage.new | |
| 284 reference_arr = %w(foo bar baz) | |
| 285 m.repeated_string += reference_arr.clone | |
| 286 | |
| 287 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 288 arr.replace(['fizz', 'buzz']) | |
| 289 end | |
| 290 end | |
| 291 | |
| 292 def test_to_a | |
| 293 m = TestMessage.new | |
| 294 reference_arr = %w(foo bar baz) | |
| 295 m.repeated_string += reference_arr.clone | |
| 296 | |
| 297 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 298 arr.to_a | |
| 299 end | |
| 300 end | |
| 301 | |
| 302 def test_to_ary | |
| 303 m = TestMessage.new | |
| 304 reference_arr = %w(foo bar baz) | |
| 305 m.repeated_string += reference_arr.clone | |
| 306 | |
| 307 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 308 arr.to_ary | |
| 309 end | |
| 310 end | |
| 311 | |
| 312 # emulate Array behavior | |
| 313 ########################## | |
| 314 | |
| 315 def test_collect! | |
| 316 m = TestMessage.new | |
| 317 reference_arr = %w(foo bar baz) | |
| 318 m.repeated_string += reference_arr.clone | |
| 319 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 320 arr.collect!{|x| x + "!" } | |
| 321 end | |
| 322 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 323 arr.collect!.with_index{|x, i| x[0...i] } | |
| 324 end | |
| 325 end | |
| 326 | |
| 327 def test_compact! | |
| 328 m = TestMessage.new | |
| 329 m.repeated_msg << TestMessage2.new(:foo => 1) | |
| 330 m.repeated_msg << nil | |
| 331 m.repeated_msg << TestMessage2.new(:foo => 2) | |
| 332 reference_arr = m.repeated_string.to_a | |
| 333 | |
| 334 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 335 arr.compact! | |
| 336 end | |
| 337 end | |
| 338 | |
| 339 def test_delete | |
| 340 m = TestMessage.new | |
| 341 reference_arr = %w(foo bar baz) | |
| 342 m.repeated_string += reference_arr.clone | |
| 343 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 344 arr.delete('bar') | |
| 345 end | |
| 346 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 347 arr.delete('nope') | |
| 348 end | |
| 349 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 350 arr.delete('nope'){'within'} | |
| 351 end | |
| 352 end | |
| 353 | |
| 354 def test_delete_at | |
| 355 m = TestMessage.new | |
| 356 reference_arr = %w(foo bar baz) | |
| 357 m.repeated_string += reference_arr.clone | |
| 358 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 359 arr.delete_at(2) | |
| 360 end | |
| 361 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 362 arr.delete_at(10) | |
| 363 end | |
| 364 end | |
| 365 | |
| 366 def test_fill | |
| 367 m = TestMessage.new | |
| 368 reference_arr = %w(foo bar baz) | |
| 369 m.repeated_string += reference_arr.clone | |
| 370 | |
| 371 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 372 arr.fill("x") | |
| 373 end | |
| 374 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 375 arr.fill("z", 2, 2) | |
| 376 end | |
| 377 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 378 arr.fill("y", 0..1) | |
| 379 end | |
| 380 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 381 arr.fill { |i| (i*i).to_s } | |
| 382 end | |
| 383 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 384 arr.fill(-2) { |i| (i*i*i).to_s } | |
| 385 end | |
| 386 end | |
| 387 | |
| 388 def test_flatten! | |
| 389 m = TestMessage.new | |
| 390 reference_arr = %w(foo bar baz) | |
| 391 m.repeated_string += reference_arr.clone | |
| 392 | |
| 393 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 394 arr.flatten! | |
| 395 end | |
| 396 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 397 arr.flatten!(1) | |
| 398 end | |
| 399 end | |
| 400 | |
| 401 def test_insert | |
| 402 m = TestMessage.new | |
| 403 reference_arr = %w(foo bar baz) | |
| 404 m.repeated_string += reference_arr.clone | |
| 405 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 406 arr.insert(2, 'fizz') | |
| 407 end | |
| 408 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 409 arr.insert(3, 'fizz', 'buzz', 'bazz') | |
| 410 end | |
| 411 end | |
| 412 | |
| 413 def test_inspect | |
| 414 m = TestMessage.new | |
| 415 assert_equal '[]', m.repeated_string.inspect | |
| 416 m.repeated_string << 'foo' | |
| 417 assert_equal m.repeated_string.to_a.inspect, m.repeated_string.inspect | |
| 418 m.repeated_string << 'bar' | |
| 419 assert_equal m.repeated_string.to_a.inspect, m.repeated_string.inspect | |
| 420 end | |
| 421 | |
| 422 def test_reverse! | |
| 423 m = TestMessage.new | |
| 424 reference_arr = %w(foo bar baz) | |
| 425 m.repeated_string += reference_arr.clone | |
| 426 | |
| 427 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 428 arr.reverse! | |
| 429 end | |
| 430 end | |
| 431 | |
| 432 def test_rotate! | |
| 433 m = TestMessage.new | |
| 434 reference_arr = %w(foo bar baz) | |
| 435 m.repeated_string += reference_arr.clone | |
| 436 | |
| 437 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 438 arr.rotate! | |
| 439 end | |
| 440 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 441 arr.rotate!(2) | |
| 442 end | |
| 443 end | |
| 444 | |
| 445 def test_select! | |
| 446 m = TestMessage.new | |
| 447 reference_arr = %w(foo bar baz) | |
| 448 m.repeated_string += reference_arr.clone | |
| 449 | |
| 450 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 451 arr.select! { |v| v =~ /[aeiou]/ } | |
| 452 end | |
| 453 end | |
| 454 | |
| 455 def test_shift | |
| 456 m = TestMessage.new | |
| 457 reference_arr = %w(foo bar baz) | |
| 458 m.repeated_string += reference_arr.clone | |
| 459 | |
| 460 # should return an element | |
| 461 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 462 arr.shift | |
| 463 end | |
| 464 # should return an array | |
| 465 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 466 arr.shift(2) | |
| 467 end | |
| 468 # should return nil | |
| 469 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 470 arr.shift | |
| 471 end | |
| 472 end | |
| 473 | |
| 474 def test_shuffle! | |
| 475 m = TestMessage.new | |
| 476 m.repeated_string += %w(foo bar baz) | |
| 477 orig_repeated_string = m.repeated_string.clone | |
| 478 result = m.repeated_string.shuffle! | |
| 479 assert_equal m.repeated_string, result | |
| 480 # NOTE: sometimes it doesn't change the order... | |
| 481 # assert_not_equal m.repeated_string.to_a, orig_repeated_string.to_a | |
| 482 end | |
| 483 | |
| 484 def test_slice! | |
| 485 m = TestMessage.new | |
| 486 reference_arr = %w(foo bar baz bar fizz buzz) | |
| 487 m.repeated_string += reference_arr.clone | |
| 488 | |
| 489 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 490 arr.slice!(2) | |
| 491 end | |
| 492 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 493 arr.slice!(1,2) | |
| 494 end | |
| 495 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 496 arr.slice!(0..1) | |
| 497 end | |
| 498 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 499 arr.slice!(10) | |
| 500 end | |
| 501 end | |
| 502 | |
| 503 def test_sort! | |
| 504 m = TestMessage.new | |
| 505 reference_arr = %w(foo bar baz) | |
| 506 m.repeated_string += reference_arr.clone | |
| 507 | |
| 508 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 509 arr.sort! | |
| 510 end | |
| 511 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 512 arr.sort! { |x,y| y <=> x } | |
| 513 end | |
| 514 end | |
| 515 | |
| 516 def test_sort_by! | |
| 517 m = TestMessage.new | |
| 518 reference_arr = %w(foo bar baz) | |
| 519 m.repeated_string += reference_arr.clone | |
| 520 | |
| 521 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 522 arr.sort_by! | |
| 523 end | |
| 524 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 525 arr.sort_by!(&:hash) | |
| 526 end | |
| 527 end | |
| 528 | |
| 529 def test_uniq! | |
| 530 m = TestMessage.new | |
| 531 reference_arr = %w(foo bar baz) | |
| 532 m.repeated_string += reference_arr.clone | |
| 533 | |
| 534 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 535 arr.uniq! | |
| 536 end | |
| 537 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 538 arr.uniq!{|s| s[0] } | |
| 539 end | |
| 540 end | |
| 541 | |
| 542 def test_unshift | |
| 543 m = TestMessage.new | |
| 544 reference_arr = %w(foo bar baz) | |
| 545 m.repeated_string += reference_arr.clone | |
| 546 | |
| 547 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 548 arr.unshift('1') | |
| 549 end | |
| 550 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 551 arr.unshift('a', 'b') | |
| 552 end | |
| 553 check_self_modifying_method(m.repeated_string, reference_arr) do |arr| | |
| 554 arr.unshift('') | |
| 555 end | |
| 556 end | |
| 557 | |
| 558 | |
| 559 ##### HELPER METHODS | |
| 560 | |
| 561 def check_self_modifying_method(repeated_field, ref_array) | |
| 562 expected_result = yield(ref_array) | |
| 563 actual_result = yield(repeated_field) | |
| 564 if expected_result.is_a?(Enumerator) | |
| 565 assert_equal expected_result.to_a, actual_result.to_a | |
| 566 else | |
| 567 assert_equal expected_result, actual_result | |
| 568 end | |
| 569 assert_equal ref_array, repeated_field | |
| 570 end | |
| 571 | |
| 572 | |
| 573 def repeated_field_names(klass) | |
| 574 klass.descriptor.find_all{|f| f.label == :repeated}.map(&:name) | |
| 575 end | |
| 576 | |
| 577 | |
| 578 def fill_test_msg(test_msg) | |
| 579 test_msg.repeated_int32 += [-10, -11] | |
| 580 test_msg.repeated_int64 += [-1_000_000, -1_000_001] | |
| 581 test_msg.repeated_uint32 += [10, 11] | |
| 582 test_msg.repeated_uint64 += [1_000_000, 1_000_001] | |
| 583 test_msg.repeated_bool += [true, false] | |
| 584 test_msg.repeated_float += [-1.01, -1.02] | |
| 585 test_msg.repeated_double += [-1.0000000000001, -1.0000000000002] | |
| 586 test_msg.repeated_string += %w(foo bar) | |
| 587 test_msg.repeated_bytes += ["bar".encode!('ASCII-8BIT'), "foo".encode!('ASC
II-8BIT')] | |
| 588 test_msg.repeated_msg << TestMessage2.new(:foo => 1) | |
| 589 test_msg.repeated_msg << TestMessage2.new(:foo => 2) | |
| 590 test_msg.repeated_enum << :A | |
| 591 test_msg.repeated_enum << :B | |
| 592 end | |
| 593 | |
| 594 | |
| 595 pool = Google::Protobuf::DescriptorPool.new | |
| 596 pool.build do | |
| 597 | |
| 598 add_message "TestMessage" do | |
| 599 optional :optional_int32, :int32, 1 | |
| 600 optional :optional_int64, :int64, 2 | |
| 601 optional :optional_uint32, :uint32, 3 | |
| 602 optional :optional_uint64, :uint64, 4 | |
| 603 optional :optional_bool, :bool, 5 | |
| 604 optional :optional_float, :float, 6 | |
| 605 optional :optional_double, :double, 7 | |
| 606 optional :optional_string, :string, 8 | |
| 607 optional :optional_bytes, :bytes, 9 | |
| 608 optional :optional_msg, :message, 10, "TestMessage2" | |
| 609 optional :optional_enum, :enum, 11, "TestEnum" | |
| 610 | |
| 611 repeated :repeated_int32, :int32, 12 | |
| 612 repeated :repeated_int64, :int64, 13 | |
| 613 repeated :repeated_uint32, :uint32, 14 | |
| 614 repeated :repeated_uint64, :uint64, 15 | |
| 615 repeated :repeated_bool, :bool, 16 | |
| 616 repeated :repeated_float, :float, 17 | |
| 617 repeated :repeated_double, :double, 18 | |
| 618 repeated :repeated_string, :string, 19 | |
| 619 repeated :repeated_bytes, :bytes, 20 | |
| 620 repeated :repeated_msg, :message, 21, "TestMessage2" | |
| 621 repeated :repeated_enum, :enum, 22, "TestEnum" | |
| 622 end | |
| 623 add_message "TestMessage2" do | |
| 624 optional :foo, :int32, 1 | |
| 625 end | |
| 626 | |
| 627 add_enum "TestEnum" do | |
| 628 value :Default, 0 | |
| 629 value :A, 1 | |
| 630 value :B, 2 | |
| 631 value :C, 3 | |
| 632 end | |
| 633 end | |
| 634 | |
| 635 TestMessage = pool.lookup("TestMessage").msgclass | |
| 636 TestMessage2 = pool.lookup("TestMessage2").msgclass | |
| 637 TestEnum = pool.lookup("TestEnum").enummodule | |
| 638 | |
| 639 | |
| 640 end | |
| OLD | NEW |