protobuf/ruby/tests/object_cache_test.rb

require 'google/protobuf'
require 'test/unit'

class PlatformTest < Test::Unit::TestCase
  def test_correct_implementation_for_platform
    if RUBY_PLATFORM == "java"
      return
    end

    if Gem::Version.new(RUBY_VERSION) >= Gem::Version.new('2.7.0') and Google::Protobuf::SIZEOF_LONG >= Google::Protobuf::SIZEOF_VALUE
      assert_instance_of Google::Protobuf::ObjectCache, Google::Protobuf::OBJECT_CACHE
    else
      assert_instance_of Google::Protobuf::LegacyObjectCache, Google::Protobuf::OBJECT_CACHE
    end
  end
end

module ObjectCacheTestModule
  def test_try_add_returns_existing_value
    cache = self.create

    keys = ["k1", "k2"]
    vals = ["v1", "v2"]
    2.times do |i|
      assert_same vals[i], cache.try_add(keys[i], vals[i])
      assert_same vals[i], cache.get(keys[i])
      assert_same vals[i], cache.try_add(keys[i], vals[(i+1)%2])
      assert_same vals[i], cache.get(keys[i])
    end
  end

  def test_multithreaded_access
    cache = self.create

    keys = ["k0", "k1", "k2", "k3", "k4", "k5", "k6", "k7"]

    threads = []
    100.times do |i|
      threads[i] = Thread.new {
        Thread.current["result"] = cache.try_add(keys[i % keys.size], i.to_s)
      }
    end

    results = {}
    threads.each_with_index {|t, i| t.join; results[i] = t["result"] }
    assert_equal 100, results.size
    assert_equal 8, results.values.uniq.size
  end
end

if RUBY_PLATFORM == "java"
  return
end

class ObjectCacheTest < Test::Unit::TestCase
  def create
    Google::Protobuf::ObjectCache.new
  end

  include ObjectCacheTestModule
end

class LegacyObjectCacheTest < Test::Unit::TestCase
  def create
    Google::Protobuf::LegacyObjectCache.new
  end

  include ObjectCacheTestModule
end